diff --git a/mindspore/lite/src/cxx_api/model/model_impl.cc b/mindspore/lite/src/cxx_api/model/model_impl.cc index e01b9f6d08..1ced137344 100644 --- a/mindspore/lite/src/cxx_api/model/model_impl.cc +++ b/mindspore/lite/src/cxx_api/model/model_impl.cc @@ -213,6 +213,7 @@ std::vector ModelImpl::GetInputs() { } res.resize(inputs.size()); for (size_t i = 0; i < inputs.size(); i++) { + inputs[i]->MutableData(); // prepare data auto impl = std::shared_ptr(new (std::nothrow) MSTensor::Impl(inputs[i])); if (impl == nullptr || impl->lite_tensor() == nullptr) { MS_LOG(ERROR) << "Create tensor failed."; diff --git a/mindspore/lite/tools/benchmark/benchmark.cc b/mindspore/lite/tools/benchmark/benchmark.cc index 8eabdf1cb4..b5e9d0c233 100644 --- a/mindspore/lite/tools/benchmark/benchmark.cc +++ b/mindspore/lite/tools/benchmark/benchmark.cc @@ -410,6 +410,10 @@ int Benchmark::MarkPerformance() { uint64_t time_avg = 0; for (int i = 0; i < flags_->loop_count_; i++) { + auto inputs = session_->GetInputs(); + for (auto tensor : inputs) { + tensor->MutableData(); // prepare data + } session_->BindThread(true); auto start = GetTimeUs(); auto status = session_->RunGraph(before_call_back_, after_call_back_);