Browse Source

!2667 Fix SpecialOpEliminater bug

Merge pull request !2667 from Kang/optimization
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
07cd3efaec
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/optimizer/irpass.cc

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

@@ -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_ =


Loading…
Cancel
Save