diff --git a/hipamd/src/hip_device_runtime.cpp b/hipamd/src/hip_device_runtime.cpp
index 871a9420fc..6e8dfdb4f0 100644
--- a/hipamd/src/hip_device_runtime.cpp
+++ b/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);