[TRTLLM-6893][infra] fix Build Docker Image tag issue (#6555)

Signed-off-by: ZhanruiSunCh <184402041+ZhanruiSunCh@users.noreply.github.com>
Signed-off-by: Zhanrui Sun <184402041+ZhanruiSunCh@users.noreply.github.com>
This commit is contained in:
Zhanrui Sun 2025-08-05 16:33:36 +08:00 committed by GitHub
parent dc84695520
commit 7cbe30e17d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -233,15 +233,11 @@ def buildImage(config, imageKeyToTag)
def dependentImageWithTag = "${IMAGE_NAME}/${dependent.dockerfileStage}:${dependentTag}"
def customImageWithTag = "${IMAGE_NAME}/${dockerfileStage}:${customTag}"
if (target == "ngc-release") {
if (TRIGGER_TYPE == "post-merge") {
echo "Use NGC artifacts for post merge build"
dependentImageWithTag = "${NGC_IMAGE_NAME}:${dependentTag}"
imageWithTag = "${NGC_IMAGE_NAME}:${tag}"
customImageWithTag = "${NGC_IMAGE_NAME}:${customTag}"
}
imageKeyToTag["NGC Devel Image ${config.arch}"] = dependentImageWithTag
imageKeyToTag["NGC Release Image ${config.arch}"] = imageWithTag
if (target == "ngc-release" && TRIGGER_TYPE == "post-merge") {
echo "Use NGC artifacts for post merge build"
dependentImageWithTag = "${NGC_IMAGE_NAME}:${dependentTag}"
imageWithTag = "${NGC_IMAGE_NAME}:${tag}"
customImageWithTag = "${NGC_IMAGE_NAME}:${customTag}"
}
args += " GITHUB_MIRROR=https://urm.nvidia.com/artifactory/github-go-remote"
@ -630,7 +626,9 @@ pipeline {
status = handle.result
if (status != "SUCCESS") {
error "Downstream job did not succeed"
catchError(buildResult: 'FAILURE', stageResult: 'FAILURE') {
error "Downstream job did not succeed"
}
}
}
}