Add bindings for all modules except cudnn and nccl

coreylowman · Jun 25, 2024 · 4eee547 · 4eee547
1 parent 0200c29
commit 4eee547
Show file tree

Hide file tree

Showing 13 changed files with 26,686 additions and 3 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -25,8 +25,10 @@ features = ["cuda-12050", "f16", "cudnn"]
 
 [features]
 default = ["std", "cublas", "cublaslt", "cudnn", "curand", "driver", "nccl", "nvrtc"]
+no-cudnn = ["std", "cublas", "cublaslt", "curand", "driver", "nvrtc"]
 
 cuda-version-from-build-system = []
+cuda-11040 = []
 cuda-11050 = []
 cuda-11060 = []
 cuda-11070 = []

diff --git a/build.rs b/build.rs
@@ -26,9 +26,11 @@ fn main() {
         (11, 6)
     } else if cfg!(feature = "cuda-11050") {
         (11, 5)
+    } else if cfg!(feature = "cuda-11040") {
+        (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-12050, cuda-12040, cuda-12030, cuda-12020, cuda-12010, cuda-12000, cuda-11080, cuda-11070, cuda-11060, cuda-11050]");
+        panic!("Must specify one of the following features: [cuda-version-from-build-system, 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")]
         {

diff --git a/src/cublas/sys/mod.rs b/src/cublas/sys/mod.rs
@@ -1,3 +1,8 @@
+#[cfg(feature = "cuda-11040")]
+mod sys_11040;
+#[cfg(feature = "cuda-11040")]
+pub use sys_11040::*;
+
 #[cfg(feature = "cuda-11050")]
 mod sys_11050;
 #[cfg(feature = "cuda-11050")]