kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[https://nvbugs/5433581][infra] Temporarily disable Docker Image use wheel from build stage (#6630)

Browse Source 
Signed-off-by: ZhanruiSunCh <184402041+ZhanruiSunCh@users.noreply.github.com>
This commit is contained in:
Zhanrui Sun 2025-08-05 21:33:11 +08:00 committed by GitHub
parent 78a75c2990
commit 6a9b4b11be
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
jenkins/BuildDockerImage.groovy
View File

@ -29,6 +29,8 @@ LLM_DEFAULT_TAG = env.defaultTag ?: "${LLM_SHORT_COMMIT}-${LLM_BRANCH_TAG}-${BUI
RUN_SANITY_CHECK = params.runSanityCheck ?: false
TRIGGER_TYPE = env.triggerType ?: "manual"
ENABLE_USE_WHEEL_FROM_BUILD_STAGE = params.useWheelFromBuildStage ?: false
WAIT_TIME_FOR_BUILD_STAGE = 60 // minutes
BUILD_JOBS = "32"
@ -193,6 +195,11 @@ def createKubernetesPodConfig(type, arch = "amd64", build_wheel = false)
def prepareWheelFromBuildStage(dockerfileStage, arch) {
if (!ENABLE_USE_WHEEL_FROM_BUILD_STAGE) {
echo "useWheelFromBuildStage is false, skip preparing wheel from build stage"
return ""
}
if (TRIGGER_TYPE != "post-merge") {
echo "Trigger type is not post-merge, skip preparing wheel from build stage"
return ""