Skip to content

Commit 630e67b

Browse files
authored
[None][ci] waive test_mamba2_chunk_scan_combined_prefill_chunking[seqlens1-8] (#7194)
Signed-off-by: junq <[email protected]>
1 parent 9c5b464 commit 630e67b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/unittest/_torch/thop/test_mamba2_chunk_ss_update.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -369,6 +369,8 @@ def test_mamba2_chunk_scan_selective_state_update(dim, headdim, ngroups, dstate,
369369
(16, 20),
370370
])
371371
def test_mamba2_chunk_scan_combined_prefill_chunking(mamba_chunk_size, seqlens):
372+
if mamba_chunk_size == 8 and seqlens == (270, 88, 212, 203):
373+
pytest.skip("https://nvbugspro.nvidia.com/bug/5477332")
372374
dim = 1024
373375
headdim = 64
374376
ngroups = 1

0 commit comments

Comments
 (0)