Skip to content

[TRTLLM-7288][infra] Download merged waive list in slurm script #8999

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
chzblych merged 2 commits into from
Nov 27, 2025
Merged

[TRTLLM-7288][infra] Download merged waive list in slurm script #8999

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
41180e8
Download merged waive list in slurm script
yiqingy0 Nov 7, 2025
9668e0f
Update L0_Test.groovy
chzblych Nov 27, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions jenkins/L0_Test.groovy
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -826,6 +826,7 @@ def runLLMTestlistWithSbatch(pipeline, platform, testList, config=VANILLA_CONFIG
Utils.exec(pipeline, script: "apt-get update && apt-get install -y sshpass openssh-client")
def tarName = BUILD_CONFIGS[config][TARNAME]
def llmTarfile = "https://urm.nvidia.com/artifactory/${ARTIFACT_PATH}/${tarName}"
def llmWaivesTxtfile = "https://urm.nvidia.com/artifactory/${ARTIFACT_PATH}/waive_list/waives.txt"
def llmPath = sh (script: "realpath .", returnStdout: true).trim()
def jobWorkspace = "/home/svc_tensorrt/bloom/scripts/${jobUID}"
def resourcePathNode = "/tmp"
Expand All @@ -834,13 +835,11 @@ def runLLMTestlistWithSbatch(pipeline, platform, testList, config=VANILLA_CONFIG
def scriptRunLocalPath = "${llmSrcLocal}/jenkins/scripts/slurm_run.sh"
def scriptRunPathNode = "${jobWorkspace}/${jobUID}-slurm_run.sh"
def testListPathNode = "${jobWorkspace}/${testList}.txt"
def waivesListPathNode = "${jobWorkspace}/waives.txt"
def outputPath = "${jobWorkspace}/job-output.log"
def scriptLaunchPathLocal = Utils.createTempLocation(pipeline, "./slurm_launch.sh")
def scriptLaunchPathNode = "${jobWorkspace}/${jobUID}-slurm_launch.sh"
def scriptExecPathLocal = Utils.createTempLocation(pipeline, "./slurm_exec.sh")
def scriptExecPathNode = "${jobWorkspace}/${jobUID}-slurm_exec.sh"
def isAarch64 = config.contains("aarch64")
def coverageConfigFile = "${jobWorkspace}/.coveragerc"

stage("[${stageName}] Initializing Test") {
Expand Down Expand Up @@ -985,6 +984,7 @@ def runLLMTestlistWithSbatch(pipeline, platform, testList, config=VANILLA_CONFIG
export jobWorkspace=$jobWorkspace
export tarName=$tarName
export llmTarfile=$llmTarfile
export llmWaivesTxtfile=$llmWaivesTxtfile
export llmSrcNode=$llmSrcNode
export stageName=$stageName
export perfMode=$perfMode
Expand Down
12 changes: 12 additions & 0 deletions jenkins/scripts/slurm_run.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,18 @@ fi
if [ $SLURM_LOCALID -eq 0 ]; then
wget -nv $llmTarfile
tar -zxf $tarName

# Download the new merged waives.txt
if ! wget -nv "$llmWaivesTxtfile"; then
echo "Failed to download merged waives.txt, use the default waives.txt."
elif [ ! -f "waives.txt" ]; then
echo "Downloaded but file does not exist, use the default waives.txt."
else
rm -f "$llmSrcNode/tests/integration/test_lists/waives.txt"
mv waives.txt "$llmSrcNode/tests/integration/test_lists/waives.txt"
echo "Download merged waives.txt successfully"
fi

which python3
python3 --version
apt-get install -y libffi-dev
Expand Down
Loading