[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:
Yiqing Yan 2026-02-10 17:53:35 +08:00 committed by GitHub
parent 2a4e70b4a9
commit 21cdc39e83
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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)
}