Browse Source

update ge/hybrid/node_executor/hccl/hccl_node_executor.cc.

pull/475/head
cclworkaccount Gitee 4 years ago
parent
commit
b4b4432128
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/hybrid/node_executor/hccl/hccl_node_executor.cc

+ 1
- 1
ge/hybrid/node_executor/hccl/hccl_node_executor.cc View File

@@ -83,7 +83,7 @@ Status HcclNodeTask::ExecuteAsync(TaskContext &context, std::function<void()> do
op_info.dataType = iter->second; op_info.dataType = iter->second;
HcclReduceOp op_type = HCCL_REDUCE_SUM; HcclReduceOp op_type = HCCL_REDUCE_SUM;
if (op_desc->GetType() == HCOMALLREDUCE || op_desc->GetType() == HCOMREDUCESCATTER || if (op_desc->GetType() == HCOMALLREDUCE || op_desc->GetType() == HCOMREDUCESCATTER ||
op_desc->GetType() == HVDCALLBACKALLREDUCE) {
op_desc->GetType() == HVDCALLBACKALLREDUCE || op_desc->GetType() == HCOMREDUCE) {
GE_CHK_STATUS_RET(HcomOmeUtil::GetHcclOperationType(op_desc, op_type), "GetHcclOperationType failed"); GE_CHK_STATUS_RET(HcomOmeUtil::GetHcclOperationType(op_desc, op_type), "GetHcclOperationType failed");
op_info.opType = op_type; op_info.opType = op_type;
} }


Loading…
Cancel
Save