From 4a467495a8cdf04ff68c01bd3e39c9c53babd2e3 Mon Sep 17 00:00:00 2001
From: tianwei <liutianweidlut@gmail.com>
Date: Fri, 5 Aug 2022 22:13:09 +0800
Subject: [PATCH] enhance(ci): parallel build starwhale image (#874)

---
 .github/workflows/release.yml | 11 ++++++++++-
 docker/Makefile               | 20 ++++++--------------
 2 files changed, 16 insertions(+), 15 deletions(-)

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index d12088a0c5..4c94393ec4 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -63,6 +63,11 @@ jobs:
     needs:
       - pypi-release
 
+    strategy:
+      matrix:
+        cuda_version: ["", "11.3", "11.3-cudnn8", "11.4", "11.4-cudnn8", "11.5", "11.5-cudnn8", "11.6", "11.6-cudnn8", "11.7"]
+        arch: ["linux/arm64", "linux/amd64"]
+
     steps:
       - uses: actions/checkout@v3
 
@@ -97,7 +102,11 @@ jobs:
           RELEASE_VERSION: ${{ steps.tag.outputs.tag }}
           PYPI_RELEASE_VERSION: ${{ steps.tag.outputs.tag }}
         run: |
-          make build-release-starwhale
+          if [ "${{matrix.cuda_version}}" = "" ]; then
+            make build-release-starwhale arch=${{matrix.arch}}
+          else
+            make build-release-starwhale-cuda arch=${{matrix.arch}} version=${{matrix.cuda_version}}
+          fi
 
   server-image-release:
     runs-on: ubuntu-latest
diff --git a/docker/Makefile b/docker/Makefile
index abd3f52d2c..b6b5621305 100644
--- a/docker/Makefile
+++ b/docker/Makefile
@@ -50,7 +50,7 @@ define push-image
 endef
 
 define build-starwhale
-	docker buildx build --platform linux/arm64,linux/amd64 \
+	docker buildx build --platform $(3) \
 		--build-arg BASE_IMAGE=$(1) \
 		--build-arg SW_VERSION=$(PYPI_RELEASE_VERSION) \
 		--build-arg HTTP_PROXY=$(HTTP_PROXY) --build-arg HTTPS_PROXY=$(HTTPS_PROXY) \
@@ -66,7 +66,7 @@ define build-starwhale
 endef
 
 define build-starwhale-cuda
-	$(call build-starwhale,${GHCR_IO_REPO}/cuda:$(1)-base${FIXED_VERSION_BASE_IMAGE},-cuda$(1))
+	$(call build-starwhale,${GHCR_IO_REPO}/cuda:$(1)-base${FIXED_VERSION_BASE_IMAGE},-cuda$(1),$(2))
 endef
 
 
@@ -88,19 +88,11 @@ build-release-base:
 		-t $(GHCR_IO_REPO)/base:latest \
 		--push -f Dockerfile.base .
 
-#Parallel to build starwhale image
 build-release-starwhale:
-	$(call build-starwhale,${GHCR_IO_REPO}/base:${FIXED_VERSION_BASE_IMAGE},)
-
-	$(call build-starwhale-cuda,11.3)
-	$(call build-starwhale-cuda,11.3-cudnn8)
-	$(call build-starwhale-cuda,11.4)
-	$(call build-starwhale-cuda,11.4-cudnn8)
-	$(call build-starwhale-cuda,11.5)
-	$(call build-starwhale-cuda,11.5-cudnn8)
-	$(call build-starwhale-cuda,11.6)
-	$(call build-starwhale-cuda,11.6-cudnn8)
-	$(call build-starwhale-cuda,11.7)
+	$(call build-starwhale,${GHCR_IO_REPO}/base:${FIXED_VERSION_BASE_IMAGE},,$(arch))
+
+build-release-starwhale-cuda:
+	$(call build-starwhale-cuda,$(version),$(arch))
 
 build-cuda:
 	cd cuda; python3 render.py --all --push --base-image $(GHCR_BASE_IMAGE)