diff --git a/jenkins/L0_Test.groovy b/jenkins/L0_Test.groovy index 941c3efb228..6f6ae7c1186 100644 --- a/jenkins/L0_Test.groovy +++ b/jenkins/L0_Test.groovy @@ -309,6 +309,7 @@ def runLLMTestlistOnSlurm_MultiNodes(pipeline, platform, testList, config=VANILL def llmSrcLocal = "${llmPath}/TensorRT-LLM/src" def scriptRunNode = "${jobWorkspace}/slurm_run.sh" def testListPathNode = "${jobWorkspace}/${testList}.txt" + def waivesListPathNode = "${jobWorkspace}/waives.txt" def isAarch64 = config.contains("aarch64") def pytestTestTimeout = "7200" @@ -325,6 +326,10 @@ def runLLMTestlistOnSlurm_MultiNodes(pipeline, platform, testList, config=VANILL Utils.exec(pipeline, script: "chmod +x ${scriptRunLocalPath}", returnStdout: true) Utils.exec(pipeline, script: "sshpass -p '${remote.passwd}' scp -r -p -oStrictHostKeyChecking=no ${scriptRunLocalPath} ${remote.user}@${remote.host}:${scriptRunNode}",) + // Upload waives.txt to Frontend node + def waivesListLocalPath = "${llmSrcLocal}/tests/integration/test_lists/waives.txt" + Utils.exec(pipeline, script: "sshpass -p '${remote.passwd}' scp -r -p -oStrictHostKeyChecking=no ${waivesListLocalPath} ${remote.user}@${remote.host}:${waivesListPathNode}",) + // Generate Test List and Upload to Frontend Node def makoArgs = getMakoArgsFromStageName(stageName, true) // TODO: currently the options will only be processed if the first @@ -362,6 +367,7 @@ def runLLMTestlistOnSlurm_MultiNodes(pipeline, platform, testList, config=VANILL export stageName=$stageName export testList=$testList export testListPathNode=$testListPathNode + export waivesListPathNode=$waivesListPathNode export pytestTestTimeout=$pytestTestTimeout export splits=$splits export splitId=$splitId diff --git a/jenkins/scripts/slurm_run.sh b/jenkins/scripts/slurm_run.sh index 9c055d8cd34..4b6337fca5d 100755 --- a/jenkins/scripts/slurm_run.sh +++ b/jenkins/scripts/slurm_run.sh @@ -45,6 +45,7 @@ testCmdLines=( "-v" "--timeout=$pytestTestTimeout" "--test-list=$testListPathNode" + "--waives-file=$waivesListPathNode" "--rootdir $llmSrcNode/tests/integration/defs" "--test-prefix=$stageName" "--splits $splits"