Enable TLX kernels when TLX is available

Differential Revision: D83983601

Pull Request resolved: #517

Author: htyu

tritonbench/operators/blackwell_attentions/operator.py

@@ -504,7 +504,7 @@ def gluon_blackwell_tutorial_persistent_fwd(
         )
 
     # Only works with triton beta, forward only.
-    @register_benchmark(enabled=False)
+    @register_benchmark(enabled=HAS_TLX)
     def tlx_blackwell_ws_pipelined_fwd(
         self,
         q: torch.Tensor,
@@ -514,7 +514,7 @@ def tlx_blackwell_ws_pipelined_fwd(
         return lambda: tlx_blackwell_fwd(q, k, v, self.sm_scale, False)
 
     # Only works with triton beta, forward only.
-    @register_benchmark(enabled=False)
+    @register_benchmark(enabled=HAS_TLX)
     def tlx_blackwell_ws_pipelined_persistent_fwd(
         self,
         q: torch.Tensor,