File tree Expand file tree Collapse file tree 1 file changed +9
-5
lines changed
Expand file tree Collapse file tree 1 file changed +9
-5
lines changed Original file line number Diff line number Diff line change @@ -56,10 +56,6 @@ NGC_STAGING_REPO ?= nvcr.io/nvstaging/tensorrt-llm
5656NGC_REPO ?= nvcr.io/nvidia/tensorrt-llm
5757NGC_USE_STAGING ?= 0
5858NGC_AUTO_REPO ?= $(if $(filter 1,$(NGC_USE_STAGING ) ) ,$(NGC_STAGING_REPO ) ,$(NGC_REPO ) )
59- SH_ENV ?= $(shell docker inspect --format='{{range .Config.Env}}{{println .}}{{end}}' $(BASE_IMAGE ) :$(BASE_TAG ) \
60- | grep '^ENV=' | sed 's/^[^=]* =//' 2>/dev/null)
61- BASH_ENV ?= $(shell docker inspect --format='{{range .Config.Env}}{{println .}}{{end}}' $(BASE_IMAGE ) :$(BASE_TAG ) \
62- | grep '^BASH_ENV=' | sed 's/^[^=]* =//' 2>/dev/null)
6359
6460define add_local_user
6561 docker build \
@@ -79,8 +75,16 @@ define rewrite_tag
7975$(shell echo $(IMAGE_WITH_TAG ) | sed "s/\/tensorrt-llm:/\/tensorrt-llm-staging:/g")
8076endef
8177
78+ base_pull :
79+ @echo " Pulling base image: $( BASE_IMAGE) :$( BASE_TAG) "
80+ docker pull $(BASE_IMAGE ) :$(BASE_TAG )
81+
8282% _build : DEVEL_IMAGE = $(if $(findstring 1,$(JENKINS_DEVEL ) ) ,$(shell . ../jenkins/current_image_tags.properties && echo $$LLM_DOCKER_IMAGE) )
83- % _build :
83+ % _build : SH_ENV = $(shell docker inspect --format='{{range .Config.Env}}{{println .}}{{end}}' $(BASE_IMAGE ) :$(BASE_TAG ) \
84+ | grep '^ENV=' | sed 's/^[^=]* =//' 2>/dev/null)
85+ % _build : BASH_ENV = $(shell docker inspect --format='{{range .Config.Env}}{{println .}}{{end}}' $(BASE_IMAGE ) :$(BASE_TAG ) \
86+ | grep '^BASH_ENV=' | sed 's/^[^=]* =//' 2>/dev/null)
87+ % _build : base_pull
8488 @echo " Building docker image: $( IMAGE_WITH_TAG) "
8589 docker buildx build $(DOCKER_BUILD_OPTS ) $(DOCKER_BUILD_ARGS ) \
8690 --progress $(DOCKER_PROGRESS ) \
You can’t perform that action at this time.
0 commit comments