Fix tritonbench fp8_gemm_rowwise TMA assertion

Summary: Fix tritonbench fp8_gemm_rowwise TMA assertion

Reviewed By: htyu

Differential Revision: D66101658

fbshipit-source-id: 2f0bf11ab863238c52bc5336816b30bae24d025c

Author: karthik-man

tritonbench/operators/fp8_gemm_rowwise/operator.py

@@ -43,14 +43,14 @@ def parse_args(args: List[str]) -> argparse.Namespace:
         matmul_fp8_row as triton_fp8_row,
     )
 
-    assert hasattr(
-        triton.runtime.driver.active.utils, "fill_1d_tma_descriptor"
-    ), "TMA is required by the Triton kernel."
+    if not torch.version.hip:
+        assert hasattr(
+            triton.runtime.driver.active.utils, "fill_1d_tma_descriptor"
+        ), "TMA is required by the Triton kernel."
     HAS_TRITON = True
 except (ImportError, AssertionError):
     HAS_TRITON = False
 
-
 try:
     cutlass_or_ck_fp8_row = torch.ops.fbgemm.f8f8bf16_rowwise
     HAS_CUTLASS_OR_CK = True