Add nightly performance test for GPT-OSS 4GPU models (#12805)

This commit is contained in:
alisonshao
2025-11-07 16:54:07 -08:00
committed by GitHub
parent d3d7f960b5
commit 0b88d520a0
3 changed files with 136 additions and 1 deletions

View File

@@ -120,7 +120,7 @@ fi
# Show current packages
$PIP_CMD list
$PIP_CMD install mooncake-transfer-engine==0.3.7.post2 "${NVRTC_SPEC}" py-spy scipy huggingface_hub[hf_xet] $PIP_INSTALL_SUFFIX
$PIP_CMD install mooncake-transfer-engine==0.3.7.post2 "${NVRTC_SPEC}" py-spy scipy huggingface_hub[hf_xet] pytest $PIP_INSTALL_SUFFIX
if [ "$IS_BLACKWELL" != "1" ]; then
# For lmms_evals evaluating MMMU