[logs] Add autotuning data (#154771)

stashuk-olek · pytorchmergebot · commit 64436c38c956 · 2025-06-06T17:12:55.000Z
Summary: Add autotuning logging data to scuba/chrome trace. Test Plan: ``` TORCHINDUCTOR_MAX_AUTOTUNE=1 tlp buck run //scripts/sashko:compilation_sample ``` Open https://interncache-all.fbcdn.net/manifold/perfetto-artifacts/tree/ui/index.html#!/viewer?local_cache_key=00000000-0000-0000-92db-f23383ebf5b5, search for template_autotuning, see in metadata strides (see screenshot) Differential Revision: D75457770 Pull Request resolved: #154771 Approved by: https://github.com/masnesral, https://github.com/PaulZhang12
diff --git a/torch/_inductor/select_algorithm.py b/torch/_inductor/select_algorithm.py
@@ -2199,10 +2199,21 @@ def make_benchmark_fn():
 
         def autotune(choices):
             log.debug("Starting autotuning")
+
+            autotune_data = {
+                "shape": ", ".join(
+                    ["x".join(map(str, n.get_size())) for n in input_nodes]
+                ),
+                "strides": ", ".join([str(n.get_stride()) for n in input_nodes]),
+                "dtypes": ", ".join([str(n.get_dtype()) for n in input_nodes]),
+                "offset": ", ".join([str(n.get_layout().offset) for n in input_nodes]),
+            }
+
             with dynamo_timed(
                 f"{name}_template_autotuning",
                 log_pt2_compile_event=True,
                 dynamo_compile_column_us="compile_time_autotune_time_us",
+                metadata={"autotune_data": autotune_data},
             ):
                 return make_benchmark_fn()(choices)
 
