zephyr: alloc: ace: calculate L3 heap size based on actual IMR size

Updates the L3 heap management to dynamically calculate heap size based on the actual IMR size reported by hardware registers instead of using hardcoded values.

Only initializes the L3 heap when the IMR is actually available and being used, as determined by the ace_imr_used() function, improving robustness by preventing the initialization of unavailable memory regions.

This change makes the L3 heap allocation more flexible and adaptable to different hardware configurations.

Signed-off-by: Adrian Bonislawski <adrian.bonislawski@intel.com>

Author: abonislawski

zephyr/lib/alloc.c

@@ -19,6 +19,10 @@
 #include <rtos/symbol.h>
 #include <rtos/wait.h>
 
+#if CONFIG_L3_HEAP && CONFIG_MMU
+#include <kernel_arch_interface.h>
+#endif
+
 #if CONFIG_VIRTUAL_HEAP
 #include <sof/lib/regions_mm.h>
 #include <zephyr/drivers/mm/mm_drv_intel_adsp_mtl_tlb.h>
@@ -158,7 +162,7 @@ static inline size_t get_l3_heap_size(void)
 	  * - IMR base address
 	  * - actual IMR heap start
 	  */
-	return ROUND_DOWN(IMR_L3_HEAP_SIZE, L3_MEM_PAGE_SIZE);
+	return ROUND_DOWN((ace_imr_get_mem_size() - (IMR_L3_HEAP_BASE - L3_MEM_BASE_ADDR)), L3_MEM_PAGE_SIZE);
 }
 
 void l3_heap_save(void)
@@ -557,11 +561,16 @@ static int heap_init(void)
 	sys_heap_init(&sof_heap.heap, heapmem, HEAPMEM_SIZE);
 
 #if CONFIG_L3_HEAP
-	if (l3_heap_copy.heap.heap)
+	if (l3_heap_copy.heap.heap) {
 		l3_heap = l3_heap_copy;
-	else
-		sys_heap_init(&l3_heap.heap, UINT_TO_POINTER(get_l3_heap_start()),
-			      get_l3_heap_size());
+	} else if (ace_imr_used()) {
+#if CONFIG_MMU
+		uintptr_t va = POINTER_TO_UINT(sys_cache_cached_ptr_get(UINT_TO_POINTER(get_l3_heap_start())));
+
+		arch_mem_map(UINT_TO_POINTER(get_l3_heap_start()), va, get_l3_heap_size(), K_MEM_PERM_RW | K_MEM_CACHE_WB);
+#endif
+		sys_heap_init(&l3_heap.heap, UINT_TO_POINTER(get_l3_heap_start()), get_l3_heap_size());
+	}
 #endif
 
 	return 0;