diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py index e5bbc37f2..32305531c 100644 --- a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py +++ b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py @@ -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 @@ -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" ),