|
23 | 23 | $matrix = @{ |
24 | 24 | 'os' = @('windows-2022') #'ubuntu-latest', |
25 | 25 | 'pyver' = @("3.11") # "3.10", , "3.12" |
26 | | - 'cuda' = @("12.4.1", "12.5.1", "12.6.3") # "12.1.1", "12.2.2", "12.3.2", "12.4.1") #, "12.5.1", |
27 | | - 'releasetag' = @("AVX", "basic") # 'releasetag' = @("basic") |
| 26 | + 'cuda' = @("12.6.3") # "12.1.1", "12.2.2", "12.3.2", "12.4.1") #, "12.5.1", |
| 27 | + 'releasetag' = @("AVX2") # 'releasetag' = @("basic") |
28 | 28 | 'cudaarch' = @("all") |
29 | 29 | } |
30 | 30 |
|
|
53 | 53 | if: runner.os == 'Windows' |
54 | 54 | uses: microsoft/setup-msbuild@main |
55 | 55 | with: |
56 | | -vs-version: '[17.12,17.15)' |
| 56 | +vs-version: '[16.11,16.12)' |
57 | 57 | msbuild-architecture: x64 |
58 | 58 |
|
59 | 59 | - uses: actions/checkout@main |
|
63 | 63 |
|
64 | 64 | # from kingbri1/flash-attention build-wheels.yml |
65 | 65 | - name: Install CUDA ${{matrix.cuda }} |
66 | | - |
| 66 | + |
67 | 67 | # 250630 temporary issue https://github.com/Jimver/cuda-toolkit/issues/395#issuecomment-2941579978 |
68 | 68 | id: cuda-toolkit |
69 | 69 | with: |
@@ -104,9 +104,6 @@ jobs: |
104 | 104 | # if ($env:AVXVER -eq 'AVX512'){ |
105 | 105 | # $env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_AVX512=on' |
106 | 106 | # } |
107 | | - if ($env:AVXVER -eq 'basic'){ |
108 | | - $env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_AVX=off -DGGML_AVX2=off -DGGML_FMA=off -DGGML_F16C=off' |
109 | | - } |
110 | 107 | python -m build --wheel |
111 | 108 | # write the build tag to the output |
112 | 109 | Write-Output "CUDA_VERSION=$cudaVersion" >> $env:GITHUB_ENV |
|
0 commit comments