Browse Source

fix AvgPoolGrad in GPU-Pynative

tags/v1.2.0-rc1
yuchaojie 5 years ago
parent
commit
67a008d5d4
2 changed files with 1 additions and 2 deletions
  1. +0
    -1
      mindspore/ccsrc/backend/optimizer/pass/const_input_to_attr_registry.cc
  2. +1
    -1
      mindspore/nn/layer/normalization.py

+ 0
- 1
mindspore/ccsrc/backend/optimizer/pass/const_input_to_attr_registry.cc View File

@@ -25,7 +25,6 @@ namespace mindspore {
namespace opt { namespace opt {
ConstInputToAttrInfoRegistry::ConstInputToAttrInfoRegistry() { ConstInputToAttrInfoRegistry::ConstInputToAttrInfoRegistry() {
Register(prim::kPrimCast->name(), {1}); Register(prim::kPrimCast->name(), {1});
Register(prim::kPrimAvgPoolGrad->name(), {0});
Register(prim::kPrimAvgPoolGradVm->name(), {0}); Register(prim::kPrimAvgPoolGradVm->name(), {0});
Register(prim::kPrimConv2DBackpropInput->name(), {2}); Register(prim::kPrimConv2DBackpropInput->name(), {2});
Register(prim::kPrimConv2DBackpropFilter->name(), {2}); Register(prim::kPrimConv2DBackpropFilter->name(), {2});


+ 1
- 1
mindspore/nn/layer/normalization.py View File

@@ -94,7 +94,7 @@ class _BatchNorm(Cell):
SYNC_BN_GROUP_NAME = "sync_bn_group"+ str(i) SYNC_BN_GROUP_NAME = "sync_bn_group"+ str(i)
management.create_group(SYNC_BN_GROUP_NAME, self.rank_list[i]) management.create_group(SYNC_BN_GROUP_NAME, self.rank_list[i])
# for SyncBatchNorm # 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_id = get_rank()
self.rank_size = get_group_size() self.rank_size = get_group_size()
if self.process_groups is not None: if self.process_groups is not None:


Loading…
Cancel
Save