Kaynağa Gözat

feat(build): enhance CUDA workflow with version-specific configurations

沉默の金 5 ay önce
ebeveyn
işleme
0d18fed6cd
1 değiştirilmiş dosya ile 13 ekleme ve 4 silme
  1. 13 4
      .github/workflows/build.yml

+ 13 - 4
.github/workflows/build.yml

@@ -468,7 +468,13 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        cuda_version: ["11.8.0", "12.9.1", "13.1.0"]
+        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"]'
     
     env:
       CUDA_VERSION: ${{ matrix.cuda_version }}
@@ -489,16 +495,19 @@ jobs:
         with:
           cuda: ${{ matrix.cuda_version }}
           method: network
-          sub-packages: '["nvcc", "cudart", "cublas", "cublas_dev", "visual_studio_integration"]'
+          sub-packages: ${{ matrix.sub_packages }}
         
       - name: Install Ninja
         run: choco install ninja -y
         
       - name: Configure and Build
         shell: cmd
-        env:
-          CUDAFLAGS: -allow-unsupported-compiler
         run: |
+          if "${{ matrix.cuda_version }}" == "11.8.0" (
+            set CUDAFLAGS=-allow-unsupported-compiler -D_ALLOW_COMPILER_AND_STL_VERSION_MISMATCH -D_DISABLE_CONSTEXPR_MUTEX_CONSTRUCTOR
+          ) else (
+            set CUDAFLAGS=
+          )
           cmake -B build -G "Ninja Multi-Config" ^
             -DGGML_DIR=ggml ^
             -DGGML_CUDA=ON ^