AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-413997 - Fixing multiple device cases.

Browse Source 
Change-Id: I10ad3fbfca887e92cd81f68392fa1acf753cbd2b
This commit is contained in:
kjayapra-amd
2024-03-15 11:31:19 -04:00
committed by Karthik Jayaprakash
parent 7de7da4016
commit d52d16c8e6
8 changed files with 177 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_vm.cpp
+3 -3
View File
@@ -314,12 +314,12 @@ hipError_t hipMemUnmap(void* ptr, size_t size) {
HIP_RETURN(hipErrorInvalidValue);
}
amd::Memory* vaddr_mem_obj = amd::MemObjMap::FindVirtualMemObj(ptr);
if (vaddr_mem_obj == nullptr && vaddr_mem_obj->getSize() != size) {
amd::Memory* vaddr_sub_obj = amd::MemObjMap::FindMemObj(ptr);
if (vaddr_sub_obj == nullptr && vaddr_sub_obj->getSize() != size) {
HIP_RETURN(hipErrorInvalidValue);
}
amd::Memory* phys_mem_obj = vaddr_mem_obj->getUserData().phys_mem_obj;
amd::Memory* phys_mem_obj = vaddr_sub_obj->getUserData().phys_mem_obj;
if (phys_mem_obj == nullptr) {
HIP_RETURN(hipErrorInvalidValue);
}