CI: reduce cuda versions for pull_request

bitsandbytes-foundation · Feb 7, 2024 · 8a5c804 · 8a5c804
1 parent 788c200
commit 8a5c804
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -116,7 +116,7 @@ jobs:
       matrix:
         os: [ubuntu-latest, windows-latest]
         arch: [x86_64, aarch64]
-        cuda-version: ['11.8.0', '12.1.1']
+        cuda-version: ${{ startsWith(github.ref, 'refs/tags') && fromJson('["11.8.0","12.1.1"]') || fromJson('["12.1.1"]') }}
         build_type: [Release]
         exclude:
           - os: windows-latest # This probably requires arm64 Windows agents