ZLUDA/zluda_dnn
2024-04-29 22:56:19 +09:00
..
build Nobody expects the Red Team 2024-02-11 20:45:51 +01:00
src Remove zluda_dnn remains. 2024-04-29 22:56:19 +09:00
Cargo.toml Disable DNN build on Windows. 2024-04-06 13:51:46 +09:00
README Nobody expects the Red Team 2024-02-11 20:45:51 +01:00

bindgen /usr/include/x86_64-linux-gnu/cudnn_v7.h -o src/cudnn_types_v7.rs --allowlist-type="^cudnn.*reference_t$"  --blocklist-function="^.*" --default-enum-style=newtype --no-layout-tests --no-derive-debug -- -I/usr/local/cuda/targets/x86_64-linux/include
bindgen /usr/include/x86_64-linux-gnu/cudnn_v8.h -o src/cudnn_types_v8.rs --allowlist-type="^libraryPropertyType$"  --allowlist-type="^cudnn.*"  --blocklist-function=".*" --default-enum-style=newtype --no-layout-tests --no-derive-debug -- -I/usr/local/cuda/targets/x86_64-linux/include -Ibuild
bindgen /usr/include/x86_64-linux-gnu/cudnn_v7.h -o src/cudnn_v7.rs --blocklist-type=".*" --allowlist-function="^cudnnGetConvolutionBackwardDataAlgorithm$" --allowlist-function="^cudnnGetConvolutionBackwardFilterAlgorithm$" --allowlist-function="^cudnnGetConvolutionForwardAlgorithm$" --allowlist-function="^cudnnSetRNNDescriptor$" --allowlist-function="^cudnnSetRNNDescriptor_v5$" --default-enum-style=newtype --no-layout-tests --no-derive-debug -- -I/usr/local/cuda/targets/x86_64-linux/include
bindgen /usr/include/x86_64-linux-gnu/cudnn_v8.h -o src/cudnn_v8.rs --blocklist-type=".*" --allowlist-function="^cudnn.*" --default-enum-style=newtype --no-layout-tests --no-derive-debug -- -I/usr/local/cuda/targets/x86_64-linux/include -I../zluda_dnn/build
sed -i -e 's/extern "C" {//g' -e 's/-> cudnnStatus_t;/-> cudnnStatus_t { crate::unsupported()/g' -e 's/pub fn /#[no_mangle] pub unsafe extern "system" fn /g' src/cudnn_v7.rs
sed -i -e 's/extern "C" {//g' -e 's/-> cudnnStatus_t;/-> cudnnStatus_t { crate::unsupported()/g' -e 's/pub fn /#[no_mangle] pub unsafe extern "system" fn /g' src/cudnn_v8.rs