[TRITON] fix gluon test skipping logic (#1566)

Boss2002n · web-flow · commit 2ae0991463b0 · 2025-12-04T20:13:56.000-05:00
diff --git a/op_tests/triton_tests/test_gemm_a8w8_blockscale.py b/op_tests/triton_tests/test_gemm_a8w8_blockscale.py
@@ -184,9 +184,9 @@ def test_gemm(dtype, M, N, K, layout, output, impl: str):
 
     block_shape_n, block_shape_k = block_shape
 
-    if impl == "gluon" and int(DEVICE_ARCH.split("MI")[1].replace("X", "")) < 350:
+    if impl == "gluon" and DEVICE_ARCH not in ("gfx950",):
         pytest.skip(
-            "Gluon implementation is not supported on this device (requires CDNA4)."
+            "Gluon implementation is not supported on this device (requires CDNA4/gfx950)."
         )
 
     dtype = str_to_torch_dtype[dtype]

Original file line number	Diff line number	Diff line change
`@@ -184,9 +184,9 @@ def test_gemm(dtype, M, N, K, layout, output, impl: str):`
`184`	`184`
`185`	`185`	`block_shape_n, block_shape_k = block_shape`
`186`	`186`
`187`		`- if impl == "gluon" and int(DEVICE_ARCH.split("MI")[1].replace("X", "")) < 350:`
	`187`	`+ if impl == "gluon" and DEVICE_ARCH not in ("gfx950",):`
`188`	`188`	`pytest.skip(`
`189`		`- "Gluon implementation is not supported on this device (requires CDNA4)."`
	`189`	`+ "Gluon implementation is not supported on this device (requires CDNA4/gfx950)."`
`190`	`190`	`)`
`191`	`191`
`192`	`192`	`dtype = str_to_torch_dtype[dtype]`