!46846 fix mirrorpad coredump on pynative

Merge pull request !46846 from 范吉斌/fix_pad
2022-12-16 03:27:02 +00:00 · 2022-12-16 03:27:02 +00:00 · b3f30e573b
parent 3797dad108 b473ca1fe1
commit b3f30e573b
1 changed files with 1 additions and 0 deletions
--- a/mindspore/core/abstract/ops/primitive_infer_map.cc
+++ b/mindspore/core/abstract/ops/primitive_infer_map.cc
@ -212,6 +212,7 @@ PrimShapeDependMap &GetHostDependsMap() {
                                         {kSparseSegmentSqrtnWithNumSegments, ShapeSet{3}},
                                         {kSparseMatrixTranspose, ShapeSet{0}},
                                         {kParameterizedTruncatedNormal, ShapeSet{0}},
+                                         {prim::kPrimMirrorPadGrad->name(), ShapeSet{1}},
                                         {kSparseToDense, ShapeSet{2}}};
  return host_depends;
 }