From 8d31caffbe30bb0b3c34b342e22fa7817c7980ad Mon Sep 17 00:00:00 2001 From: brian Date: Fri, 10 Mar 2023 16:32:41 +0100 Subject: [PATCH] Adding cuDNN support Signed-off-by: brian --- brutex-extended-tests/src/test/java/net/brutex/gan/App.java | 2 +- .../src/main/java/org/nd4j/linalg/factory/Nd4j.java | 6 +++--- .../org/deeplearning4j/common/config/DL4JClassLoading.java | 3 ++- 3 files changed, 6 insertions(+), 5 deletions(-) diff --git a/brutex-extended-tests/src/test/java/net/brutex/gan/App.java b/brutex-extended-tests/src/test/java/net/brutex/gan/App.java index f4feb6fdf..5d4704f2c 100644 --- a/brutex-extended-tests/src/test/java/net/brutex/gan/App.java +++ b/brutex-extended-tests/src/test/java/net/brutex/gan/App.java @@ -167,7 +167,7 @@ public class App { trainData.reset(); int j = 0; - for (int i = 0; i < 10; i++) { + for (int i = 0; i < 20; i++) { while (trainData.hasNext()) { j++; diff --git a/cavis-dnn/cavis-dnn-api/src/main/java/org/nd4j/linalg/factory/Nd4j.java b/cavis-dnn/cavis-dnn-api/src/main/java/org/nd4j/linalg/factory/Nd4j.java index f542e3cce..359f30b02 100644 --- a/cavis-dnn/cavis-dnn-api/src/main/java/org/nd4j/linalg/factory/Nd4j.java +++ b/cavis-dnn/cavis-dnn-api/src/main/java/org/nd4j/linalg/factory/Nd4j.java @@ -4877,7 +4877,7 @@ public class Nd4j { * Create an ndarray of zeros * * @param shape the shape of the array - * @return an ndarray with ones filled in + * @return an ndarray with zeros filled in */ public static INDArray zeros(int[] shape, char order) { checkShapeValues(shape); @@ -4896,7 +4896,7 @@ public class Nd4j { * Create an ndarray of zeros * * @param shape the shape of the array - * @return an ndarray with ones filled in + * @return an ndarray with zeros filled in */ public static INDArray zeros(@NonNull int... shape) { return Nd4j.create(shape); @@ -4907,7 +4907,7 @@ public class Nd4j { * Create an ndarray of zeros * * @param shape the shape of the array - * @return an ndarray with ones filled in + * @return an ndarray with zeros filled in */ public static INDArray zeros(@NonNull long... shape) { return Nd4j.create(shape); diff --git a/cavis-dnn/cavis-dnn-common/src/main/java/org/deeplearning4j/common/config/DL4JClassLoading.java b/cavis-dnn/cavis-dnn-common/src/main/java/org/deeplearning4j/common/config/DL4JClassLoading.java index 16f6f134a..55481d875 100644 --- a/cavis-dnn/cavis-dnn-common/src/main/java/org/deeplearning4j/common/config/DL4JClassLoading.java +++ b/cavis-dnn/cavis-dnn-common/src/main/java/org/deeplearning4j/common/config/DL4JClassLoading.java @@ -105,7 +105,8 @@ public class DL4JClassLoading { throw new RuntimeException(instantiationException); } catch (InvocationTargetException instantiationException) { - log.error(String.format("InvocationTargetException was '%s'.", instantiationException.getTargetException().getMessage()), instantiationException); + log.error(String.format("---------- ----------- ---------- \nInvocationTargetException was '%s'.", instantiationException.getTargetException().getMessage()), instantiationException); + log.error(String.format("java.library.path was '%s'\n---------- ---------- ----------", System.getProperty("java.library.path"))); throw new RuntimeException(instantiationException); } }