Add backward compatibility for TensorDescriptor

This commits adds a guard around tensor_descriptor import in warp spec for backward compatibility

Signed-off-by: Bryan Bowyer <[email protected]>

Author: bdbowyer

tritonbench/operators/gemm/warp_spec_persistent_matmul.py

@@ -9,7 +9,12 @@
 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