fast-get: add support for vregion allocations

When userspace modules use fast-get to access smaller buffers, their
copies should be allocated on module's vregion, if available, not on
the global SOF heap.

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

Author: lyakh

zephyr/lib/fast-get.c

@@ -10,6 +10,7 @@
 
 #include <sof/audio/module_adapter/module/generic.h>
 #include <sof/lib/fast-get.h>
+#include <sof/lib/vregion.h>
 #include <rtos/alloc.h>
 #include <rtos/cache.h>
 #include <rtos/kernel.h>
@@ -216,12 +217,12 @@ const void *fast_get(struct mod_alloc_ctx *alloc, const void *dram_ptr, size_t s
 		goto out;
 	}
 
-	/*
-	 * If a userspace threads is the first user to fast-get the buffer, an
-	 * SRAM copy will be allocated on its own heap, so it will have access
-	 * to it
-	 */
-	ret = sof_heap_alloc(alloc->heap, alloc_flags, alloc_size, alloc_align);
+	if (alloc->vreg && size <= FAST_GET_MAX_COPY_SIZE)
+		/* A userspace allocation, that won't be shared */
+		ret = vregion_alloc_align(alloc->vreg, VREGION_MEM_TYPE_INTERIM, alloc_size,
+					  alloc_align);
+	else
+		ret = sof_heap_alloc(alloc->heap, alloc_flags, alloc_size, alloc_align);
 	if (!ret)
 		goto out;
 
@@ -284,7 +285,10 @@ void fast_put(struct mod_alloc_ctx *alloc, struct k_mem_domain *mdom, const void
 
 	if (!entry->refcount) {
 		LOG_DBG("freeing buffer %p", sram_ptr);
-		sof_heap_free(alloc->heap, entry->sram_ptr);
+		if (alloc->vreg && entry->size <= FAST_GET_MAX_COPY_SIZE)
+			vregion_free(alloc->vreg, entry->sram_ptr);
+		else
+			sof_heap_free(alloc->heap, entry->sram_ptr);
 	}
 
 #if CONFIG_USERSPACE