SWDEV-336550 - added nullptr check for svm

Change-Id: Icfd7dab1c7818958108385261542526d4545721d
Dieser Commit ist enthalten in:
pghafari
2022-05-20 22:14:18 -04:00
committet von Payam Ghafari
Ursprung 84f94fd134
Commit 2870b7ddd5
+7
Datei anzeigen
@@ -600,6 +600,13 @@ hipError_t hipMemPtrGetInfo(void *ptr, size_t *size) {
hipError_t hipHostFree(void* ptr) {
HIP_INIT_API(hipHostFree, ptr);
CHECK_STREAM_CAPTURE_SUPPORTED();
size_t offset = 0;
amd::Memory* memory_object = getMemoryObject(ptr, offset);
if (memory_object != nullptr) {
if (memory_object->getSvmPtr() == nullptr) {
return hipErrorInvalidValue;
}
}
HIP_RETURN(ihipFree(ptr));
}