cudaPackages_12_8: init at 12.8.0

pkgs/development/cuda-modules/cuda/extension.nix

@@ -18,6 +18,7 @@ let
     "12.3" = "12.3.0";
     "12.4" = "12.4.0";
     "12.6" = "12.6.0";
+    "12.8" = "12.8.0";
   };
 
   # Check if the current CUDA version is supported.

pkgs/development/cuda-modules/cudatoolkit/releases.nix

@@ -99,4 +99,10 @@
     url = "https://developer.download.nvidia.com/compute/cuda/12.6.0/local_installers/cuda_12.6.0_560.28.03_linux.run";
     sha256 = "sha256-MasEOU5psU3YZW4rRMKHfbGg6Jjf+KdUakxihDgQG5Q=";
   };
+
+  "12.8" = {
+    version = "12.8.0";
+    url = "https://developer.download.nvidia.com/compute/cuda/12.8.0/local_installers/cuda_12.8.0_570.86.10_linux.run";
+    sha256 = "sha256-YQhn3NbZTE42xJJPHQG52yjsCBZOivbHZPIbhCAGlfg=";
+  };
 }

pkgs/development/cuda-modules/gpus.nix

@@ -184,4 +184,58 @@
     dontDefaultAfter = null;
     maxCudaVersion = null;
   }
+  {
+    # NVIDIA B100
+    archName = "Blackwell";
+    computeCapability = "10.0";
+    isJetson = false;
+    minCudaVersion = "12.8";
+    dontDefaultAfter = null;
+    maxCudaVersion = null;
+  }
+  {
+    # NVIDIA B100 Accelerated
+    archName = "Blackwell";
+    computeCapability = "10.0a";
+    isJetson = false;
+    minCudaVersion = "12.8";
+    dontDefaultAfter = null;
+    maxCudaVersion = null;
+  }
+  {
+    # NVIDIA Blackwell
+    archName = "Blackwell";
+    computeCapability = "10.1";
+    isJetson = false;
+    minCudaVersion = "12.8";
+    dontDefaultAfter = null;
+    maxCudaVersion = null;
+  }
+  {
+    # NVIDIA Blackwell Accelerated
+    archName = "Blackwell";
+    computeCapability = "10.1a";
+    isJetson = false;
+    minCudaVersion = "12.8";
+    dontDefaultAfter = null;
+    maxCudaVersion = null;
+  }
+  {
+    # NVIDIA GeForce RTX 5090 (GB202), RTX 5080 (GB203), RTX 5070 (GB205)
+    archName = "Blackwell";
+    computeCapability = "12.0";
+    isJetson = false;
+    minCudaVersion = "12.8";
+    dontDefaultAfter = null;
+    maxCudaVersion = null;
+  }
+  {
+    # NVIDIA Blackwell Accelerated
+    archName = "Blackwell";
+    computeCapability = "12.0a";
+    isJetson = false;
+    minCudaVersion = "12.8";
+    dontDefaultAfter = null;
+    maxCudaVersion = null;
+  }
 ]

pkgs/development/cuda-modules/nvcc-compatibilities.nix

@@ -98,7 +98,8 @@ let
     # No changes from 12.2 to 12.3
     "12.3" = attrs."12.2";
 
-    # No changes from 12.2 to 12.3
+    # Added support for Clang 17 and GCC 13
+    # https://docs.nvidia.com/cuda/archive/12.4.0/cuda-installation-guide-linux/index.html#host-compiler-support-policy
     "12.4" = attrs."12.3" // {
       clangMaxMajorVersion = "17";
       gccMaxMajorVersion = "13";
@@ -110,6 +111,12 @@ let
       clangMaxMajorVersion = "18";
     };
 
+    # Added support for Clang 19 and GCC 14
+    # https://docs.nvidia.com/cuda/archive/12.8.0/cuda-installation-guide-linux/index.html#host-compiler-support-policy
+    "12.8" = attrs."12.6" // {
+      clangMaxMajorVersion = "19";
+      gccMaxMajorVersion = "14";
+    };
   };
 in
 attrs

pkgs/top-level/all-packages.nix

@@ -2968,6 +2968,7 @@ with pkgs;
   cudaPackages_12_3 = callPackage ./cuda-packages.nix { cudaVersion = "12.3"; };
   cudaPackages_12_4 = callPackage ./cuda-packages.nix { cudaVersion = "12.4"; };
   cudaPackages_12_6 = callPackage ./cuda-packages.nix { cudaVersion = "12.6"; };
+  cudaPackages_12_8 = callPackage ./cuda-packages.nix { cudaVersion = "12.8"; };
   cudaPackages_12 = cudaPackages_12_4; # Latest supported by cudnn
 
   cudaPackages = recurseIntoAttrs cudaPackages_12;