[CI Perf]Prune Tests in kernel/mamba (#26538)

Signed-off-by: Fardin Hoque <kfhfar@amazon.com>
Signed-off-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>
Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>
This commit is contained in:
Fardin Hoque
2025-10-13 15:22:31 -07:00
committed by GitHub
parent 314285d4f2
commit 577c72a227
4 changed files with 21 additions and 36 deletions

View File

@@ -25,7 +25,6 @@ from vllm.utils import update_environment_variables
(64, 1),
(64, 2),
(64, 4), # hidden_size be divisible by num_gpus
(100, 5), # and n_groups must divide hidden_size
],
)
@pytest.mark.parametrize("dtype", [torch.float16])