diff --git a/op_tests/triton_tests/attention/test_pa_mqa_logits_mxfp4.py b/op_tests/triton_tests/attention/test_pa_mqa_logits_mxfp4.py index 995bb6625b6..445b9258f7f 100644 --- a/op_tests/triton_tests/attention/test_pa_mqa_logits_mxfp4.py +++ b/op_tests/triton_tests/attention/test_pa_mqa_logits_mxfp4.py @@ -11,8 +11,14 @@ mfma_nonk_dim, paged_mxfp4_mqa_logits, ) +from aiter.ops.triton.utils._triton import arch_info from op_tests.triton_tests.utils.pa_mqa_logits_mxfp4_ref import preshuffle_cache +pytestmark = pytest.mark.skipif( + arch_info.get_arch() != "gfx950", + reason="paged MXFP4 MQA logits is gfx950 only", +) + SCALE_GROUP = 32 TOL = 1e-12 E8M0_BIAS = 127 diff --git a/op_tests/triton_tests/rope/test_rope_mxfp4.py b/op_tests/triton_tests/rope/test_rope_mxfp4.py index deb4ec55e6c..b995ddf76b7 100644 --- a/op_tests/triton_tests/rope/test_rope_mxfp4.py +++ b/op_tests/triton_tests/rope/test_rope_mxfp4.py @@ -9,6 +9,7 @@ ) from aiter.ops.triton.fusions.k_norm_rope_mxfp4_cache import k_norm_rope_mxfp4_cache from aiter.ops.triton.rope.q_rope_mxfp4_quant import q_rope_mxfp4_quant +from aiter.ops.triton.utils._triton import arch_info from op_tests.triton_tests.utils.pa_mqa_logits_mxfp4_ref import ( unshuffle_scales, unshuffle_values, @@ -244,6 +245,10 @@ def test_q_quant_matches_reference(num_tokens, num_heads, with_weights): @pytest.mark.parametrize("page_size", [64, 128]) +@pytest.mark.skipif( + arch_info.get_arch() != "gfx950", + reason="paged MXFP4 MQA logits is gfx950 only", +) def test_round_trip_through_logits(page_size): """Keys written by the cache op and a query from the quant op score as the dequantized values do: the writer and paged_mxfp4_mqa_logits agree."""