Browse Source

!14255 [GraphKernel] open Assign fusion and fix OptimizeAssign

From: @lingyunli63
Reviewed-by: @gaoxiong1,@dylangeng
Signed-off-by: @dylangeng
pull/14255/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
15ed2daa31
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc
  2. +1
    -1
      mindspore/ccsrc/backend/optimizer/graph_kernel/optimize_assign.cc

+ 1
- 1
mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc View File

@@ -584,7 +584,7 @@ std::vector<PrimitivePtr> GetFusibleOpList() {
prim::kPrimExpandDims, prim::kPrimMul, prim::kPrimMinimum, prim::kPrimMaximum, prim::kPrimLog,
prim::kPrimPow, prim::kPrimSub, prim::kPrimRsqrt, prim::kPrimSqrt, prim::kPrimAddN,
prim::kPrimEqual, prim::kPrimReciprocal, prim::kPrimTanh, prim::kPrimReshape, prim::kPrimTranspose,
prim::kPrimCast, prim::kPrimRealDiv, prim::kPrimMatMul};
prim::kPrimCast, prim::kPrimRealDiv, prim::kPrimMatMul, prim::kPrimAssign};
#elif ENABLE_GPU
std::vector<PrimitivePtr> fusible_basic_ops = {
prim::kPrimAbs, prim::kPrimRound, prim::kPrimNeg, prim::kPrimExp, prim::kPrimAdd,


+ 1
- 1
mindspore/ccsrc/backend/optimizer/graph_kernel/optimize_assign.cc View File

@@ -146,7 +146,7 @@ void UpdateUsersOfGraphKernel(const FuncGraphPtr &func_graph, const AnfNodePtr &
// input, without Depend node.
// 2. If the `cnode` has another path to the getitem_user, it's unnecessary to add update_state and load node to
// keep exec_order.
if (!AnfAlgo::IsRealKernel(getitem_user) || HasPathToParamUser(cnode, getitem_user, getitem)) {
if (HasPathToParamUser(cnode, getitem_user, getitem)) {
mng->Replace(getitem, assign_to);
continue;
}


Loading…
Cancel
Save