Skip to content

Commit 5b8b257

Browse files
committed
Merge remote-tracking branch 'origin/master'
2 parents dd1249b + a4e8251 commit 5b8b257

15 files changed

+5026
-0
lines changed

crates/cudnn/Cargo.toml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
[package]
2+
authors = ["frjnn <[email protected]>"]
3+
edition = "2021"
4+
name = "cudnn"
5+
version = "0.1.0"
6+
7+
[dependencies]
8+
cust = "0.2.2"

crates/cudnn/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
# cudnn-rs
2+
Type safe cuDNN wrapper for the Rust programming language.

crates/cudnn/bindgen.sh

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
bindgen "/usr/include/cudnn.h" \
2+
--size_t-is-usize \
3+
--allowlist-type "cudnn.*" \
4+
--allowlist-function "cudnn.*" \
5+
--allowlist-var "CUDNN.*" \
6+
--no-layout-tests \
7+
--no-doc-comments \
8+
--default-enum-style rust \
9+
-- -I "/usr/local/cuda/include" > src/sys.rs

crates/cudnn/build.rs

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
fn main() {
2+
println!("cargo:include=/usr/local/cuda/include");
3+
println!("cargo:rustc-link-lib=dylib=cudnn");
4+
println!("cargo:rerun-if-changed=build.rs");
5+
}

0 commit comments

Comments
 (0)