diff --git a/mindspore/ccsrc/backend/optimizer/pass/const_input_to_attr_registry.cc b/mindspore/ccsrc/backend/optimizer/pass/const_input_to_attr_registry.cc index 66d510b1b9..d4effc3c50 100644 --- a/mindspore/ccsrc/backend/optimizer/pass/const_input_to_attr_registry.cc +++ b/mindspore/ccsrc/backend/optimizer/pass/const_input_to_attr_registry.cc @@ -25,7 +25,6 @@ namespace mindspore { namespace opt { ConstInputToAttrInfoRegistry::ConstInputToAttrInfoRegistry() { Register(prim::kPrimCast->name(), {1}); - Register(prim::kPrimAvgPoolGrad->name(), {0}); Register(prim::kPrimAvgPoolGradVm->name(), {0}); Register(prim::kPrimConv2DBackpropInput->name(), {2}); Register(prim::kPrimConv2DBackpropFilter->name(), {2}); diff --git a/mindspore/nn/layer/normalization.py b/mindspore/nn/layer/normalization.py index 748ab7be3d..954d8d29fa 100644 --- a/mindspore/nn/layer/normalization.py +++ b/mindspore/nn/layer/normalization.py @@ -94,7 +94,7 @@ class _BatchNorm(Cell): SYNC_BN_GROUP_NAME = "sync_bn_group"+ str(i) management.create_group(SYNC_BN_GROUP_NAME, self.rank_list[i]) # for SyncBatchNorm - if self.process_groups != 0 and self.parallel_mode != context.ParallelMode.STAND_ALONE: + if self.process_groups != 0: self.rank_id = get_rank() self.rank_size = get_group_size() if self.process_groups is not None: