Skip to content

Commit 200ca77

Browse files
committed
always enable cuda graph
1 parent 34e755a commit 200ca77

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tritonbench/utils/triton_op.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -681,7 +681,7 @@ class BenchmarkOperator(metaclass=PostInitProcessor):
681681
_input_iter: Optional[Generator] = None
682682
extra_args: List[str] = []
683683
example_inputs: Any = None
684-
use_cuda_graphs: bool = False
684+
use_cuda_graphs: bool = True
685685
is_compute_bound = True
686686
# reset dynamo to avoid errors like https://github.com/meta-pytorch/tritonbench/issues/90
687687
reset_dynamo = True

0 commit comments

Comments
 (0)