SWDEV-446399 - Fixed segfault in hipMemSetAccess

Change-Id: Ia1200d9bee03e8abade211287505f081e635ceec
This commit is contained in:
Ioannis Assiouras
2024-02-17 00:48:16 +00:00
parent 7bfee3481b
commit 1f6d416684
5 changed files with 9 additions and 10 deletions
+2 -2
View File
@@ -283,7 +283,7 @@ hipError_t hipMemSetAccess(void* ptr, size_t size, const hipMemAccessDesc* desc,
}
for (size_t desc_idx = 0; desc_idx < count; ++desc_idx) {
if (desc[desc_idx].location.id >= g_devices.size()) {
HIP_RETURN(hipErrorInvalidValue)
}
@@ -295,7 +295,7 @@ hipError_t hipMemSetAccess(void* ptr, size_t size, const hipMemAccessDesc* desc,
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)) {
if (!dev->devices()[0]->SetMemAccess(ptr, size, access_flags)) {
HIP_RETURN(hipErrorInvalidValue);
}
}