Skip iHipWaitActiveStreams if stream is NonBlocking

SWDEV-234484 Change-Id: I279bdc8485d8218e0aaa89d094c08a84f002a608 [ROCm/hip commit: f0b49d0560]
2020-05-06 10:36:53 -07:00
parent 9f0b428aa4
commit d6f0c4fc41
1 changed files with 3 additions and 1 deletions
@@ -84,7 +84,9 @@ amd::HostQueue* getQueue(hipStream_t stream) {
  } else {
    constexpr bool WaitNullStreamOnly = true;
    amd::HostQueue* queue = reinterpret_cast<hip::Stream*>(stream)->asHostQueue();
-    iHipWaitActiveStreams(queue, WaitNullStreamOnly);
+    if (!(reinterpret_cast<hip::Stream*>(stream)->Flags() & hipStreamNonBlocking)) {
+      iHipWaitActiveStreams(queue, WaitNullStreamOnly);
+    }
    return queue;
  }
 }