From 67e32adb5605894791f421530e8a2750c13a324e Mon Sep 17 00:00:00 2001 From: brian Date: Fri, 10 Mar 2023 17:17:19 +0100 Subject: [PATCH] Adding cuDNN support Signed-off-by: brian --- cavis-dnn/cavis-dnn-cudnn/build.gradle | 2 +- cavis-native/cavis-native-jcublas/build.gradle | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/cavis-dnn/cavis-dnn-cudnn/build.gradle b/cavis-dnn/cavis-dnn-cudnn/build.gradle index 725ca1f85..8f40cf470 100644 --- a/cavis-dnn/cavis-dnn-cudnn/build.gradle +++ b/cavis-dnn/cavis-dnn-cudnn/build.gradle @@ -7,7 +7,7 @@ ext { dependencies { implementation platform(projects.cavisCommonPlatform) - implementation projects.cavisNative.cavisNativeJcublas + implementation project(":cavis-native:cavis-native-jcublas") implementation projects.cavisDnn.cavisDnnApi implementation projects.cavisDnn.cavisDnnNn diff --git a/cavis-native/cavis-native-jcublas/build.gradle b/cavis-native/cavis-native-jcublas/build.gradle index 0e0a9dd22..cf3372c5c 100644 --- a/cavis-native/cavis-native-jcublas/build.gradle +++ b/cavis-native/cavis-native-jcublas/build.gradle @@ -11,7 +11,8 @@ ext { dependencies { 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", classifier: buildTarget