From 34038d56da7667e31829974213a5a9115fb3c00d Mon Sep 17 00:00:00 2001 From: "Sam Larsen (Meta Employee)" Date: Mon, 9 Dec 2024 10:45:08 -0800 Subject: [PATCH] Add triton_compile_time_us column to dynamo_compile (#142068) Summary: X-link: https://github.com/pytorch/pytorch/pull/142068 Approved by: https://github.com/c00w Reviewed By: c00w, clee2000 Differential Revision: D66799565 Pulled By: masnesral fbshipit-source-id: 9231089fa0603d6e6e031f76f92460fb88264d7b --- userbenchmark/dynamo/dynamobench/_dynamo/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py index a4e033bcf..0f45dd084 100644 --- 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