mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-02-16 15:55:08 +08:00
[TRTLLM-10331][infra] Upload unittest sub results in slurm (#10834)
Signed-off-by: Yiqing Yan <yiqingy@nvidia.com> Co-authored-by: Yanchao Lu <yanchaol@nvidia.com>
This commit is contained in:
parent
2a4e70b4a9
commit
21cdc39e83
@ -128,7 +128,7 @@ def uploadResults(def pipeline, SlurmCluster cluster, String nodeName, String st
|
||||
def timeoutTestFilePath = "/home/svc_tensorrt/bloom/scripts/${nodeName}/unfinished_test.txt"
|
||||
def downloadTimeoutTestSucceed = Utils.exec(pipeline, script: "sshpass -p '${remote.passwd}' scp -P ${remote.port} -r -p ${COMMON_SSH_OPTIONS} ${remote.user}@${remote.host}:${timeoutTestFilePath} ${stageName}/", returnStatus: true, numRetries: 3) == 0
|
||||
if (downloadTimeoutTestSucceed) {
|
||||
sh "ls ${stageName}"
|
||||
sh "ls -al ${stageName}/"
|
||||
def timeoutTestXml = generateTimeoutTestResultXml(stageName, "unfinished_test.txt")
|
||||
if (timeoutTestXml != null) {
|
||||
sh """
|
||||
@ -140,7 +140,7 @@ EOF_TIMEOUT_XML
|
||||
}
|
||||
}
|
||||
// Download normal test results
|
||||
def resultsFilePath = "/home/svc_tensorrt/bloom/scripts/${nodeName}/results.xml"
|
||||
def resultsFilePath = "/home/svc_tensorrt/bloom/scripts/${nodeName}/results*.xml"
|
||||
downloadResultSucceed = Utils.exec(pipeline, script: "sshpass -p '${remote.passwd}' scp -P ${remote.port} -r -p ${COMMON_SSH_OPTIONS} ${remote.user}@${remote.host}:${resultsFilePath} ${stageName}/", returnStatus: true, numRetries: 3) == 0
|
||||
|
||||
// Download perf test results
|
||||
@ -165,7 +165,7 @@ EOF_TIMEOUT_XML
|
||||
|
||||
echo "hasTimeoutTest: ${hasTimeoutTest}, downloadResultSucceed: ${downloadResultSucceed}, downloadPerfResultSucceed: ${downloadPerfResultSucceed}"
|
||||
if (hasTimeoutTest || downloadResultSucceed || downloadPerfResultSucceed) {
|
||||
sh "ls ${stageName}"
|
||||
sh "ls -al ${stageName}/"
|
||||
echo "Upload test results."
|
||||
sh "tar -czvf results-${stageName}.tar.gz ${stageName}/"
|
||||
ensureStageResultNotUploaded(stageName)
|
||||
@ -774,7 +774,7 @@ def executeLLMTestOnSlurm(pipeline, platform, testList, config=VANILLA_CONFIG, p
|
||||
sh "cd ${stageName} && sed -i 's/testsuite name=\"pytest\"/testsuite name=\"${stageName}\"/g' *.xml || true"
|
||||
// Copy CPP test result
|
||||
sh "cp ${llmSrc}/cpp/build_backup/*.xml ${stageName} || true"
|
||||
sh "ls ${stageName}/ -all"
|
||||
sh "ls -al ${stageName}/"
|
||||
})
|
||||
}
|
||||
}
|
||||
@ -2948,7 +2948,7 @@ def runLLMTestlistOnPlatform(pipeline, platform, testList, config=VANILLA_CONFIG
|
||||
sh "cd ${stageName} && sed -i 's/testsuite name=\"pytest\"/testsuite name=\"${stageName}\"/g' *.xml || true"
|
||||
// Copy CPP test result
|
||||
sh "cp ${llmSrc}/cpp/build_backup/*.xml ${stageName} || true"
|
||||
sh "ls ${stageName}/ -all"
|
||||
sh "ls -al ${stageName}/"
|
||||
}, false, postTag)
|
||||
}
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user