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

SWDEV-364576 - use hipLimitMallocHeapSize to set initial heap buffer size

Browse Source 
Change-Id: I2f1d3b229bc28571ce6c55f93f8ef73dda516b50
This commit is contained in:
Anusha GodavarthySurya
2023-01-05 20:20:08 +00:00
committed by Anusha Godavarthy Surya
parent 3e54bad5b3
commit 33eef6d95e
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_device_runtime.cpp
+5
View File
@@ -489,6 +489,11 @@ hipError_t hipDeviceSetLimit ( hipLimit_t limit, size_t value ) {
HIP_RETURN(hipErrorInvalidValue);
}
break;
case hipLimitMallocHeapSize:
if (!hip::getCurrentDevice()->devices()[0]->UpdateInitialHeapSize(value)) {
HIP_RETURN(hipErrorInvalidValue);
}
break;
default:
LogPrintfError("UnsupportedLimit = %d is passed", limit);
HIP_RETURN(hipErrorUnsupportedLimit);