Skip to content

Commit f48ad1e

Browse files
xuzhao9facebook-github-bot
authored andcommitted
Triton kernel requires TMA
Summary: Fix a minor bug in fp8_gemm_rowwise where TMA is not available on AMD/HIP Reviewed By: nmacchioni Differential Revision: D65915379 fbshipit-source-id: de6effd2f7e830115eadaf17a4c1add2101ef78e
1 parent c2273e5 commit f48ad1e

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tritonbench/operators/fp8_gemm_rowwise/operator.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,11 @@ def parse_args(args: List[str]) -> argparse.Namespace:
4343
matmul_fp8_row as triton_fp8_row,
4444
)
4545

46+
assert hasattr(
47+
triton.runtime.driver.active.utils, "fill_1d_tma_descriptor"
48+
), "TMA is required by the Triton kernel."
4649
HAS_TRITON = True
47-
except ImportError:
50+
except (ImportError, AssertionError):
4851
HAS_TRITON = False
4952

5053

0 commit comments

Comments
 (0)