From b8bbe5f02548b4a9dc664456d6209bf2ff726636 Mon Sep 17 00:00:00 2001
From: gukecai <gukecai@huawei.com>
Date: Mon, 26 Oct 2020 16:43:22 +0800
Subject: [PATCH] fix stream for memreuse

---
 .../device/ascend/ascend_stream_assign.cc     | 25 +++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

diff --git a/mindspore/ccsrc/runtime/device/ascend/ascend_stream_assign.cc b/mindspore/ccsrc/runtime/device/ascend/ascend_stream_assign.cc
index 50af13ab26..6b446b135d 100644
--- a/mindspore/ccsrc/runtime/device/ascend/ascend_stream_assign.cc
+++ b/mindspore/ccsrc/runtime/device/ascend/ascend_stream_assign.cc
@@ -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;
   }