|
|
@@ -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); |
|
|
|
} |
|
|
|