xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!16732 [MSLITE]rollback to avoid CI failed problem 

From: @probiotics_53
Reviewed-by: @zhang_xue_tong,@hangangqiang
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2021-05-22 14:36:59 +08:00 committed by Gitee
parent 9f0d6ec8da 011d0c1b99
commit 90991518bc
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/lite/src/lite_mindrt.h
View File

@ -42,12 +42,7 @@ class LiteOpActor : public OpActor<lite::Tensor> {
void RunOpData(OpDataPtr<Tensor> inputs, OpContext<Tensor> *context = nullptr) override { void RunOpData(OpDataPtr<Tensor> inputs, OpContext<Tensor> *context = nullptr) override {
auto op_uuid = context->sequential_num_; auto op_uuid = context->sequential_num_;
input_op_datas_[op_uuid].push_back(inputs); input_op_datas_[op_uuid].push_back(inputs);
inputs_data_.push_back(inputs->data_);
if (inputs->index_ >= static_cast<int>(inputs_data_.size()) || inputs->index_ < 0) {
return;
}
inputs_data_[inputs->index_] = inputs->data_;
if (input_op_datas_[op_uuid].size() < kernel_->in_tensors().size()) { if (input_op_datas_[op_uuid].size() < kernel_->in_tensors().size()) {
return; return;
} }
@ -80,6 +75,7 @@ class LiteOpActor : public OpActor<lite::Tensor> {
return; return;
} }
input_op_datas_.erase(op_uuid); input_op_datas_.erase(op_uuid);
inputs_data_.clear();
AsyncOutput(context); AsyncOutput(context);
BindThreads(static_cast<const lite::InnerContext *>(kernel_->Context())->thread_pool_, false, cpu_bind_mode); BindThreads(static_cast<const lite::InnerContext *>(kernel_->Context())->thread_pool_, false, cpu_bind_mode);
@ -91,7 +87,6 @@ class LiteOpActor : public OpActor<lite::Tensor> {
} }
int CastTensorData(Tensor *dst, Tensor *src); int CastTensorData(Tensor *dst, Tensor *src);
void Init() override { void Init() override {
inputs_data_.resize(kernel_->in_tensors().size());
auto ret = CompileArrow(); auto ret = CompileArrow();
if (ret != RET_OK) { if (ret != RET_OK) {
MS_LOG(ERROR) << "CompileArrow failed, name: " << kernel_->name(); MS_LOG(ERROR) << "CompileArrow failed, name: " << kernel_->name();