From b1f8819bdecf67ea310b245a43d14c6d10a475e1 Mon Sep 17 00:00:00 2001
From: agibsonccc <adam@konduit.ai>
Date: Tue, 23 Mar 2021 20:21:56 +0900
Subject: [PATCH] Update Nd4jTestsC.java

---
 .../nd4j-tests/src/test/java/org/nd4j/linalg/Nd4jTestsC.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/Nd4jTestsC.java b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/Nd4jTestsC.java
index 56a713d6c..4d2ffed77 100644
--- a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/Nd4jTestsC.java
+++ b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/Nd4jTestsC.java
@@ -8591,7 +8591,7 @@ public class Nd4jTestsC extends BaseNd4jTestWithBackends {
     }
 
     private static INDArray fwd(INDArray input, INDArray W, INDArray b){
-        INDArray ret = Nd4j.createUninitialized(input.size(0), W.size(1));
+        INDArray ret = Nd4j.createUninitialized(input.size(0), W.size(1)).castTo(DataType.DOUBLE);
         input.mmuli(W, ret);
         ret.addiRowVector(b);
         return ret;