allenchen121
diff --git a/‎.github/workflows/build-wheels-cuda.yaml
Copy file name to clipboardExpand all lines: .github/workflows/build-wheels-cuda.yaml
+13-24Lines changed: 13 additions & 24 deletions b/‎.github/workflows/build-wheels-cuda.yaml
Copy file name to clipboardExpand all lines: .github/workflows/build-wheels-cuda.yaml
+13-24Lines changed: 13 additions & 24 deletions
@@ -8,7 +8,7 @@ permissions:
 jobs:
   define_matrix:
     name: Define Build Matrix
-    runs-on: windows-latest
+    runs-on: ubuntu-latest
     outputs:
       matrix: ${{ steps.set-matrix.outputs.matrix }}
     defaults:
@@ -20,9 +20,9 @@ jobs:
         id: set-matrix
         run: |
           $matrix = @{
-              'os' = @('windows-latest')
+              'os' = @('ubuntu-latest', 'windows-2019')
               'pyver' = @("3.12")
-              'cuda' = @("12.5")
+              'cuda' = @("12.5.1")
               'releasetag' = @("basic")
           }
 
@@ -43,33 +43,12 @@ jobs:
       AVXVER: ${{ matrix.releasetag }}
 
     steps:
-      - name: Debug Before Adding MSBuild
-        run: |
-          Write-Output "Current PATH:"
-          Write-Output $env:PATH
-
       - name: Add MSBuild to PATH
         if: runner.os == 'Windows'
         uses: microsoft/setup-msbuild@v2
         with:
           vs-version: '[16.11,16.12)'
 
-      - name: Debug After Adding MSBuild
-        run: |
-          Write-Output "Updated PATH:"
-          Write-Output $env:PATH
-          Get-ChildItem "C:\Program Files (x86)\Microsoft Visual Studio\2019\Enterprise\MSBuild\Current\Bin"
-
-      - name: Explicitly Add MSBuild to PATH
-        if: runner.os == 'Windows'
-        run: |
-          $msbuildPath = "C:\Program Files (x86)\Microsoft Visual Studio\2019\Enterprise\MSBuild\Current\Bin"
-          if (-Not (Test-Path $msbuildPath)) {
-            throw "MSBuild path not found: $msbuildPath"
-          }
-          $env:PATH += ";$msbuildPath"
-          Write-Output "##[add-path]$env:PATH"
-
       - uses: actions/checkout@v4
         with:
           submodules: "recursive"
@@ -135,13 +114,23 @@ jobs:
           $env:VERBOSE = '1'
           $env:CMAKE_ARGS = '-DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=all'
           $env:CMAKE_ARGS = "-DGGML_CUDA_FORCE_MMQ=ON $env:CMAKE_ARGS"
+          # if ($env:AVXVER -eq 'AVX') {
           $env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_AVX2=off -DGGML_FMA=off -DGGML_F16C=off'
+          # }
+          # if ($env:AVXVER -eq 'AVX512') {
+          #  $env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_AVX512=on'
+          # }
+          # if ($env:AVXVER -eq 'basic') {
+          #  $env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_AVX=off -DGGML_AVX2=off -DGGML_FMA=off -DGGML_F16C=off'
+          # }
           python -m build --wheel
+          # write the build tag to the output
           Write-Output "CUDA_VERSION=$cudaVersion" >> $env:GITHUB_ENV
 
       - uses: softprops/action-gh-release@v2
         with:
           files: dist/*
+          # Set tag_name to <tag>-cu<cuda_version>
           tag_name: ${{ github.ref_name }}-cu${{ env.CUDA_VERSION }}
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}