diff --git a/tritonbench/utils/python_utils.py b/tritonbench/utils/python_utils.py
index f71d19121..2179486fe 100644
--- a/tritonbench/utils/python_utils.py
+++ b/tritonbench/utils/python_utils.py
@@ -9,5 +9,5 @@ def try_import(cond_name: str):
     try:
         yield
         _caller_globals[cond_name] = True
-    except (ImportError, ModuleNotFoundError, OSError) as e:
+    except (ImportError, ModuleNotFoundError, OSError, AttributeError) as e:
         _caller_globals[cond_name] = False