AI/rocm-systems
2
0
Vork 0
Code Kwesties Pull-aanvragen Actions Paketten Projecten Publicaties Wiki Activiteit

SWDEV-413997 - Fixing multiple access desc cases for hipMemSetAccess.

Bladeren bron 
Change-Id: Iec7d1207ec0a7136e4e74ef2e390f3347e753d40


[ROCm/clr commit: e226ae109b]
This commit is contained in:
kjayapra-amd
2023-10-19 10:31:13 -04:00
gecommit door Karthik Jayaprakash
bovenliggende 2da4764fc2
commit 87f90197f2
1 gewijzigde bestanden met toevoegingen van 10 en 7 verwijderingen
Download Patch-bestand Download Diff-bestand Expand all files Collapse all files
projects/clr/hipamd/src/hip_vm.cpp
+10 -7
Bestand weergeven
@@ -280,15 +280,18 @@ hipError_t hipMemSetAccess(void* ptr, size_t size, const hipMemAccessDesc* desc,
HIP_RETURN(hipErrorInvalidValue);
}
if (desc->location.id >= g_devices.size()) {
HIP_RETURN(hipErrorInvalidValue)
}
for (size_t desc_idx = 0; desc_idx < count; ++desc_idx) {
auto& dev = g_devices[desc->location.id];
amd::Device::VmmAccess access_flags = static_cast<amd::Device::VmmAccess>(desc->flags);
if (desc[desc_idx].location.id >= g_devices.size()) {
HIP_RETURN(hipErrorInvalidValue)
}
if (!dev->devices()[0]->SetMemAccess(ptr, size, access_flags, count)) {
HIP_RETURN(hipErrorInvalidValue);
auto& dev = g_devices[desc[desc_idx].location.id];
amd::Device::VmmAccess access_flags = static_cast<amd::Device::VmmAccess>(desc[desc_idx].flags);
if (!dev->devices()[0]->SetMemAccess(ptr, size, access_flags, count)) {
HIP_RETURN(hipErrorInvalidValue);
}
}
HIP_RETURN(hipSuccess);