Browse Source

update ge/graph/build/stream_allocator.cc.

pull/475/head
cclworkaccount Gitee 4 years ago
parent
commit
6d3bd4dc40
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/build/stream_allocator.cc

+ 1
- 1
ge/graph/build/stream_allocator.cc View File

@@ -49,7 +49,7 @@ inline bool HasContinuousStreamLabel(const ge::OpDescPtr &op_desc, std::string &
}

bool IsHcclOp(const string &op_type) {
const set<string> hccl_op_types({ge::HCOMBROADCAST, ge::HCOMALLGATHER, ge::HCOMALLREDUCE, ge::HCOMREDUCESCATTER});
const set<string> hccl_op_types({ge::HCOMBROADCAST, ge::HCOMALLGATHER, ge::HCOMALLREDUCE, ge::HCOMREDUCESCATTER, ge::HCOMREDUCE});
return hccl_op_types.find(op_type) != hccl_op_types.end();
}
} // namespace


Loading…
Cancel
Save