Browse Source

!15975 [GraphKernel]Fix compilation error

From: @dayschan
Reviewed-by: @gaoxiong1,@gaoxiong1,@xsmq
Signed-off-by: @xsmq
pull/15975/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
0f4898fd46
2 changed files with 2 additions and 4 deletions
  1. +1
    -2
      mindspore/ccsrc/backend/optimizer/graph_kernel/add_stitch_atomic_clean_gpu.cc
  2. +1
    -2
      mindspore/ccsrc/backend/optimizer/graph_kernel/add_stitch_atomic_clean_gpu.h

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

@@ -88,8 +88,7 @@ CNodePtr StitchAtomicCleanInsertter::CreateInplaceAssignNode(const FuncGraphPtr
return inplace_assign_node;
}

void StitchAtomicCleanInsertter::ProcessOriginCNode(const AnfNodePtr &composite_node, const AnfNodePtr &new_input,
const FuncGraphManagerPtr &mng) {
void StitchAtomicCleanInsertter::ProcessOriginCNode(const AnfNodePtr &composite_node, const AnfNodePtr &new_input) {
auto sub_graph = AnfAlgo::GetCNodeFuncGraphPtr(composite_node);
auto mng_sub = sub_graph->manager();
if (mng_sub == nullptr) {


+ 1
- 2
mindspore/ccsrc/backend/optimizer/graph_kernel/add_stitch_atomic_clean_gpu.h View File

@@ -38,8 +38,7 @@ class StitchAtomicCleanInsertter : public AtomicCleanInsertter {
CNodePtr CreateInplaceAssignNode(const FuncGraphPtr &sub_graph, const AnfNodePtr &new_parameter);
std::vector<std::pair<AnfNodePtr, int>> FindInnerCNodeUsers(const AnfNodePtr &inner_node,
const CNodePtr &target) const;
void ProcessOriginCNode(const AnfNodePtr &composite_node, const AnfNodePtr &new_input,
const FuncGraphManagerPtr &mng) override;
void ProcessOriginCNode(const AnfNodePtr &composite_node, const AnfNodePtr &new_input) override;
bool IsStitchWithAtomic(const AnfNodePtr &anf_node);

AnfNodePtr stitch_node_{nullptr};


Loading…
Cancel
Save