diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000..ee6259b --- /dev/null +++ b/.gitattributes @@ -0,0 +1,3 @@ +tensorrt-sys/3rdParty/TensorRT-5.1.5/libnvinfer_plugin_static.a filter=lfs diff=lfs merge=lfs -text +tensorrt-sys/3rdParty/TensorRT-5.1.5/libnvinfer_static.a filter=lfs diff=lfs merge=lfs -text +tensorrt-sys/3rdParty/TensorRT-5.1.5/libnvparsers_static.a filter=lfs diff=lfs merge=lfs -text diff --git a/tensorrt-sys/3rdParty/TensorRT-5.1.5/libnvinfer_plugin_static.a b/tensorrt-sys/3rdParty/TensorRT-5.1.5/libnvinfer_plugin_static.a new file mode 100644 index 0000000..06fdf8d --- /dev/null +++ b/tensorrt-sys/3rdParty/TensorRT-5.1.5/libnvinfer_plugin_static.a @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8c5e0094984b9462c18d91f9affcc56606e3eeb0ea72322ca452422d6fc0c11b +size 2636966 diff --git a/tensorrt-sys/3rdParty/TensorRT-5.1.5/libnvinfer_static.a b/tensorrt-sys/3rdParty/TensorRT-5.1.5/libnvinfer_static.a new file mode 100644 index 0000000..2931378 --- /dev/null +++ b/tensorrt-sys/3rdParty/TensorRT-5.1.5/libnvinfer_static.a @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:92e4c858e3eac82a3f161da87c7c36eb1620672e19210fd2640ec7cfcfdd027b +size 164301090 diff --git a/tensorrt-sys/3rdParty/TensorRT-5.1.5/libnvparsers_static.a b/tensorrt-sys/3rdParty/TensorRT-5.1.5/libnvparsers_static.a new file mode 100644 index 0000000..c561d37 --- /dev/null +++ b/tensorrt-sys/3rdParty/TensorRT-5.1.5/libnvparsers_static.a @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:896ac86fd8d9281635d70e63ea077798e49914b0db046d0caf27ad10a8901b7e +size 4377674 diff --git a/tensorrt-sys/build.rs b/tensorrt-sys/build.rs index 265dfcd..4dcca3a 100644 --- a/tensorrt-sys/build.rs +++ b/tensorrt-sys/build.rs @@ -43,6 +43,8 @@ fn configuration(full_library_path: &PathBuf) { println!("cargo:rustc-link-search=native={}", dst.display()); println!("cargo:rustc-flags=-l dylib=nvinfer"); println!("cargo:rustc-flags=-l dylib=nvparsers"); + println!("cargo:rustc-flags=-L {}/3rdParty/TensorRT-5.1.5", env!("CARGO_MANIFEST_DIR")); + println!("cargo:rustc-flags=-l static=nvinfer_plugin_static"); cuda_configuration(); } else { panic!("Invalid nvinfer version found. Expected: libnvinfer.so.5.1.5, Found: {}", full_library_path.to_str().unwrap()); @@ -67,6 +69,9 @@ fn configuration(full_library_path: &PathBuf) { fn cuda_configuration() { println!("cargo:rustc-flags=-L /usr/local/cuda-10.1/lib64"); println!("cargo:rustc-flags=-l dylib=cudart"); + println!("cargo:rustc-flags=-l dylib=cublas"); + println!("cargo:rustc-flags=-l dylib=cublasLt"); + println!("cargo:rustc-flags=-l dylib=cudnn"); } #[cfg(feature = "cuda-102")]