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

[ROCm/clr commit: d96016bcc5]
This commit is contained in:
Li, Todd tiantuo authored and GitHub committed 2025-05-28 13:48:48 +05:30
1 parent e5792ebe7d
commit 306a7bb5a1
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)) {