diff --git a/test/unittest/unittest_nntrainer_tensor_v2.cpp b/test/unittest/unittest_nntrainer_tensor_v2.cpp index 854c525308..de7d2d7935 100644 --- a/test/unittest/unittest_nntrainer_tensor_v2.cpp +++ b/test/unittest/unittest_nntrainer_tensor_v2.cpp @@ -861,6 +861,9 @@ TEST(nntrainer_Tensor, multiply_strided_06_p) { float *indata = input.getData(); ASSERT_NE(nullptr, indata); + float *data = output.getData(); + ASSERT_NE(nullptr, data); + float *outdata_beta = new float[(input.size())]; float *indata_mul = new float[(input.size())]; float *outdata = new float[(input.size())]; @@ -876,9 +879,6 @@ TEST(nntrainer_Tensor, multiply_strided_06_p) { input.multiply_strided(input, output, 10.0); - float *data = output.getData(); - ASSERT_NE(nullptr, data); - for (int i = 0; i < batch * height * width; ++i) { if (data[i] != outdata[i]) { status = ML_ERROR_RESULT_OUT_OF_RANGE;