Browse Source

update ge/graph/load/new_model_manager/davinci_model.cc.

pull/475/head
王笑天 Gitee 4 years ago
parent
commit
2c03211509
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      ge/graph/load/new_model_manager/davinci_model.cc

+ 2
- 1
ge/graph/load/new_model_manager/davinci_model.cc View File

@@ -411,7 +411,8 @@ void DavinciModel::CheckHasHcomOp() {
(op_desc->GetType() == HCOMALLREDUCE) || (op_desc->GetType() == HCOMSEND) ||
(op_desc->GetType() == HCOMRECEIVE) || (op_desc->GetType() == HCOMREDUCESCATTER) ||
(op_desc->GetType() == HVDCALLBACKALLREDUCE) || (op_desc->GetType() == HVDCALLBACKALLGATHER) ||
(op_desc->GetType() == HVDCALLBACKBROADCAST) || (op_desc->GetType() == HVDWAIT)),
(op_desc->GetType() == HVDCALLBACKBROADCAST) || (op_desc->GetType() == HVDWAIT) ||
(op_desc->GetType() == HCOMREDUCE)),
uint32_t stream_id = static_cast<uint32_t>(op_desc->GetStreamId());
(void)hcom_streams_.emplace(stream_id); GELOGD("hcom stream: %u.", stream_id); continue);
}


Loading…
Cancel
Save