Browse Source

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

pull/475/head
cclworkaccount Gitee 4 years ago
parent
commit
12bb35b0a5
1 changed files with 1 additions and 1 deletions
  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) {
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) {


Loading…
Cancel
Save