| @@ -1634,8 +1634,18 @@ StreamActiveKind AscendStreamAssign::GetStreamActiveKind(const NotNull<KernelGra | |||||
| if (name == kSendOpName || name == kRecvOpName) { | if (name == kSendOpName || name == kRecvOpName) { | ||||
| continue; | continue; | ||||
| } | } | ||||
| auto stream = AnfAlgo::GetStreamId(cnode); | |||||
| auto it = hcom_stream_map_.find(stream); | |||||
| if (it != hcom_stream_map_.end()) { | |||||
| continue; | |||||
| } | |||||
| it = independent_stream_map_.find(stream); | |||||
| if (it != independent_stream_map_.end()) { | |||||
| continue; | |||||
| } | |||||
| pre_stream_id = AnfAlgo::GetStreamId(cnode); | |||||
| pre_stream_id = stream; | |||||
| break; | break; | ||||
| } | } | ||||
| @@ -1645,7 +1655,18 @@ StreamActiveKind AscendStreamAssign::GetStreamActiveKind(const NotNull<KernelGra | |||||
| continue; | continue; | ||||
| } | } | ||||
| next_stream_id = AnfAlgo::GetStreamId(cnode); | |||||
| auto stream = AnfAlgo::GetStreamId(cnode); | |||||
| auto it = hcom_stream_map_.find(stream); | |||||
| if (it != hcom_stream_map_.end()) { | |||||
| continue; | |||||
| } | |||||
| it = independent_stream_map_.find(stream); | |||||
| if (it != independent_stream_map_.end()) { | |||||
| continue; | |||||
| } | |||||
| next_stream_id = stream; | |||||
| break; | break; | ||||
| } | } | ||||