!12917 fix AvgPoolGrad in GPU-Pynative

From: @yuchaojie
Reviewed-by: @zhoufeng54,@jjfeing
Signed-off-by: @jjfeing
pull/12917/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit d78f1786c5

@ -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});

@ -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:

Loading…
Cancel
Save