minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Fix mamba2 mixer ci test (#19658)

Browse Source
This commit is contained in:
Ke Bao
2026-03-02 14:55:15 +08:00
committed by GitHub
parent 922aad2faa
commit 4726073865
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/layers/mamba/test_mamba2_mixer.py
+3 -1
View File
@@ -40,7 +40,9 @@ def test_mixer2_gated_norm_multi_gpu(
if not torch.cuda.is_available(): if not torch.cuda.is_available():
pytest.skip("CUDA device not available") pytest.skip("CUDA device not available")
assert torch.cuda.device_count() == NUM_GPUS assert (
torch.cuda.device_count() >= NUM_GPUS
), f"This test requires at least {NUM_GPUS} GPUs, but only {torch.cuda.device_count()} available"
hidden_size, n_groups = hidden_size_n_groups hidden_size, n_groups = hidden_size_n_groups
num_processes = NUM_GPUS num_processes = NUM_GPUS