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

Change-Id: I2f1d3b229bc28571ce6c55f93f8ef73dda516b50

hipamd/src/hip_device_runtime.cpp

@@ -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);