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

This commit is contained in:
Li, Todd tiantuo
2025-05-28 01:18:48 -07:00
committed by GitHub
parent 24cdff6cce
commit d96016bcc5
+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)) {