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

SWDEV-499927 - Add nullchecks to hipMemCreate and GetAllocationGranularity

Browse Source 
Change-Id: I9bb879ce2c702e660ae5e6372cb27fdb8600566b


[ROCm/hipother commit: aea1990fe0]
This commit is contained in:
Marko Arandjelovic
2025-02-20 22:24:41 +05:30
parent 1f7400d7bf
commit 71418041e7
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
+12 -4
View File
@@ -2857,15 +2857,23 @@ inline static CUmemAccessDesc* hipMemAccessDescToCUmemAccessDesc(const hipMemAcc
inline static hipError_t hipMemGetAllocationGranularity(size_t* granularity,
const hipMemAllocationProp* prop,
hipMemAllocationGranularity_flags option) {
CUmemAllocationProp cuProp = hipMemAllocationPropToCUmemAllocationProp(prop);
return hipCUResultTohipError(cuMemGetAllocationGranularity(granularity, &cuProp, option));
if (prop == NULL) {
return hipCUResultTohipError(cuMemGetAllocationGranularity(granularity, NULL, option));
} else {
CUmemAllocationProp cuProp = hipMemAllocationPropToCUmemAllocationProp(prop);
return hipCUResultTohipError(cuMemGetAllocationGranularity(granularity, &cuProp, option));
}
}
inline static hipError_t hipMemCreate(hipMemGenericAllocationHandle_t* handle,
size_t size,
const hipMemAllocationProp* prop,
unsigned long long flags) {
CUmemAllocationProp cuProp = hipMemAllocationPropToCUmemAllocationProp(prop);
return hipCUResultTohipError(cuMemCreate(handle, size, &cuProp, flags));
if (prop == NULL) {
return hipCUResultTohipError(cuMemCreate(handle, size, NULL, flags));
} else {
CUmemAllocationProp cuProp = hipMemAllocationPropToCUmemAllocationProp(prop);
return hipCUResultTohipError(cuMemCreate(handle, size, &cuProp, flags));
}
}
inline static hipError_t hipMemRelease(hipMemGenericAllocationHandle_t handle) {
return hipCUResultTohipError(cuMemRelease(handle));