SWDEV-347592 - validate hipMemAllocationType & hipMemLocationType

Change-Id: Ia8c4d3edc7ad96efa429b3530a118199d4f92902


[ROCm/clr commit: 0fed29f77c]
Tá an tiomantas seo le fáil i:
Jaydeep Patel
2022-07-25 15:32:59 +00:00
tiomanta ag Jaydeepkumar Patel
tuismitheoir ab694c23fb
tiomantas f4642fc0d4
+14
Féach ar an gComhad
@@ -171,6 +171,20 @@ hipError_t hipMemPoolCreate(hipMemPool_t* mem_pool, const hipMemPoolProps* pool_
if (mem_pool == nullptr) {
HIP_RETURN(hipErrorInvalidValue);
}
// validate hipMemAllocationType value
if (pool_props->allocType != hipMemAllocationTypeInvalid &&
pool_props->allocType != hipMemAllocationTypePinned &&
pool_props->allocType != hipMemAllocationTypeMax) {
HIP_RETURN(hipErrorInvalidValue);
}
// validate hipMemLocationType value
if (pool_props->location.type != hipMemLocationTypeInvalid &&
pool_props->location.type != hipMemLocationTypeDevice) {
HIP_RETURN(hipErrorInvalidValue);
}
// Make sure the pool creation occurs on a valid device
if ((pool_props->location.type != hipMemLocationTypeDevice) ||
(pool_props->location.id >= g_devices.size())) {