AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-326798 - [ABI Break] Fixes null stream sync behavior

Browse Source 
Change-Id: I84ad21f61779145c198dc68bf0fe02a57d34bf64
This commit is contained in:
Sourabh Betigeri
2023-04-26 20:27:11 +00:00
committed by Rakesh Roy
parent 15ba5c0004
commit 3f88fe850b
3 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_context.cpp
+2 -2
View File
@@ -100,12 +100,12 @@ void setCurrentDevice(unsigned int index) {
amd::Os::setPreferredNumaNode(preferredNumaNode);
}
hip::Stream* getStream(hipStream_t stream) {
hip::Stream* getStream(hipStream_t stream, bool wait) {
if (stream == nullptr) {
return getNullStream();
} else {
hip::Stream* hip_stream = reinterpret_cast<hip::Stream*>(stream);
if (!(hip_stream->Flags() & hipStreamNonBlocking)) {
if (wait && !(hip_stream->Flags() & hipStreamNonBlocking)) {
constexpr bool WaitNullStreamOnly = true;
iHipWaitActiveStreams(hip_stream, WaitNullStreamOnly);
}