diff --git a/.github/workflows/backends.yml b/.github/workflows/backends.yml index ac25b79..f8fe875 100644 --- a/.github/workflows/backends.yml +++ b/.github/workflows/backends.yml @@ -31,7 +31,7 @@ jobs: - build: 'avx512' defines: '-DGGML_AVX512=ON -DSD_BUILD_SHARED_LIBS=ON' - build: 'cuda12' - defines: '-DSD_CUDA=ON -DSD_BUILD_SHARED_LIBS=ON' + defines: '-DSD_CUDA=ON -DCMAKE_CUDA_ARCHITECTURES=61,75,86,89,100 -DSD_BUILD_SHARED_LIBS=ON' - build: 'vulkan' defines: "-DSD_VULKAN=ON -DSD_BUILD_SHARED_LIBS=ON" @@ -84,7 +84,7 @@ jobs: matrix: include: - build: 'cuda11' - defines: '-DSD_CUDA=ON -DSD_BUILD_SHARED_LIBS=ON -DCMAKE_CUDA_FLAGS="-allow-unsupported-compiler"' + defines: '-DSD_CUDA=ON -DSD_BUILD_SHARED_LIBS=ON -DCMAKE_CUDA_ARCHITECTURES=61 -DCMAKE_CUDA_FLAGS="-allow-unsupported-compiler"' steps: - name: Checkout @@ -225,9 +225,9 @@ jobs: - build: 'avx512' defines: '-DGGML_AVX512=ON -DSD_BUILD_SHARED_LIBS=ON' - build: 'cuda11' - defines: '-DSD_CUDA=ON -DSD_BUILD_SHARED_LIBS=ON' + defines: '-DSD_CUDA=ON -DCMAKE_CUDA_ARCHITECTURES=61 -DSD_BUILD_SHARED_LIBS=ON' - build: 'cuda12' - defines: '-DSD_CUDA=ON -DSD_BUILD_SHARED_LIBS=ON' + defines: '-DSD_CUDA=ON -DCMAKE_CUDA_ARCHITECTURES=61,75,86,89,100 -DSD_BUILD_SHARED_LIBS=ON' - build: 'sycl' defines: '-DSD_SYCL=ON -DCMAKE_C_COMPILER=icx -DCMAKE_CXX_COMPILER=icpx -DSD_BUILD_SHARED_LIBS=ON'