Explorar el Código

build(workflow): update CUDA toolkit installation strategy for build workflow

沉默の金 hace 5 meses
padre
commit
7e908ffbff
Se han modificado 1 ficheros con 11 adiciones y 8 borrados
  1. 11 8
      .github/workflows/build.yml

+ 11 - 8
.github/workflows/build.yml

@@ -468,13 +468,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        include:
-          - cuda_version: "11.8.0"
-            sub_packages: '["nvcc", "cudart", "cublas", "cublas_dev", "nvrtc", "nvrtc_dev", "cccl", "visual_studio_integration"]'
-          - cuda_version: "12.9.1"
-            sub_packages: '["nvcc", "cudart", "cublas", "cublas_dev", "nvrtc", "nvrtc_dev", "cccl", "visual_studio_integration"]'
-          - cuda_version: "13.1.0"
-            sub_packages: '["nvcc", "cudart", "cublas", "cublas_dev", "nvrtc", "nvrtc_dev", "cccl", "crt", "nvvm", "visual_studio_integration"]'
+        cuda_version: ["11.8.0", "12.9.1", "13.1.0"]
     
     env:
       CUDA_VERSION: ${{ matrix.cuda_version }}
@@ -491,12 +485,21 @@ jobs:
         run: git clone --depth 1 https://github.com/ggerganov/ggml.git ggml
       
       - name: Install CUDA Toolkit
+        if: ${{ matrix.cuda_version != '13.1.0' }}
         uses: Jimver/cuda-toolkit@master
         with:
           cuda: ${{ matrix.cuda_version }}
           method: network
-          sub-packages: ${{ matrix.sub_packages }}
+          sub-packages: '["nvcc", "cudart", "cublas", "cublas_dev", "thrust", "visual_studio_integration"]'
         
+      - name: Install CUDA Toolkit(13.1.0)
+        if: ${{ matrix.cuda_version == '13.1.0' }}
+        uses: Jimver/cuda-toolkit@master
+        with:
+          cuda: ${{ matrix.cuda_version }}
+          method: network
+          sub-packages: '["nvcc", "cudart", "cublas", "cublas_dev", "nvrtc", "nvrtc_dev", "cccl", "crt", "nvvm", "visual_studio_integration"]'
+
       - name: Install Ninja
         run: choco install ninja -y