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

!18957 [MS][LITE]fix bug of ir manager and tensorlist op infer 

Merge pull request !18957 from mengyuanli/master
This commit is contained in:
i-robot 2021-06-28 03:13:51 +00:00 committed by Gitee
parent 37e9acd372 e8a41a8509
commit 961918f24f
3 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/tensorlist_setitem_infer.c
View File

@ -59,10 +59,6 @@ int TensorListSetItemInferShape(const TensorC *const *inputs, size_t inputs_size
}
int index = ((int *)(get_index->data_))[0];
if (index < 0 || (index >= ((int)(input0->element_num_)) && index != 0)) {
return NNACL_ERR;
}
output0->max_elements_num_ = input0->max_elements_num_;
if (input0->element_num_ == 0 && input0->element_shape_size_ == 0 && index == 0) {

2
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/tensorlist_stack_infer.c
View File

@ -35,7 +35,7 @@ int TensorListStackInferShape(const TensorC *const *inputs, size_t inputs_size,
return NNACL_INFER_INVALID;
}
if (input0->element_num_ == 0) {
return NNACL_ERR;
return NNACL_INFER_INVALID;
}
const TensorC *ele_shape = inputs[1]; // element shape
if (ele_shape->data_ == NULL) {

2
mindspore/core/ir/manager.cc
View File

@ -845,7 +845,7 @@ void FVTotalComputer::RealRecompute() {
continue;
}
auto curr = fg;
while (curr != p) {
while (curr != nullptr && curr != p) {
fv_total_analysis_[curr][iter.first] = iter.second;
curr = manager->parent(curr);
}