Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Adding cuDNN support 

Signed-off-by: brian <brian@brutex.de>
master
Brian Rosenberger 2023-03-10 17:17:19 +01:00
parent e59e14da35
commit 67e32adb56
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cavis-dnn/cavis-dnn-cudnn/build.gradle
View File

@ -7,7 +7,7 @@ ext {
dependencies { dependencies {
implementation platform(projects.cavisCommonPlatform) implementation platform(projects.cavisCommonPlatform)
implementation projects.cavisNative.cavisNativeJcublas implementation project(":cavis-native:cavis-native-jcublas")
implementation projects.cavisDnn.cavisDnnApi implementation projects.cavisDnn.cavisDnnApi
implementation projects.cavisDnn.cavisDnnNn implementation projects.cavisDnn.cavisDnnNn

3
cavis-native/cavis-native-jcublas/build.gradle
View File

@ -11,7 +11,8 @@ ext {
dependencies { dependencies {
implementation platform(projects.cavisCommonPlatform) implementation platform(projects.cavisCommonPlatform)
implementation project(":cavis-native:cavis-native-blas") //implementation project(":cavis-native:cavis-native-blas")
implementation projects.cavisNative.cavisNativeBlas
implementation group: "org.bytedeco", name: "cuda" implementation group: "org.bytedeco", name: "cuda"
implementation group: "org.bytedeco", name: "cuda", classifier: buildTarget implementation group: "org.bytedeco", name: "cuda", classifier: buildTarget