diff --git a/tritonbench/utils/triton_op.py b/tritonbench/utils/triton_op.py index e1196694..4fe6d159 100644 --- a/tritonbench/utils/triton_op.py +++ b/tritonbench/utils/triton_op.py @@ -693,7 +693,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