feat(add cuda-12080)

Cargo.toml

@@ -10,18 +10,12 @@ documentation = "https://docs.rs/cudarc"
 repository = "https://github.com/coreylowman/cudarc"
 readme = "README.md"
 
-keywords = [
-    "cuda",
-    "nvidia",
-    "gpu",
-    "nvrtc",
-    "cublas",
-]
+keywords = ["cuda", "nvidia", "gpu", "nvrtc", "cublas"]
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [package.metadata.docs.rs]
-features = ["cuda-12060", "f16", "cudnn"]
+features = ["cuda-12080", "f16", "cudnn"]
 
 [features]
 default = ["std", "cublas", "cublaslt", "curand", "driver", "runtime", "nvrtc"]
@@ -39,6 +33,7 @@ cuda-12030 = []
 cuda-12040 = []
 cuda-12050 = []
 cuda-12060 = []
+cuda-12080 = []
 
 dynamic-linking = []
 
@@ -56,7 +51,12 @@ no-std = ["no-std-compat/std", "dep:spin"]
 f16 = ["dep:half"]
 
 [dependencies]
-spin = { version = "0.9.8", optional = true, features = ["rwlock"], default-features = false }
-no-std-compat = { version = "0.4.1", optional = true, features = [ "alloc" ] }
-half = { version = "2.4.1", optional = true, default-features = false, features = ["num-traits", "rand_distr"] }
+spin = { version = "0.9.8", optional = true, features = [
+    "rwlock",
+], default-features = false }
+no-std-compat = { version = "0.4.1", optional = true, features = ["alloc"] }
+half = { version = "2.4.1", optional = true, default-features = false, features = [
+    "num-traits",
+    "rand_distr",
+] }
 libloading = "0.8.5"

build.rs

@@ -6,7 +6,9 @@ fn main() {
     println!("cargo:rerun-if-env-changed=CUDA_PATH");
     println!("cargo:rerun-if-env-changed=CUDA_TOOLKIT_ROOT_DIR");
 
-    let (major, minor): (usize, usize) = if cfg!(feature = "cuda-12060") {
+    let (major, minor): (usize, usize) = if cfg!(feature = "cuda-12080") {
+        (12, 8)
+    } else if cfg!(feature = "cuda-12060") {
         (12, 6)
     } else if cfg!(feature = "cuda-12050") {
         (12, 5)
@@ -32,7 +34,7 @@ fn main() {
         (11, 4)
     } else {
         #[cfg(not(feature = "cuda-version-from-build-system"))]
-        panic!("Must specify one of the following features: [cuda-version-from-build-system, cuda-12060, cuda-12050, cuda-12040, cuda-12030, cuda-12020, cuda-12010, cuda-12000, cuda-11080, cuda-11070, cuda-11060, cuda-11050, cuda-11040]");
+        panic!("Must specify one of the following features: [cuda-version-from-build-system, cuda-12080, cuda-12060, cuda-12050, cuda-12040, cuda-12030, cuda-12020, cuda-12010, cuda-12000, cuda-11080, cuda-11070, cuda-11060, cuda-11050, cuda-11040]");
 
         #[cfg(feature = "cuda-version-from-build-system")]
         {
@@ -70,6 +72,7 @@ fn cuda_version_from_build_system() -> (usize, usize) {
     let version_number = release_section.split(' ').nth(1).unwrap();
 
     match version_number {
+        "12.8" => (12, 8),
         "12.6" => (12, 6),
         "12.5" => (12, 5),
         "12.4" => (12, 4),

run-bindgen.sh

@@ -1,6 +1,15 @@
+if [ -z "${CUDA_ROOT}" ]; then
+    CUDA_DIR=$CUDA_ROOT
+elif [ -d "/opt/cuda" ]; then
+    CUDA_DIR="/opt/cuda"
+else
+    CUDA_DIR="/usr/local/cuda"
+fi
+
+export CUDA_INCLUDES="${CUDA_DIR}/include"
+
 declare -a modules=("cublas" "cublaslt" "cudnn" "curand" "driver" "runtime" "nccl" "nvrtc")
-for path in "${modules[@]}"
-do
+for path in "${modules[@]}"; do
     cd src/${path}/sys
     bash bindgen.sh
     cd ../../../

src/cublas/sys/bindgen.sh

@@ -2,19 +2,19 @@
 set -exu
 
 bindgen \
-  --allowlist-var="^CUDA_VERSION" \
-  --allowlist-type="^cublas.*" \
-  --allowlist-function="^cublas.*" \
-  --default-enum-style=rust \
-  --no-doc-comments \
-  --with-derive-default \
-  --with-derive-eq \
-  --with-derive-hash \
-  --with-derive-ord \
-  --use-core \
-  --dynamic-loading Lib \
-  wrapper.h -- -I/usr/local/cuda/include \
-  > tmp.rs
+    --allowlist-var="^CUDA_VERSION" \
+    --allowlist-type="^cublas.*" \
+    --allowlist-function="^cublas.*" \
+    --default-enum-style=rust \
+    --no-doc-comments \
+    --with-derive-default \
+    --with-derive-eq \
+    --with-derive-hash \
+    --with-derive-ord \
+    --use-core \
+    --dynamic-loading Lib \
+    wrapper.h -- -I$CUDA_INCLUDES \
+    >tmp.rs
 
 CUDA_VERSION=$(cat tmp.rs | grep "CUDA_VERSION" | awk '{ print $6 }' | sed 's/.$//')
 mv tmp.rs sys_${CUDA_VERSION}.rs

src/cublas/sys/mod.rs

@@ -58,6 +58,11 @@ mod sys_12060;
 #[cfg(feature = "cuda-12060")]
 pub use sys_12060::*;
 
+#[cfg(feature = "cuda-12080")]
+mod sys_12080;
+#[cfg(feature = "cuda-12080")]
+pub use sys_12080::*;
+
 pub unsafe fn lib() -> &'static Lib {
     static LIB: std::sync::OnceLock<Lib> = std::sync::OnceLock::new();
     LIB.get_or_init(|| {