Skip to content

Commit 04a3638

Browse files
authored
fix(ci): pass CUDA compiler arg for Windows detection (#2202)
1 parent 2bfd80c commit 04a3638

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ jobs:
156156
if ($IsWindows) {
157157
$nvccPathCmake = $nvccPath.Replace('\', '/')
158158
$env:CUDACXX = $nvccPathCmake
159-
$env:CMAKE_ARGS = "-DCMAKE_CUDA_COMPILER=$nvccPathCmake -DCMAKE_MAKE_PROGRAM=$env:CMAKE_MAKE_PROGRAM $env:CMAKE_ARGS"
159+
$env:CMAKE_ARGS = "-DCMAKE_CUDA_COMPILER=$nvccPathCmake -DCMAKE_CUDA_COMPILER_ARG1=-allow-unsupported-compiler -DCMAKE_MAKE_PROGRAM=$env:CMAKE_MAKE_PROGRAM $env:CMAKE_ARGS"
160160
}
161161
$nvccVersion = ((& $nvccPath --version) | Select-String 'release ([0-9]+\.[0-9]+)').Matches[0].Groups[1].Value
162162
if (-not $nvccVersion) {
@@ -167,7 +167,7 @@ jobs:
167167
# Build real cubins for the supported GPUs, including sm_70, and keep
168168
# one forward-compatible PTX target instead of embedding PTX for every
169169
# SM. This keeps the wheel under GitHub's 2 GiB release-asset limit.
170-
$env:CMAKE_ARGS = "-DGGML_CUDA_FORCE_MMQ=ON -DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=70-real;75-real;80-real;86-real;89-real;90-real;90-virtual -DCMAKE_CUDA_FLAGS=--allow-unsupported-compiler -DCMAKE_CUDA_FLAGS_INIT=--allow-unsupported-compiler $env:CMAKE_ARGS"
170+
$env:CMAKE_ARGS = "-DGGML_CUDA_FORCE_MMQ=ON -DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=70-real;75-real;80-real;86-real;89-real;90-real;90-virtual -DCMAKE_CUDA_FLAGS=-allow-unsupported-compiler -DCMAKE_CUDA_FLAGS_INIT=-allow-unsupported-compiler $env:CMAKE_ARGS"
171171
$env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_AVX2=off -DGGML_FMA=off -DGGML_F16C=off'
172172
python -m build --wheel
173173
# Publish tags that reflect the actual installed toolkit version.

0 commit comments

Comments
 (0)