From 9faa18060deca4e292c5eb720b3de72c7d5bf1fe Mon Sep 17 00:00:00 2001 From: lianliguang Date: Thu, 6 May 2021 17:28:57 +0800 Subject: [PATCH] rectify abstract if backend infer method has been converted to attr --- mindspore/ccsrc/backend/optimizer/common/helper.cc | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/backend/optimizer/common/helper.cc b/mindspore/ccsrc/backend/optimizer/common/helper.cc index c215b10b7c..f83387d46c 100644 --- a/mindspore/ccsrc/backend/optimizer/common/helper.cc +++ b/mindspore/ccsrc/backend/optimizer/common/helper.cc @@ -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()