diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/space_to_depth.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/space_to_depth.cc index 6b0819798111e0e5e0d12707e44ca1fc604664e6..3ecb34a6a6030d391471eb2a861813928fa84570 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp32/space_to_depth.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp32/space_to_depth.cc @@ -77,6 +77,8 @@ int SpaceToDepthRun(int task_id, LiteParallelGroupEnv *penv, void *cdata) { } int SpaceToDepthCPUKernel::Run() { + input_ptr_ = reinterpret_cast(inputs_[0]->Data()); + output_ptr_ = reinterpret_cast(outputs_[0]->Data()); if (inputs_[0]->GetFormat() == schema::Format_NHWC) { int ret = LiteBackendParallelLaunch(SpaceToDepthRun, this, thread_h_num_); if (ret != RET_OK) { diff --git a/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/space_to_batch_fp32_tests.cc b/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/space_to_batch_fp32_tests.cc index eafafe801193efa20644317254f736b8e40f864f..7445f0a75e01a9849629ef5b30b1ec9267876dc0 100644 --- a/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/space_to_batch_fp32_tests.cc +++ b/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/space_to_batch_fp32_tests.cc @@ -158,6 +158,8 @@ TEST_F(SpaceToBatchTestFp32, SpaceToBatchTest3) { } std::cout << "\n"; CompareOutputData(output.data(), expect_out, out_size, 0.000001); + input_tensor.SetData(nullptr); + output_tensor.SetData(nullptr); } } // namespace mindspore diff --git a/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/space_to_depth_fp32_tests.cc b/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/space_to_depth_fp32_tests.cc index ba48fa11c0c3babe13c5742023a84ba05541cb0c..2c25eeca0bb410d940dce9b0599e22a8c29e24df 100644 --- a/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/space_to_depth_fp32_tests.cc +++ b/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/space_to_depth_fp32_tests.cc @@ -56,7 +56,7 @@ TEST_F(SpaceToDepthTestFp32, SpaceToDepthTest2) { input_tensor.SetFormat(schema::Format_NHWC); input_tensor.set_data_type(kNumberTypeFloat32); std::vector inputs_tensor; - inputs_tensor.emplace_back(&input_tensor); + inputs_tensor.push_back(&input_tensor); const int out_size = 16; float expect_out[16] = {1, 2, 10, 20, 5, 6, 3, 8, 18, 10, 11, 55, 3, 4, 15, 25}; @@ -68,7 +68,7 @@ TEST_F(SpaceToDepthTestFp32, SpaceToDepthTest2) { output_tensor.SetFormat(schema::Format_NHWC); output_tensor.set_data_type(kNumberTypeFloat32); std::vector outputs_tensor; - outputs_tensor.emplace_back(&output_tensor); + outputs_tensor.push_back(&output_tensor); SpaceToDepthParameter op_param; op_param.op_parameter_.type_ = schema::PrimitiveType_SpaceToDepth;