Add NVIDIA Tools Extension wrapper

Cargo.toml

@@ -4,4 +4,5 @@ members = [
   "cuda-config",
   "cuda-driver-sys",
   "cuda-runtime-sys",
+  "nvtx-sys",
 ]

bindgen.sh

@@ -0,0 +1,16 @@
+#!/bin/bash
+set -exu
+
+bindgen \
+  --whitelist-type="^nvtx.*" \
+  --whitelist-var="^nvtx.*" \
+  --whitelist-function="^nvtx.*" \
+  --default-enum-style=rust \
+  --no-doc-comments \
+  --with-derive-default \
+  --with-derive-eq \
+  --with-derive-hash \
+  --with-derive-ord \
+  --size_t-is-usize \
+  "$CUDA_PATH/include/nvToolsExt.h" \
+  > src/nv_tools_ext.rs

nvtx-sys/Cargo.toml

@@ -0,0 +1,20 @@
+[package]
+name = "nvtx-sys"
+version = "0.3.0-alpha.0"
+authors = ["Toshiki Teramura <[email protected]>", "Clemens Lutz <[email protected]>"]
+edition = "2018"
+
+build = "build.rs"
+links = "nvToolsExt"
+
+description   = "Rust binding to NVIDIA Tools Extension Library"
+documentation = "https://docs.rs/nvtx-sys/"
+repository    = "https://github.com/rust-cuda/cuda-sys"
+keywords      = ["GPGPU", "CUDA", "ffi"]
+license       = "MIT/Apache-2.0"
+readme        = "../README.md"
+categories    = []
+
+[build-dependencies.cuda-config]
+path = "../cuda-config"
+version = "0.1.0"

nvtx-sys/bindgen.sh

@@ -0,0 +1,16 @@
+#!/bin/bash
+set -exu
+
+bindgen \
+  --whitelist-type="^nvtx.*" \
+  --whitelist-var="^nvtx.*" \
+  --whitelist-function="^nvtx.*" \
+  --default-enum-style=rust \
+  --no-doc-comments \
+  --with-derive-default \
+  --with-derive-eq \
+  --with-derive-hash \
+  --with-derive-ord \
+  --size_t-is-usize \
+  "$CUDA_PATH/include/nvToolsExt.h" \
+  > src/nv_tools_ext.rs

nvtx-sys/build.rs

@@ -0,0 +1,18 @@
+use cuda_config::*;
+
+fn main() {
+    if cfg!(target_os = "windows") {
+        println!(
+            "cargo:rustc-link-search=native={}",
+            find_cuda_windows().display()
+        );
+    } else {
+        for path in find_cuda() {
+            println!("cargo:rustc-link-search=native={}", path.display());
+        }
+    };
+
+    println!("cargo:rustc-link-lib=dylib=nvToolsExt");
+    println!("cargo:rerun-if-changed=build.rs");
+    println!("cargo:rerun-if-env-changed=CUDA_LIBRARY_PATH");
+}

nvtx-sys/src/lib.rs

@@ -0,0 +1,18 @@
+#![allow(non_snake_case, non_camel_case_types, non_upper_case_globals)]
+#![allow(deref_nullptr)] // https://github.com/rust-lang/rust-bindgen/issues/1651
+include!("nv_tools_ext.rs");
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use std::ffi::CStr;
+
+    // not a test for cublas initialization, but a test for linking cublasCreate_v2
+    #[test]
+    fn link_test() {
+        unsafe {
+            let name = CStr::from_bytes_with_nul_unchecked(b"my mark\0");
+            nvtxMarkA(name.as_ptr());
+        }
+    }
+}