make cuda-go-rust-builder multi platform

scroll-tech · Apr 22, 2024 · 340e32d · 340e32d
1 parent 69c0f7e
commit 340e32d
Showing 1 changed file with 13 additions and 16 deletions.
diff --git a/.github/workflows/intermediate-docker.yml b/.github/workflows/intermediate-docker.yml
@@ -41,19 +41,16 @@ jobs:
       with:
         username: ${{ secrets.DOCKERHUB_USERNAME }}
         password: ${{ secrets.DOCKERHUB_TOKEN }}
-    - name: Build
-      run: |
-        make all
-      env:
-        GO_VERSION: ${{ inputs.GO_VERSION }}
-        RUST_VERSION: ${{ inputs.RUST_VERSION }}
-        PYTHON_VERSION: ${{ inputs.PYTHON_VERSION }}
-        CUDA_VERSION: ${{ inputs.CUDA_VERSION }}
-    - name: Publish
-      run: |
-        make publish
-      env:
-        GO_VERSION: ${{ inputs.GO_VERSION }}
-        RUST_VERSION: ${{ inputs.RUST_VERSION }}
-        PYTHON_VERSION: ${{ inputs.PYTHON_VERSION }}
-        CUDA_VERSION: ${{ inputs.CUDA_VERSION }}
+    #cuda-go-rust-builder
+    - name: Build image
+      id: build
+      uses: docker/build-push-action@v3
+      with:
+        push: true
+        file: cuda-go-rust-builder.Dockerfile
+        platforms: linux/amd64,linux/arm64
+        tags: scrolltech/cuda-go-rust-builder:${{ github.event.inputs.CUDA_VERSION }}
+        build-args: |
+          CUDA_VERSION: ${{ github.event.inputs.CUDA_VERSION }}
+          GO_VERSION: ${{ github.event.inputs.GO_VERSION }}
+          RUST_VERSION: ${{ github.event.inputs.RUST_VERSION }}