Skip to content

Commit 1cc4949

Browse files
authored
[Infra] - Add wiave list for pytest when using slurm (#6130)
Signed-off-by: qqiao <[email protected]>
1 parent 8c1c9ef commit 1cc4949

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

jenkins/L0_Test.groovy

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -309,6 +309,7 @@ def runLLMTestlistOnSlurm_MultiNodes(pipeline, platform, testList, config=VANILL
309309
def llmSrcLocal = "${llmPath}/TensorRT-LLM/src"
310310
def scriptRunNode = "${jobWorkspace}/slurm_run.sh"
311311
def testListPathNode = "${jobWorkspace}/${testList}.txt"
312+
def waivesListPathNode = "${jobWorkspace}/waives.txt"
312313
def isAarch64 = config.contains("aarch64")
313314
def pytestTestTimeout = "7200"
314315

@@ -325,6 +326,10 @@ def runLLMTestlistOnSlurm_MultiNodes(pipeline, platform, testList, config=VANILL
325326
Utils.exec(pipeline, script: "chmod +x ${scriptRunLocalPath}", returnStdout: true)
326327
Utils.exec(pipeline, script: "sshpass -p '${remote.passwd}' scp -r -p -oStrictHostKeyChecking=no ${scriptRunLocalPath} ${remote.user}@${remote.host}:${scriptRunNode}",)
327328

329+
// Upload waives.txt to Frontend node
330+
def waivesListLocalPath = "${llmSrcLocal}/tests/integration/test_lists/waives.txt"
331+
Utils.exec(pipeline, script: "sshpass -p '${remote.passwd}' scp -r -p -oStrictHostKeyChecking=no ${waivesListLocalPath} ${remote.user}@${remote.host}:${waivesListPathNode}",)
332+
328333
// Generate Test List and Upload to Frontend Node
329334
def makoArgs = getMakoArgsFromStageName(stageName, true)
330335
// TODO: currently the options will only be processed if the first
@@ -362,6 +367,7 @@ def runLLMTestlistOnSlurm_MultiNodes(pipeline, platform, testList, config=VANILL
362367
export stageName=$stageName
363368
export testList=$testList
364369
export testListPathNode=$testListPathNode
370+
export waivesListPathNode=$waivesListPathNode
365371
export pytestTestTimeout=$pytestTestTimeout
366372
export splits=$splits
367373
export splitId=$splitId

jenkins/scripts/slurm_run.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ testCmdLines=(
4545
"-v"
4646
"--timeout=$pytestTestTimeout"
4747
"--test-list=$testListPathNode"
48+
"--waives-file=$waivesListPathNode"
4849
"--rootdir $llmSrcNode/tests/integration/defs"
4950
"--test-prefix=$stageName"
5051
"--splits $splits"

0 commit comments

Comments
 (0)