From 539041b78374b06597f3ea4a35647c514c605e7e Mon Sep 17 00:00:00 2001 From: guohongzilong <2713219276@qq.com> Date: Thu, 6 Aug 2020 15:02:25 +0800 Subject: [PATCH] fix nasnet model bug --- mindspore/lite/src/runtime/kernel/arm/fp32/split.cc | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/split.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/split.cc index 2d064989c4..2b8082ef93 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp32/split.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp32/split.cc @@ -36,11 +36,7 @@ int SplitCPUKernel::Init() { return RET_OK; } SplitBaseCPUKernel::Init(); - auto in_tensor = inputs_.front(); - input_ptr_ = reinterpret_cast(in_tensor->Data()); - for (int i = 0; i < param->num_split_; i++) { - output_ptr_.push_back(reinterpret_cast(outputs_.at(i)->Data())); - } + output_ptr_.resize(param->num_split_); return RET_OK; } @@ -73,6 +69,11 @@ int SplitRun(int task_id, LiteParallelGroupEnv *penv, void *cdata) { int SplitCPUKernel::Run() { auto ret = Prepare(); + auto in_tensor = inputs_.front(); + input_ptr_ = reinterpret_cast(in_tensor->Data()); + for (int i = 0; i < output_ptr_.size(); i++) { + output_ptr_[i] = reinterpret_cast(outputs_.at(i)->Data()); + } if (ret != RET_OK) { MS_LOG(ERROR) << "Prepare failed."; return RET_ERROR;