Browse Source

!16015 revert abstract when backend do infer

From: @lianliguang
Reviewed-by: @jjfeing,@zhoufeng54
Signed-off-by: @zhoufeng54
pull/16015/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
4957ae30be
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