From a2e2e1d8c026951baa345f0dd17668bd1718eda5 Mon Sep 17 00:00:00 2001
From: Sam Larsen <slarsen@meta.com>
Date: Fri, 10 Oct 2025 17:57:37 +0000
Subject: [PATCH] Add pytorch_version and mast_application_packages to pt2
 compile scuba logging (#165018)

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
---
 test/dynamo/test_utils.py | 5 +++++
 torch/_dynamo/utils.py    | 2 ++
 2 files changed, 7 insertions(+)

diff --git a/test/dynamo/test_utils.py b/test/dynamo/test_utils.py
index b7166c5ce6d1..1708da900056 100644
--- a/test/dynamo/test_utils.py
+++ b/test/dynamo/test_utils.py
@@ -500,6 +500,7 @@ class TestDynamoTimed(TestCase):
             e.cuda_version = None
             e.triton_version = None
             e.python_version = None
+            e.pytorch_version = None
             e.stack_trace = None
             e.graph_node_shapes = None
             e.exception_stack_trace = None
@@ -576,6 +577,7 @@ class TestDynamoTimed(TestCase):
  'post_grad_pass_time_us': 0,
  'pre_grad_pass_time_us': 0,
  'python_version': None,
+ 'pytorch_version': None,
  'recompile_reason': None,
  'recompile_user_contexts': None,
  'remote_cache_time_saved_s': None,
@@ -661,6 +663,7 @@ class TestDynamoTimed(TestCase):
  'post_grad_pass_time_us': 0,
  'pre_grad_pass_time_us': 0,
  'python_version': None,
+ 'pytorch_version': None,
  'recompile_reason': None,
  'recompile_user_contexts': None,
  'remote_cache_time_saved_s': None,
@@ -757,6 +760,7 @@ class TestDynamoTimed(TestCase):
  'post_grad_pass_time_us': 0,
  'pre_grad_pass_time_us': None,
  'python_version': None,
+ 'pytorch_version': None,
  'recompile_reason': None,
  'recompile_user_contexts': None,
  'remote_cache_time_saved_s': None,
@@ -842,6 +846,7 @@ class TestDynamoTimed(TestCase):
  'post_grad_pass_time_us': 0,
  'pre_grad_pass_time_us': None,
  'python_version': None,
+ 'pytorch_version': None,
  'recompile_reason': None,
  'recompile_user_contexts': None,
  'remote_cache_time_saved_s': None,
diff --git a/torch/_dynamo/utils.py b/torch/_dynamo/utils.py
index 8da851d66b98..0e0c69c8548f 100644
--- a/torch/_dynamo/utils.py
+++ b/torch/_dynamo/utils.py
@@ -1375,6 +1375,7 @@ class CompilationMetrics:
     param_count: Optional[int] = None
     recompile_user_contexts: Optional[set[str]] = None
     inline_inbuilt_nn_modules_candidate: Optional[bool] = False
+    pytorch_version: Optional[str] = None
 
     @classmethod
     def create(cls, metrics: dict[str, Any]) -> CompilationMetrics:
@@ -1645,6 +1646,7 @@ def record_compilation_metrics(
         "remote_cache_version": remote_cache_version,
         "inductor_fx_remote_cache_backend_type": inductor_fx_remote_cache_backend_type,
         "python_version": sys.version,
+        "pytorch_version": torch.__version__,
     }
 
     compilation_metrics = CompilationMetrics.create({**common_metrics, **metrics})