fast-get: switch to module allocation context

fast-get has to choose which allocation method to use, it needs
access to the full module allocation context.

Signed-off-by: Guennadi Liakhovetski <guennadi.liakhovetski@linux.intel.com>

Author: lyakh

src/audio/module_adapter/module/generic.c

@@ -323,7 +323,7 @@ const void *z_impl_mod_fast_get(struct processing_module *mod, const void * cons
 	if (!container)
 		return NULL;
 
-	ptr = fast_get(res->alloc->heap, dram_ptr, size);
+	ptr = fast_get(res->alloc, dram_ptr, size);
 	if (!ptr) {
 		container_put(mod, container);
 		return NULL;
@@ -362,7 +362,7 @@ static int free_contents(struct processing_module *mod, struct module_resource *
 #else
 		mdom = NULL;
 #endif
-		fast_put(res->alloc->heap, mdom, container->sram_ptr);
+		fast_put(res->alloc, mdom, container->sram_ptr);
 		return 0;
 #endif
 	default:

src/include/sof/lib/fast-get.h

@@ -10,8 +10,8 @@
 
 #include <stddef.h>
 
-struct k_heap;
 struct k_mem_domain;
+struct mod_alloc_ctx;
 
 /*
  * When built for SOF, fast_get() and fast_put() are only needed when DRAM
@@ -25,14 +25,16 @@ struct k_mem_domain;
 #if (CONFIG_COLD_STORE_EXECUTE_DRAM &&					\
 	(CONFIG_LLEXT_TYPE_ELF_RELOCATABLE || !defined(LL_EXTENSION_BUILD))) || \
 	!CONFIG_SOF_FULL_ZEPHYR_APPLICATION
-const void *fast_get(struct k_heap *heap, const void * const dram_ptr, size_t size);
-void fast_put(struct k_heap *heap, struct k_mem_domain *mdom, const void *sram_ptr);
+const void *fast_get(struct mod_alloc_ctx *alloc, const void * const dram_ptr, size_t size);
+void fast_put(struct mod_alloc_ctx *alloc, struct k_mem_domain *mdom, const void *sram_ptr);
 #else
-static inline const void *fast_get(struct k_heap *heap, const void * const dram_ptr, size_t size)
+static inline const void *fast_get(struct mod_alloc_ctx *alloc, const void * const dram_ptr,
+				   size_t size)
 {
 	return dram_ptr;
 }
-static inline void fast_put(struct k_heap *heap, struct k_mem_domain *mdom, const void *sram_ptr) {}
+static inline void fast_put(struct mod_alloc_ctx *alloc, struct k_mem_domain *mdom,
+			    const void *sram_ptr) {}
 #endif
 
 #endif /* __SOF_LIB_FAST_GET_H__ */

zephyr/lib/fast-get.c

@@ -8,6 +8,7 @@
 #include <stdint.h>
 #include <errno.h>
 
+#include <sof/audio/component.h>
 #include <sof/lib/fast-get.h>
 #include <rtos/alloc.h>
 #include <rtos/cache.h>
@@ -125,8 +126,9 @@ static int fast_get_access_grant(struct k_mem_domain *mdom, void *addr, size_t s
 }
 #endif /* CONFIG_USERSPACE */
 
-const void *fast_get(struct k_heap *heap, const void *dram_ptr, size_t size)
+const void *fast_get(struct mod_alloc_ctx *alloc, const void *dram_ptr, size_t size)
 {
+	struct k_heap *heap = alloc ? alloc->heap : NULL;
 #if CONFIG_USERSPACE
 	bool current_is_userspace = thread_is_userspace(k_current_get());
 #endif
@@ -266,8 +268,9 @@ static struct sof_fast_get_entry *fast_put_find_entry(struct sof_fast_get_data *
 	return NULL;
 }
 
-void fast_put(struct k_heap *heap, struct k_mem_domain *mdom, const void *sram_ptr)
+void fast_put(struct mod_alloc_ctx *alloc, struct k_mem_domain *mdom, const void *sram_ptr)
 {
+	struct k_heap *heap = alloc ? alloc->heap : NULL;
 	struct sof_fast_get_data *data = &fast_get_data;
 	struct sof_fast_get_entry *entry;
 	k_spinlock_key_t key;