Adding feature flags for new supported cuda toolkit verions

coreylowman · Mar 21, 2024 · de96b8c · de96b8c
1 parent 3de960c
commit de96b8c
Show file tree

Hide file tree

Showing 8 changed files with 72 additions and 0 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -26,6 +26,8 @@ features = ["ci-check", "f16", "cudnn"]
 [features]
 default = ["std", "driver", "nvrtc", "cublas", "curand", "cuda_version_11_8"]
 cuda_version_11_8 = []
+cuda_version_12_0 = []
+cuda_version_12_1 = []
 cuda_version_12_2 = []
 nvrtc = []
 driver = ["nvrtc"]

diff --git a/src/cublas/sys/mod.rs b/src/cublas/sys/mod.rs
@@ -3,6 +3,16 @@ mod sys_11080;
 #[cfg(feature = "cuda_version_11_8")]
 pub use sys_11080::*;
 
+#[cfg(feature = "cuda_version_12_0")]
+mod sys_12000;
+#[cfg(feature = "cuda_version_12_0")]
+pub use sys_12000::*;
+
+#[cfg(feature = "cuda_version_12_1")]
+mod sys_12010;
+#[cfg(feature = "cuda_version_12_1")]
+pub use sys_12010::*;
+
 #[cfg(feature = "cuda_version_12_2")]
 mod sys_12020;
 #[cfg(feature = "cuda_version_12_2")]

diff --git a/src/cublaslt/sys/mod.rs b/src/cublaslt/sys/mod.rs
@@ -3,6 +3,16 @@ mod sys_11080;
 #[cfg(feature = "cuda_version_11_8")]
 pub use sys_11080::*;
 
+#[cfg(feature = "cuda_version_12_0")]
+mod sys_12000;
+#[cfg(feature = "cuda_version_12_0")]
+pub use sys_12000::*;
+
+#[cfg(feature = "cuda_version_12_1")]
+mod sys_12010;
+#[cfg(feature = "cuda_version_12_1")]
+pub use sys_12010::*;
+
 #[cfg(feature = "cuda_version_12_2")]
 mod sys_12020;
 #[cfg(feature = "cuda_version_12_2")]

diff --git a/src/cudnn/sys/mod.rs b/src/cudnn/sys/mod.rs
@@ -3,6 +3,16 @@ mod sys_11080;
 #[cfg(feature = "cuda_version_11_8")]
 pub use sys_11080::*;
 
+#[cfg(feature = "cuda_version_12_0")]
+mod sys_12000;
+#[cfg(feature = "cuda_version_12_0")]
+pub use sys_12000::*;
+
+#[cfg(feature = "cuda_version_12_1")]
+mod sys_12010;
+#[cfg(feature = "cuda_version_12_1")]
+pub use sys_12010::*;
+
 #[cfg(feature = "cuda_version_12_2")]
 mod sys_12020;
 #[cfg(feature = "cuda_version_12_2")]

diff --git a/src/curand/sys/mod.rs b/src/curand/sys/mod.rs
@@ -3,6 +3,16 @@ mod sys_11080;
 #[cfg(feature = "cuda_version_11_8")]
 pub use sys_11080::*;
 
+#[cfg(feature = "cuda_version_12_0")]
+mod sys_12000;
+#[cfg(feature = "cuda_version_12_0")]
+pub use sys_12000::*;
+
+#[cfg(feature = "cuda_version_12_1")]
+mod sys_12010;
+#[cfg(feature = "cuda_version_12_1")]
+pub use sys_12010::*;
+
 #[cfg(feature = "cuda_version_12_2")]
 mod sys_12020;
 #[cfg(feature = "cuda_version_12_2")]

diff --git a/src/driver/sys/mod.rs b/src/driver/sys/mod.rs
@@ -3,6 +3,16 @@ mod sys_11080;
 #[cfg(feature = "cuda_version_11_8")]
 pub use sys_11080::*;
 
+#[cfg(feature = "cuda_version_12_0")]
+mod sys_12000;
+#[cfg(feature = "cuda_version_12_0")]
+pub use sys_12000::*;
+
+#[cfg(feature = "cuda_version_12_1")]
+mod sys_12010;
+#[cfg(feature = "cuda_version_12_1")]
+pub use sys_12010::*;
+
 #[cfg(feature = "cuda_version_12_2")]
 mod sys_12020;
 #[cfg(feature = "cuda_version_12_2")]

diff --git a/src/nccl/sys/mod.rs b/src/nccl/sys/mod.rs
@@ -3,6 +3,16 @@ mod sys_11080;
 #[cfg(feature = "cuda_version_11_8")]
 pub use sys_11080::*;
 
+#[cfg(feature = "cuda_version_12_0")]
+mod sys_12000;
+#[cfg(feature = "cuda_version_12_0")]
+pub use sys_12000::*;
+
+#[cfg(feature = "cuda_version_12_1")]
+mod sys_12010;
+#[cfg(feature = "cuda_version_12_1")]
+pub use sys_12010::*;
+
 #[cfg(feature = "cuda_version_12_2")]
 mod sys_12020;
 #[cfg(feature = "cuda_version_12_2")]

diff --git a/src/nvrtc/sys/mod.rs b/src/nvrtc/sys/mod.rs
@@ -3,6 +3,16 @@ mod sys_11080;
 #[cfg(feature = "cuda_version_11_8")]
 pub use sys_11080::*;
 
+#[cfg(feature = "cuda_version_12_0")]
+mod sys_12000;
+#[cfg(feature = "cuda_version_12_0")]
+pub use sys_12000::*;
+
+#[cfg(feature = "cuda_version_12_1")]
+mod sys_12010;
+#[cfg(feature = "cuda_version_12_1")]
+pub use sys_12010::*;
+
 #[cfg(feature = "cuda_version_12_2")]
 mod sys_12020;
 #[cfg(feature = "cuda_version_12_2")]