diff --git a/test/registered/attention/test_wave_attention_kernels.py b/test/registered/attention/test_wave_attention_kernels.py index ddb417f41..0b92d57e4 100644 --- a/test/registered/attention/test_wave_attention_kernels.py +++ b/test/registered/attention/test_wave_attention_kernels.py @@ -25,7 +25,11 @@ from sglang.srt.utils import get_device from sglang.test.ci.ci_register import register_amd_ci # Wave attention kernel unit tests (AMD only - requires wave_lang) -register_amd_ci(est_time=60, suite="stage-a-test-1-gpu-small-amd") +register_amd_ci( + est_time=60, + suite="stage-a-test-1-gpu-small-amd", + disabled="Wave prefill corrupts the native heap on ROCm 10", +) class TestWaveAttention(unittest.TestCase):