Add triton_compile_time_us column to dynamo_compile (#142068)

Summary: X-link: pytorch/pytorch#142068 Approved by: https://github.com/c00w Reviewed By: c00w, clee2000 Differential Revision: D66799565 Pulled By: masnesral fbshipit-source-id: 9231089fa0603d6e6e031f76f92460fb88264d7b
pytorch · Dec 9, 2024 · 34038d5 · 34038d5
1 parent 692bf64
commit 34038d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py
@@ -856,7 +856,7 @@ class CompilationMetrics:
     aot_autograd_cumulative_compile_time_us: Optional[int] = None
     inductor_cumulative_compile_time_us: Optional[int] = None
     inductor_code_gen_cumulative_compile_time_us: Optional[int] = None
-    triton_compile_time_us: Optional[int] = None  # TODO: instrument
+    triton_compile_time_us: Optional[int] = None
     runtime_cudagraphify_time_us: Optional[int] = None  # TODO: instrument
     runtime_triton_autotune_time_us: Optional[int] = None  # TODO: instrument
     dynamo_compile_time_before_restart_us: Optional[int] = None