SWDEV-468133 - Fixed hipDeviceGetLimit for hipLimitMallocHeapSize
Change-Id: I91bede414ebe46831509cbd24ffb53cf129d6a40
This commit is contained in:
@@ -514,9 +514,7 @@ hipError_t hipDeviceGetLimit(size_t* pValue, hipLimit_t limit) {
|
||||
|
||||
switch (limit) {
|
||||
case hipLimitMallocHeapSize:
|
||||
hipDeviceProp_tR0600 prop;
|
||||
HIP_RETURN_ONFAIL(ihipGetDeviceProperties(&prop, ihipGetDevice()));
|
||||
*pValue = prop.totalGlobalMem;
|
||||
*pValue = hip::getCurrentDevice()->devices()[0]->InitialHeapSize();
|
||||
break;
|
||||
case hipLimitStackSize:
|
||||
*pValue = hip::getCurrentDevice()->devices()[0]->StackSize();
|
||||
|
||||
Reference in New Issue
Block a user