hummingbird
/
graphengine
Not watched
2
0
Fork 0
Code
Releases 13 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

update ge/graph/manager/util/hcom_util.cc.

pull/475/head
cclworkaccount Gitee 4 years ago
parent
431e79e744
commit
12bb35b0a5
1 changed files with 1 additions and 1 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      ge/graph/manager/util/hcom_util.cc

+ 1
- 1
ge/graph/manager/util/hcom_util.cc View File

@@ -281,7 +281,7 @@ Status HcomOmeUtil::GetAllRootId(const ge::ConstOpDescPtr &op_desc,


bool HcomOmeUtil::IsHCOMOp(const string &op_type) { bool HcomOmeUtil::IsHCOMOp(const string &op_type) {
return (op_type == HCOMALLREDUCE) || (op_type == HCOMALLGATHER) || (op_type == HCOMBROADCAST) || return (op_type == HCOMALLREDUCE) || (op_type == HCOMALLGATHER) || (op_type == HCOMBROADCAST) ||
(op_type == HCOMSEND) || (op_type == HCOMRECEIVE) || (op_type == HCOMREDUCESCATTER || op_desc->GetType() == HCOMREDUCE);
(op_type == HCOMSEND) || (op_type == HCOMRECEIVE) || (op_type == HCOMREDUCESCATTER) || (op_type == HCOMREDUCESCATTER);
} }


bool HcomOmeUtil::IsHorovodOp(const string &op_type) { bool HcomOmeUtil::IsHorovodOp(const string &op_type) {


Write Preview
Loading…
Cancel
Save