fix: adding matrix partitioning for h200 and b200 nightly tests (#17091)

This commit is contained in:
Douglas Yang
2026-01-15 11:23:19 -08:00
committed by GitHub
parent 3f44268fe5
commit 655d2c7c2a
11 changed files with 21 additions and 13 deletions
@@ -9,7 +9,7 @@ from sglang.test.test_utils import ModelLaunchSettings, is_blackwell_system
# Runs on both H200 and B200 via nightly-8-gpu-common suite
# Note: trtllm_mla backend may have hardware-specific behavior
register_cuda_ci(est_time=12000, suite="nightly-8-gpu-common", nightly=True)
register_cuda_ci(est_time=1800, suite="nightly-8-gpu-common", nightly=True)
MISTRAL_LARGE3_MODEL_PATH = "mistralai/Mistral-Large-3-675B-Instruct-2512"
MISTRAL_LARGE3_EAGLE_MODEL_PATH = "mistralai/Mistral-Large-3-675B-Instruct-2512-Eagle"