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

SWDEV-446219 - Validate option value.

Browse Source 
Change-Id: Ifccdddfc8ac71585f6ba045578e5bbb7e3cd1649
This commit is contained in:
Jaydeep Patel
2024-02-16 06:13:10 +00:00
committed by Jaydeepkumar Patel
parent 4feb1f9337
commit 66f6a3e255
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_vm.cpp
+3 -1
View File
@@ -168,7 +168,9 @@ hipError_t hipMemGetAllocationGranularity(size_t* granularity, const hipMemAlloc
HIP_INIT_API(hipMemGetAllocationGranularity, granularity, prop, option);
if (granularity == nullptr || prop == nullptr || prop->type != hipMemAllocationTypePinned ||
prop->location.type != hipMemLocationTypeDevice || prop->location.id >= g_devices.size()) {
prop->location.type != hipMemLocationTypeDevice || prop->location.id >= g_devices.size() ||
(option != hipMemAllocationGranularityMinimum &&
option != hipMemAllocationGranularityRecommended)) {
HIP_RETURN(hipErrorInvalidValue);
}