rocr/aie: Correct reporting of dev heap size

Storing the correct dev heap size in the memory region.

Change-Id: I14b053330c187da1d7d0213256625e50795b9902

runtime/hsa-runtime/core/driver/xdna/amd_xdna_driver.cpp

@@ -84,6 +84,10 @@ hsa_status_t XdnaDriver::DiscoverDriver() {
   return HSA_STATUS_ERROR;
 }
 
+uint64_t XdnaDriver::GetDevHeapByteSize() {
+  return dev_heap_size;
+}
+
 hsa_status_t XdnaDriver::Init() { return InitDeviceHeap(); }
 
 hsa_status_t XdnaDriver::QueryKernelModeDriver(core::DriverQuery query) {