|
|
|
@@ -1634,8 +1634,18 @@ StreamActiveKind AscendStreamAssign::GetStreamActiveKind(const NotNull<KernelGra |
|
|
|
if (name == kSendOpName || name == kRecvOpName) { |
|
|
|
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; |
|
|
|
} |
|
|
|
|
|
|
|
@@ -1645,7 +1655,18 @@ StreamActiveKind AscendStreamAssign::GetStreamActiveKind(const NotNull<KernelGra |
|
|
|
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; |
|
|
|
} |
|
|
|
|
|
|
|
|