From 630e67b8452d32f3593ca67e8b519cab2cd36063 Mon Sep 17 00:00:00 2001 From: QI JUN <22017000+QiJune@users.noreply.github.com> Date: Mon, 25 Aug 2025 11:52:59 +0800 Subject: [PATCH] [None][ci] waive test_mamba2_chunk_scan_combined_prefill_chunking[seqlens1-8] (#7194) Signed-off-by: junq <22017000+QiJune@users.noreply.github.com> --- tests/unittest/_torch/thop/test_mamba2_chunk_ss_update.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/unittest/_torch/thop/test_mamba2_chunk_ss_update.py b/tests/unittest/_torch/thop/test_mamba2_chunk_ss_update.py index e26fe00776..8331857853 100644 --- a/tests/unittest/_torch/thop/test_mamba2_chunk_ss_update.py +++ b/tests/unittest/_torch/thop/test_mamba2_chunk_ss_update.py @@ -369,6 +369,8 @@ def test_mamba2_chunk_scan_selective_state_update(dim, headdim, ngroups, dstate, (16, 20), ]) def test_mamba2_chunk_scan_combined_prefill_chunking(mamba_chunk_size, seqlens): + if mamba_chunk_size == 8 and seqlens == (270, 88, 212, 203): + pytest.skip("https://nvbugspro.nvidia.com/bug/5477332") dim = 1024 headdim = 64 ngroups = 1