p19647583
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

prepare data when getting inputs

This commit is contained in:
zhaizhiqiang 2021-04-13 19:33:12 +08:00
parent c9d9e1cf32
commit 9c3f29c23a
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/lite/src/cxx_api/model/model_impl.cc
View File

@ -213,6 +213,7 @@ std::vector<MSTensor> 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<MSTensor::Impl>(new (std::nothrow) MSTensor::Impl(inputs[i]));
if (impl == nullptr || impl->lite_tensor() == nullptr) {
MS_LOG(ERROR) << "Create tensor failed.";

4
mindspore/lite/tools/benchmark/benchmark.cc
View File

@ -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_);