initial support blackwell

NVlabs · Jan 21, 2025 · 9144e12 · 9144e12
1 parent 1e38444
commit 9144e12
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -15,9 +15,15 @@ jobs:
     strategy:
       matrix:
         include:
+          - os: ubuntu-24.04
+          - cuda: "12.8"
+          - arch: 120 # RTX50
+          - os: ubuntu-24.04
+          - cuda: "12.7"
+          - arch: 100 # B100
           - os: ubuntu-24.04
             cuda: "12.6"
-            arch: 89
+            arch: 89 # ADA
           - os: ubuntu-24.04
             cuda: "12.5"
             arch: 86
@@ -72,6 +78,10 @@ jobs:
     strategy:
       matrix:
         include:
+          - os: windows-2025
+            visual_studio: "Visual Studio 17 2022"
+            cuda: "12.8"
+            arch: 120
           - os: windows-2025
             visual_studio: "Visual Studio 17 2022"
             cuda: "12.6.3"

diff --git a/dependencies/cuda-cmake-github-actions/scripts/actions/install_cuda_windows.ps1 b/dependencies/cuda-cmake-github-actions/scripts/actions/install_cuda_windows.ps1
@@ -29,6 +29,7 @@ $CUDA_KNOWN_URLS = @{
     "11.8.0" = "https://developer.download.nvidia.com/compute/cuda/11.8.0/network_installers/cuda_11.8.0_windows_network.exe";
     "12.5.0" = "https://developer.download.nvidia.com/compute/cuda/12.5.0/network_installers/cuda_12.5.0_windows_network.exe";
  	"12.6.3" = "https://developer.download.nvidia.com/compute/cuda/12.6.3/network_installers/cuda_12.6.3_windows_network.exe";
+ 	"12.8.0" = "https://developer.download.nvidia.com/compute/cuda/12.8.0/network_installers/cuda_12.8.0_windows_network.exe";
 }
 
 # @todo - change this to be based on _MSC_VER intead, or invert it to be CUDA keyed instead?