diff --git a/deeplearning4j/deeplearning4j-cuda/src/test/java/org/deeplearning4j/ValidateCuDNN.java b/deeplearning4j/deeplearning4j-cuda/src/test/java/org/deeplearning4j/ValidateCuDNN.java index 1d393aaf4..eed9f2efa 100644 --- a/deeplearning4j/deeplearning4j-cuda/src/test/java/org/deeplearning4j/ValidateCuDNN.java +++ b/deeplearning4j/deeplearning4j-cuda/src/test/java/org/deeplearning4j/ValidateCuDNN.java @@ -248,7 +248,7 @@ public class ValidateCuDNN extends BaseDL4JTest { Nd4j.getRandom().setSeed(12345); INDArray features = Nd4j.rand(fShape); INDArray labels = Nd4j.rand(lShape); - labels = Nd4j.exec(new IsMax(labels, 1)); + labels = Nd4j.exec(new IsMax(labels, 1))[0].castTo(features.dataType()); List testCaseList = new ArrayList<>(); @@ -256,7 +256,7 @@ public class ValidateCuDNN extends BaseDL4JTest { for (int i = 0; i < 6; i++) { INDArray f = Nd4j.rand(fShape); INDArray l = Nd4j.rand(lShape); - Nd4j.exec(new IsMax(l, 1))[0]; + l = Nd4j.exec(new IsMax(l, 1))[0].castTo(features.dataType()); dataSets.add(new DataSet(f, l)); } DataSetIterator iter = new ExistingDataSetIterator(dataSets); diff --git a/deeplearning4j/deeplearning4j-scaleout/spark/dl4j-spark/pom.xml b/deeplearning4j/deeplearning4j-scaleout/spark/dl4j-spark/pom.xml index 8bc852fbd..8b31872c5 100644 --- a/deeplearning4j/deeplearning4j-scaleout/spark/dl4j-spark/pom.xml +++ b/deeplearning4j/deeplearning4j-scaleout/spark/dl4j-spark/pom.xml @@ -70,6 +70,13 @@ deeplearning4j-play_2.11 ${deeplearning4j.version} test + + + + net.jpountz.lz4 + lz4 + +