SWDEV-514950 - fix hipMallocManaged & hipMemAdvise when capturing stream (#413)

This commit is contained in:
Li, Todd tiantuo authored and GitHub committed 2025-05-28 13:48:48 +05:30
1 parent 24cdff6cce
commit d96016bcc5
1 file changed
+8
+8
View File
@@ -72,6 +72,10 @@ hipError_t hipMallocManaged(void** dev_ptr, size_t size, unsigned int flags) {
HIP_RETURN(hipErrorInvalidValue);
}
if (!hip::tls.capture_streams_.empty() || !g_captureStreams.empty()) {
HIP_RETURN(hipErrorStreamCaptureUnsupported);
}
HIP_RETURN(ihipMallocManaged(dev_ptr, size), *dev_ptr);
}
@@ -151,6 +155,10 @@ hipError_t hipMemAdvise(const void* dev_ptr, size_t count, hipMemoryAdvise advic
HIP_RETURN(hipErrorInvalidValue);
}
if (!hip::tls.capture_streams_.empty() || !g_captureStreams.empty()) {
HIP_RETURN(hipErrorStreamCaptureUnsupported);
}
size_t offset = 0;
amd::Memory* memObj = getMemoryObject(dev_ptr, offset);
if (memObj && count > (memObj->getSize() - offset)) {