From 200ca77491d5e44fe3ff600c1df0ec380ae540bf Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Thu, 21 Aug 2025 09:45:26 -0700 Subject: [PATCH] always enable cuda graph --- tritonbench/utils/triton_op.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tritonbench/utils/triton_op.py b/tritonbench/utils/triton_op.py index ae67d603e..594270fc1 100644 --- a/tritonbench/utils/triton_op.py +++ b/tritonbench/utils/triton_op.py @@ -681,7 +681,7 @@ class BenchmarkOperator(metaclass=PostInitProcessor): _input_iter: Optional[Generator] = None extra_args: List[str] = [] example_inputs: Any = None - use_cuda_graphs: bool = False + use_cuda_graphs: bool = True is_compute_bound = True # reset dynamo to avoid errors like https://github.com/meta-pytorch/tritonbench/issues/90 reset_dynamo = True