modified: ge/graph/optimize/mem_rw_conflict_optimize.cc

4 years ago · c599e8a9d0
parent 5f9e80b73e
commit c599e8a9d0
1 changed files with 1 additions and 1 deletions
--- a/ge/graph/optimize/mem_rw_conflict_optimize.cc
+++ b/ge/graph/optimize/mem_rw_conflict_optimize.cc
@ -234,7 +234,7 @@ InputRWType GetSingleNodeInputRWTypeByIndex(const Node &node, uint32_t index) {
    return InputRWType::kInvalidRWType;
  }
  if (op_desc->GetType() == HCOMALLREDUCE || op_desc->GetType() == HCOMALLGATHER
-      || op_desc->GetType() == HCOMREDUCESCATTER) {
+      || op_desc->GetType() == HCOMREDUCESCATTER || op_desc->GetType() == HCOMREDUCE) {
    return InputRWType::kScopeWriteable;
  }
  // check if it is ref input