diff --git a/tritonbench/operators/gemm/warp_spec_persistent_matmul.py b/tritonbench/operators/gemm/warp_spec_persistent_matmul.py
index b97682c34..d04e3a29a 100644
--- a/tritonbench/operators/gemm/warp_spec_persistent_matmul.py
+++ b/tritonbench/operators/gemm/warp_spec_persistent_matmul.py
@@ -9,7 +9,13 @@
 import torch
 import triton
 import triton.language as tl
-from triton.tools.tensor_descriptor import TensorDescriptor
+
+# Conditional import for TensorDescriptor - only available in newer Triton versions
+try:
+    from triton.tools.tensor_descriptor import TensorDescriptor
+except ModuleNotFoundError:
+    TensorDescriptor = None
+
 
 # TODO: Add proton support