support cuda toolkit version: 11.7 (#214)

* support cuda toolkit version: 11.7 * support cuda toolkit version: 11.7 add cudnn & nccl 1 cudnn/sys/sys_11070.rs 2 nccl/sys/sys_11070.rs --------- Co-authored-by: wenhaozhao <wenhaozhao@sohu-inc.com>
coreylowman · Mar 28, 2024 · c45c1d2 · c45c1d2
1 parent 005a999
commit c45c1d2
Show file tree

Hide file tree

Showing 15 changed files with 15,287 additions and 0 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -31,6 +31,7 @@ cuda_11080 = []
 cuda_12000 = []
 cuda_12010 = []
 cuda_12020 = []
+cuda_11070 = []
 
 nvrtc = []
 driver = ["nvrtc"]

diff --git a/build.rs b/build.rs
@@ -38,6 +38,7 @@ fn link_cuda() {
             "12010" => println!("cargo:rustc-cfg=feature=\"cuda_12010\""),
             "12000" => println!("cargo:rustc-cfg=feature=\"cuda_12000\""),
             "11080" => println!("cargo:rustc-cfg=feature=\"cuda_11080\""),
+            "11070" => println!("cargo:rustc-cfg=feature=\"cuda_11070\""),
             v => panic!("Unsupported cuda toolkit version: `{v}`. Please raise a github issue."),
         }
     }

diff --git a/src/cublas/sys/mod.rs b/src/cublas/sys/mod.rs
@@ -1,3 +1,8 @@
+#[cfg(feature = "cuda_11070")]
+mod sys_11070;
+#[cfg(feature = "cuda_11070")]
+pub use sys_11070::*;
+
 #[cfg(feature = "cuda_11080")]
 mod sys_11080;
 #[cfg(feature = "cuda_11080")]