frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add pytorch_version and mast_application_packages to pt2 compile scuba logging (#165018)

Browse Source 
Summary: Two more fields requested for conda-on-mast jobs

Differential Revision: D84214442

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165018
Approved by: https://github.com/c00w
This commit is contained in:
Sam Larsen
2025-10-10 17:57:37 +00:00
committed by PyTorch MergeBot
parent b67785d9eb
commit a2e2e1d8c0
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/dynamo/test_utils.py
View File

@ -500,6 +500,7 @@ class TestDynamoTimed(TestCase):
e.cuda_version = None e.cuda_version = None
e.triton_version = None e.triton_version = None
e.python_version = None e.python_version = None
e.pytorch_version = None
e.stack_trace = None e.stack_trace = None
e.graph_node_shapes = None e.graph_node_shapes = None
e.exception_stack_trace = None e.exception_stack_trace = None
@ -576,6 +577,7 @@ class TestDynamoTimed(TestCase):
'post_grad_pass_time_us': 0, 'post_grad_pass_time_us': 0,
'pre_grad_pass_time_us': 0, 'pre_grad_pass_time_us': 0,
'python_version': None, 'python_version': None,
'pytorch_version': None,
'recompile_reason': None, 'recompile_reason': None,
'recompile_user_contexts': None, 'recompile_user_contexts': None,
'remote_cache_time_saved_s': None, 'remote_cache_time_saved_s': None,
@ -661,6 +663,7 @@ class TestDynamoTimed(TestCase):
'post_grad_pass_time_us': 0, 'post_grad_pass_time_us': 0,
'pre_grad_pass_time_us': 0, 'pre_grad_pass_time_us': 0,
'python_version': None, 'python_version': None,
'pytorch_version': None,
'recompile_reason': None, 'recompile_reason': None,
'recompile_user_contexts': None, 'recompile_user_contexts': None,
'remote_cache_time_saved_s': None, 'remote_cache_time_saved_s': None,
@ -757,6 +760,7 @@ class TestDynamoTimed(TestCase):
'post_grad_pass_time_us': 0, 'post_grad_pass_time_us': 0,
'pre_grad_pass_time_us': None, 'pre_grad_pass_time_us': None,
'python_version': None, 'python_version': None,
'pytorch_version': None,
'recompile_reason': None, 'recompile_reason': None,
'recompile_user_contexts': None, 'recompile_user_contexts': None,
'remote_cache_time_saved_s': None, 'remote_cache_time_saved_s': None,
@ -842,6 +846,7 @@ class TestDynamoTimed(TestCase):
'post_grad_pass_time_us': 0, 'post_grad_pass_time_us': 0,
'pre_grad_pass_time_us': None, 'pre_grad_pass_time_us': None,
'python_version': None, 'python_version': None,
'pytorch_version': None,
'recompile_reason': None, 'recompile_reason': None,
'recompile_user_contexts': None, 'recompile_user_contexts': None,
'remote_cache_time_saved_s': None, 'remote_cache_time_saved_s': None,

2
torch/_dynamo/utils.py
View File

@ -1375,6 +1375,7 @@ class CompilationMetrics:
param_count: Optional[int] = None param_count: Optional[int] = None
recompile_user_contexts: Optional[set[str]] = None recompile_user_contexts: Optional[set[str]] = None
inline_inbuilt_nn_modules_candidate: Optional[bool] = False inline_inbuilt_nn_modules_candidate: Optional[bool] = False
pytorch_version: Optional[str] = None
@classmethod @classmethod
def create(cls, metrics: dict[str, Any]) -> CompilationMetrics: def create(cls, metrics: dict[str, Any]) -> CompilationMetrics:
@ -1645,6 +1646,7 @@ def record_compilation_metrics(
"remote_cache_version": remote_cache_version, "remote_cache_version": remote_cache_version,
"inductor_fx_remote_cache_backend_type": inductor_fx_remote_cache_backend_type, "inductor_fx_remote_cache_backend_type": inductor_fx_remote_cache_backend_type,
"python_version": sys.version, "python_version": sys.version,
"pytorch_version": torch.__version__,
} }
compilation_metrics = CompilationMetrics.create({**common_metrics, **metrics}) compilation_metrics = CompilationMetrics.create({**common_metrics, **metrics})