Browse Source

rectify abstract if backend infer method has been converted to attr

pull/16015/head
lianliguang 4 years ago
parent
commit
9faa18060d
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      mindspore/ccsrc/backend/optimizer/common/helper.cc

+ 5
- 1
mindspore/ccsrc/backend/optimizer/common/helper.cc View File

@@ -933,7 +933,11 @@ AbstractBasePtr CppInferShape(const PrimitivePtr &prim, const AbstractBasePtrLis
auto ret_backend = prim_backend_eval_impl_map.find(prim);
if (ret_backend != prim_backend_eval_impl_map.end()) {
MS_EXCEPTION_IF_NULL(ret_backend->second.infer_shape_impl_);
return ret_backend->second.infer_shape_impl_(nullptr, prim, args_spec_list);
auto infer_spec_list = args_spec_list;
if (!ret_backend->second.in_white_list_) {
infer_spec_list = RectifyAbstract(prim, args_spec_list);
}
return ret_backend->second.infer_shape_impl_(nullptr, prim, infer_spec_list);
}
}
MS_LOG(EXCEPTION) << "Get infer shape function failed, primitive name:" << prim->name()


Loading…
Cancel
Save