From 2f5b0178e5c817fb3b0b753b27bee72149a18036 Mon Sep 17 00:00:00 2001 From: Bohan Hou Date: Sat, 5 Sep 2026 21:28:55 +0000 Subject: [PATCH] [TIRx][Test] Gate collector-A-only MMA on CUDA 13.4 --- tests/python/tirx/codegen/test_ptx_dialect.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/python/tirx/codegen/test_ptx_dialect.py b/tests/python/tirx/codegen/test_ptx_dialect.py index 835979b6f2f9..0e4737292f52 100644 --- a/tests/python/tirx/codegen/test_ptx_dialect.py +++ b/tests/python/tirx/codegen/test_ptx_dialect.py @@ -2419,6 +2419,10 @@ def sm107_collector_kernel(a_ptr: T.handle): _assert_ptxas_ok(collector_src, arch="sm_107f") +@pytest.mark.skipif( + not env.has_nvcc_version(13, 4), + reason="collector-qualified block_scale MMA is a PTX 9.4 form; need nvcc >= 13.4", +) @requires_nvcc def test_ptx_tcgen05_mma_block_scale_collector_a_without_block_size(): """SM107 activation-stationary FP8 accepts collector A without `.block*`."""