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

SWDEV-333581 - Pass free and/or total size_t as nullptr.

Browse Source 
Change-Id: Ib83467a351b6df22ef641c0e82284c7e0fac32e3
This commit is contained in:
Jaydeep Patel
2022-04-21 05:56:28 +00:00
committed by Jaydeepkumar Patel
parent e8194dca76
commit a527c241a2
1 changed files with 18 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_memory.cpp
+18 -9
View File
@@ -661,18 +661,27 @@ hipError_t hipMemGetAddressRange(hipDeviceptr_t* pbase, size_t* psize, hipDevice
hipError_t hipMemGetInfo(size_t* free, size_t* total) {
HIP_INIT_API(hipMemGetInfo, free, total);
size_t freeMemory[2];
amd::Device* device = hip::getCurrentDevice()->devices()[0];
if(device == nullptr) {
HIP_RETURN(hipErrorInvalidDevice);
}
if(!device->globalFreeMemory(freeMemory)) {
if (free == nullptr && total == nullptr) {
HIP_RETURN(hipErrorInvalidValue);
}
*free = freeMemory[0] * Ki;
*total = device->info().globalMemSize_;
size_t freeMemory[2];
amd::Device* device = hip::getCurrentDevice()->devices()[0];
if (device == nullptr) {
HIP_RETURN(hipErrorInvalidDevice);
}
if (!device->globalFreeMemory(freeMemory)) {
HIP_RETURN(hipErrorInvalidValue);
}
if (free != nullptr) {
*free = freeMemory[0] * Ki;
}
if (total != nullptr) {
*total = device->info().globalMemSize_;
}
HIP_RETURN(hipSuccess);
}