fix SpecialOpEliminater bug

9ef744db · wuyongkang · e83c5630 · 9ef744db
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

mindspore/ccsrc/optimizer/irpass.cc mindspore/ccsrc/optimizer/irpass.cc +1 -1

未找到文件。
--- a/mindspore/ccsrc/optimizer/irpass.cc
+++ b/mindspore/ccsrc/optimizer/irpass.cc
@@ -55,7 +55,7 @@ OptimizeIRPassLib::OptimizeIRPassLib() {
  special_op_eliminate_ =
    MakeSubstitution(std::make_shared<SpecialOpEliminater>(), "special_op_eliminate",
                     {prim::kPrimInsertGradientOf, prim::kPrimStopGradient, prim::kPrimHookBackward,
-                      prim::kPrimPrintShapeType, prim::kPrimGetRefKey, prim::kPrimMirror, prim::kPrimVirtualDiv});
+                      prim::kPrimPrintShapeType, prim::kPrimGetRefValue, prim::kPrimMirror, prim::kPrimVirtualDiv});
  zero_like_fill_zero_ =
    MakeSubstitution(std::make_shared<ZeroLikeFillZero>(), "zero_like_fill_zero", prim::kPrimZerosLike);
  adjust_all_reduce_mul_add_ =