Fix mode

pytorch-labs · Nov 18, 2024 · 19d6db9 · 19d6db9
1 parent 2133c0a
commit 19d6db9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tritonbench/utils/triton_op.py b/tritonbench/utils/triton_op.py
@@ -523,7 +523,7 @@ def __call__(cls, *args, **kwargs):
 
 def _translate_mode(tb_args):
     def _has_and_true(attr):
-        if hasattr(tb_args, attr) and attr:
+        if hasattr(tb_args, attr) and getattr(tb_args, attr):
             return True
         return False
     if _has_and_true("fwd"):