Skip to content

Commit 5acaa80

Browse files
authored
Refine CUDA versions and CMake arguments in workflow
Updated the CUDA versions and modified CMake arguments for building wheels. Disable AVX512
1 parent 346853c commit 5acaa80

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

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

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,11 +26,11 @@ jobs:
2626
id: set-matrix
2727
run: |
2828
$matrix = @{
29-
'os' = @('ubuntu-22.04', 'windows-2022')
29+
'os' = @('ubuntu-22.04')
3030
# wheel.py-api = "py3" makes the CUDA wheel interpreter-agnostic,
3131
# so one builder per toolkit version is sufficient.
3232
'pyver' = @("3.9")
33-
'cuda' = @("11.8.0", "12.1.1", "12.2.2", "12.3.2", "12.4.1", "12.5.1", "13.0.2", "13.2.1")
33+
'cuda' = @("12.4.1")
3434
'releasetag' = @("basic")
3535
'exclude' = @(
3636
@{ 'os' = 'windows-2022'; 'cuda' = '12.1.1' },
@@ -231,7 +231,9 @@ jobs:
231231
# one forward-compatible PTX target instead of embedding PTX for every
232232
# SM. This keeps the wheel under GitHub's 2 GiB release-asset limit.
233233
$env:CMAKE_ARGS = "-DGGML_CUDA_FORCE_MMQ=ON -DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=$cudaArchs -DCMAKE_CUDA_FLAGS=-allow-unsupported-compiler -DCMAKE_CUDA_FLAGS_INIT=-allow-unsupported-compiler $env:CMAKE_ARGS"
234-
$env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_AVX2=off -DGGML_FMA=off -DGGML_F16C=off'
234+
$env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_NATIVE=OFF -DGGML_AVX512=OFF'
235+
$env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_AVX2=ON -DGGML_FMA=ON -DGGML_F16C=ON'
236+
235237
if ($IsLinux) {
236238
$env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_OPENMP=OFF'
237239
}

0 commit comments

Comments
 (0)