File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed
.github/actions/docker-build Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -115,7 +115,11 @@ runs:
115115 EXTRA_ARGS="--cache-to type=inline "
116116 EXTRA_ARGS+="--cache-from type=registry,ref=${ECR_HOSTNAME}/ai-dynamo/dynamo:${{ inputs.framework }}-cuda${CUDA_VERSION_MAJOR}-${PLATFORM##*/}-cache "
117117 EXTRA_ARGS+="--cache-from type=registry,ref=${ECR_HOSTNAME}/ai-dynamo/dynamo:main-${{ inputs.framework }}-cuda${CUDA_VERSION_MAJOR}-${PLATFORM##*/} "
118- if [[ "$GITHUB_REF_NAME" == "main" ]]; then
118+ EXTRA_ARGS+="--cache-from type=registry,ref=${ECR_HOSTNAME}/ai-dynamo/dynamo:release-${{ inputs.framework }}-cuda${CUDA_VERSION_MAJOR}-${PLATFORM##*/}-cache "
119+ if [[ "$GITHUB_REF_NAME" =~ ^release ]]; then
120+ # Release branches also use release cache
121+ EXTRA_ARGS+="--cache-to type=registry,ref=${ECR_HOSTNAME}/ai-dynamo/dynamo:release-${{ inputs.framework }}-cuda${CUDA_VERSION_MAJOR}-${PLATFORM##*/}-cache,mode=max "
122+ elif [[ "$GITHUB_REF_NAME" == "main" ]]; then
119123 EXTRA_ARGS+="--cache-to type=registry,ref=${ECR_HOSTNAME}/ai-dynamo/dynamo:${{ inputs.framework }}-cuda${CUDA_VERSION_MAJOR}-${PLATFORM##*/}-cache,mode=max "
120124 fi
121125 fi
You can’t perform that action at this time.
0 commit comments