add auto push mode

cupy · Apr 11, 2024 · a0e93fe · a0e93fe
1 parent 489aa0d
commit a0e93fe
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/builder/base/cuda-runfile/build.sh b/builder/base/cuda-runfile/build.sh
@@ -144,4 +144,13 @@ docker buildx build --platform "${PLATFORM}" -t "cupy/cupy-release-tools:cuda-ru
     --build-arg BASE_IMAGE="${BASE_IMAGE}" \
     --build-arg CUDA_INSTALLER_URL="${CUDA_INSTALLER_URL}"
 
-echo "Done. Run docker push cupy/cupy-release-tools:cuda-runfile-${CUDA_VERSION}-${IMAGE_SUFFIX} to push the image."
+PUSH_COMMAND="docker push cupy/cupy-release-tools:cuda-runfile-${CUDA_VERSION}-${IMAGE_SUFFIX}"
+
+echo "Done."
+
+if [[ ${AUTO_PUSH:-0} == 1 ]]; then
+  echo "Pushing: ${PUSH_COMMAND}"
+  eval "${PUSH_COMMAND}"
+else
+  echo "Run ${PUSH_COMMAND} to push the image."
+fi