From 2b41fa979fed568535204f1ce62ff4efae6a0745 Mon Sep 17 00:00:00 2001
From: Xu Zhao <xzhao9@fb.com>
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 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