Skip to content

Commit b91da84

Browse files
authored
Update build-wheels-cuda.yaml
1 parent d769bc6 commit b91da84

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

‎.github/workflows/build-wheels-cuda.yaml‎

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ jobs:
2323
$matrix = @{
2424
'os' = @('windows-2022') #'ubuntu-latest',
2525
'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")
2828
'cudaarch' = @("all")
2929
}
3030
@@ -53,7 +53,7 @@ jobs:
5353
if: runner.os == 'Windows'
5454
uses: microsoft/setup-msbuild@main
5555
with:
56-
vs-version: '[17.12,17.15)'
56+
vs-version: '[16.11,16.12)'
5757
msbuild-architecture: x64
5858

5959
- uses: actions/checkout@main
@@ -63,7 +63,7 @@ jobs:
6363

6464
# from kingbri1/flash-attention build-wheels.yml
6565
- name: Install CUDA ${{matrix.cuda }}
66-
uses: Jimver/[email protected].24
66+
uses: Jimver/[email protected].21
6767
# 250630 temporary issue https://github.com/Jimver/cuda-toolkit/issues/395#issuecomment-2941579978
6868
id: cuda-toolkit
6969
with:
@@ -104,9 +104,6 @@ jobs:
104104
# if ($env:AVXVER -eq 'AVX512'){
105105
# $env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_AVX512=on'
106106
# }
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-
}
110107
python -m build --wheel
111108
# write the build tag to the output
112109
Write-Output "CUDA_VERSION=$cudaVersion" >> $env:GITHUB_ENV

0 commit comments

Comments
(0)