SWDEV-492768 - Match hipStreamAddCallback capture behavior with nvidia
Change-Id: I7a084d8eeffe8b5095f7eb9969a565a40e76bb4b
This commit is contained in:
@@ -609,6 +609,18 @@ hipError_t hipStreamAddCallback_common(hipStream_t stream, hipStreamCallback_t c
|
|||||||
if (callback == nullptr || flags != 0) {
|
if (callback == nullptr || flags != 0) {
|
||||||
return hipErrorInvalidValue;
|
return hipErrorInvalidValue;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (stream != nullptr) {
|
||||||
|
// Not supported while stream is capturing
|
||||||
|
if (hip::Stream::StreamCaptureOngoing(stream) == true) {
|
||||||
|
HIP_RETURN(hipErrorStreamCaptureUnsupported);
|
||||||
|
}
|
||||||
|
} else if (Stream::StreamCaptureBlocking() == true) {
|
||||||
|
// If any of the blocking streams is capturing, return error for implicit capture and invalidate
|
||||||
|
// capture for all capturing streams
|
||||||
|
CHECK_STREAM_CAPTURING();
|
||||||
|
}
|
||||||
|
|
||||||
StreamCallback* cbo = new StreamAddCallback(stream, callback, userData);
|
StreamCallback* cbo = new StreamAddCallback(stream, callback, userData);
|
||||||
return streamCallback_common(stream, cbo, userData);
|
return streamCallback_common(stream, cbo, userData);
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user