From 0adce9a4fa971906e77fc84408e87b8b7975ca9d Mon Sep 17 00:00:00 2001 From: raver119 Date: Wed, 21 Aug 2019 16:34:07 +0300 Subject: [PATCH] minor fix for msvc Signed-off-by: raver119 --- libnd4j/tests_cpu/layers_tests/DeclarableOpsTests9.cpp | 2 +- libnd4j/tests_cpu/layers_tests/testlayers.h | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/libnd4j/tests_cpu/layers_tests/DeclarableOpsTests9.cpp b/libnd4j/tests_cpu/layers_tests/DeclarableOpsTests9.cpp index d27aa4e46..ba090703e 100644 --- a/libnd4j/tests_cpu/layers_tests/DeclarableOpsTests9.cpp +++ b/libnd4j/tests_cpu/layers_tests/DeclarableOpsTests9.cpp @@ -656,7 +656,7 @@ TEST_F(DeclarableOpsTests9, concat_test19) { context.setInputArray(e, array, true); } - auto z = NDArrayFactory::create('c', {arrays.size(), 5, 20}); + auto z = NDArrayFactory::create('c', {(Nd4jLong) arrays.size(), 5, 20}); context.setOutputArray(0, &z, false); context.setIArguments(&axis, 1); diff --git a/libnd4j/tests_cpu/layers_tests/testlayers.h b/libnd4j/tests_cpu/layers_tests/testlayers.h index 723b241bf..95814faae 100644 --- a/libnd4j/tests_cpu/layers_tests/testlayers.h +++ b/libnd4j/tests_cpu/layers_tests/testlayers.h @@ -36,5 +36,6 @@ #include #include #include +#include #endif //LIBND4J_TESTLAYERS_H