CUDA runtime api

Cargo.toml

@@ -24,7 +24,7 @@ keywords = [
 features = ["cuda-12060", "f16", "cudnn"]
 
 [features]
-default = ["std", "cublas", "cublaslt", "curand", "driver", "nvrtc"]
+default = ["std", "cublas", "cublaslt", "curand", "driver", "runtime", "nvrtc"]
 
 cuda-version-from-build-system = []
 cuda-11040 = []
@@ -46,6 +46,7 @@ nvrtc = []
 driver = ["nvrtc"]
 cublas = ["driver"]
 cublaslt = ["driver"]
+runtime = ["driver"]
 cudnn = ["driver"]
 curand = ["driver"]
 nccl = ["driver"]

build.rs

@@ -138,6 +138,8 @@ fn dynamic_linking(major: usize, minor: usize) {
     println!("cargo:rustc-link-lib=dylib=cublasLt");
     #[cfg(feature = "cudnn")]
     println!("cargo:rustc-link-lib=dylib=cudnn");
+    #[cfg(feature = "runtime")]
+    println!("cargo:rustc-link-lib=dylib=cudart");
 }
 
 #[allow(unused)]

run-bindgen.sh

@@ -1,4 +1,4 @@
-declare -a modules=("cublas" "cublaslt" "cudnn" "curand" "driver" "nccl" "nvrtc")
+declare -a modules=("cublas" "cublaslt" "cudnn" "curand" "driver" "runtime" "nccl" "nvrtc")
 for path in "${modules[@]}"
 do
     cd src/${path}/sys

src/lib.rs

@@ -87,6 +87,8 @@ pub mod cudnn;
 pub mod curand;
 #[cfg(feature = "driver")]
 pub mod driver;
+#[cfg(feature = "runtime")]
+pub mod runtime;
 #[cfg(feature = "nccl")]
 pub mod nccl;
 #[cfg(feature = "nvrtc")]

src/runtime/mod.rs

@@ -0,0 +1,6 @@
+//! Wrappers around the [CUDA Runtime API](https://docs.nvidia.com/cuda/cuda-runtime-api/index.html),
+//! in two levels: an unsafe low-level API and a (still unsafe) thin wrapper around it.
+
+pub mod result;
+#[allow(warnings)]
+pub mod sys;