test: change cmake cuda architecture.

NexaAI · Feb 24, 2025 · ab6c764 · ab6c764
1 parent 6ab809c
commit ab6c764
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/build-wheels-cuda-linux.yaml b/.github/workflows/build-wheels-cuda-linux.yaml
@@ -95,7 +95,7 @@ jobs:
           $env:VERBOSE = '1'
 
           $env:CMAKE_BUILD_PARALLEL_LEVEL = $(nproc)
-          $env:CMAKE_ARGS = '-DSD_CUDA=ON -DGGML_CUDA=ON -DCMAKE_CUDA_ARCHITECTURES=all'
+          $env:CMAKE_ARGS = '-DSD_CUDA=ON -DGGML_CUDA=ON -DCMAKE_CUDA_ARCHITECTURES=native'
           $env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_CUDA_FORCE_MMQ=ON -DGGML_AVX2=OFF -DGGML_FMA=OFF -DGGML_F16C=OFF'
 
           python -m build --wheel