Skip to content

Commit

Permalink
Added cuda and triton versions to dynamo_compile (#141290)
Browse files Browse the repository at this point in the history
Summary:
Opening another PR since #141140 was reverted.

X-link: pytorch/pytorch#141290
Approved by: https://github.com/masnesral

Reviewed By: izaitsevfb

Differential Revision: D66384168

Pulled By: jovianjaison

fbshipit-source-id: 9e689094683a14cd919791cde8264dfb3fa95b17
  • Loading branch information
jovianjaison authored and facebook-github-bot committed Nov 23, 2024
1 parent 930f14b commit 03e12ce
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions userbenchmark/dynamo/dynamobench/_dynamo/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -870,6 +870,8 @@ class CompilationMetrics:
inductor_fx_remote_cache_backend_type: Optional[str] = None
inductor_fx_remote_cache_hit_keys: Optional[str] = None
inductor_fx_remote_cache_miss_keys: Optional[str] = None
cuda_version: Optional[str] = None
triton_version: Optional[str] = None
feature_usage: Optional[dict[str, bool]] = None


Expand Down Expand Up @@ -987,6 +989,8 @@ def safe_str(item: Any) -> str:

common_metrics = {
"inductor_config": _scrubbed_inductor_config_for_logging(),
"cuda_version": torch.version.cuda,
"triton_version": triton.__version__ if has_triton() else "",
"inductor_fx_remote_cache_hit_keys": _convert_collection_to_str(
"inductor_fx_remote_cache_hit_keys"
),
Expand Down

0 comments on commit 03e12ce

Please sign in to comment.