fast_get: Enable buffer sharing when using module driver heap

Allow fast_get sram buffer sharing across multiple userspace module
instances when CONFIG_SOF_USERSPACE_USE_DRIVER_HEAP is enabled. The module
driver heap is shared by all instances of a given module, so allocated
buffers can safely be reused between them.

Simplify checking whether a calling thread runs in userspace by verifying
if the K_USER flag is set.

Signed-off-by: Adrian Warecki <adrian.warecki@intel.com>

Author: softwarecki

zephyr/lib/fast-get.c

@@ -150,7 +150,11 @@ const void *fast_get(struct k_heap *heap, const void *dram_ptr, size_t size)
 		alloc_align = PLATFORM_DCACHE_ALIGN;
 	}
 
-	if (size > FAST_GET_MAX_COPY_SIZE || !IS_ENABLED(CONFIG_USERSPACE))
+	if (size > FAST_GET_MAX_COPY_SIZE || !IS_ENABLED(CONFIG_USERSPACE) ||
+	    /* The module driver heap is shared by all instances of a given module.
+	     * Instances can share the allocated buffer.
+	     */
+	    IS_ENABLED(CONFIG_SOF_USERSPACE_USE_DRIVER_HEAP))
 		alloc_ptr = dram_ptr;
 	else
 		/* When userspace is enabled only share large buffers */
@@ -183,8 +187,8 @@ const void *fast_get(struct k_heap *heap, const void *dram_ptr, size_t size)
 		ret = entry->sram_ptr;
 
 #if CONFIG_USERSPACE
-		/* We only get there for large buffers */
-		if (k_current_get()->mem_domain_info.mem_domain->num_partitions > 1) {
+		/* We only get there for large buffers or module driver heap is in use */
+		if (k_current_get()->base.user_options & K_USER && size > FAST_GET_MAX_COPY_SIZE) {
 			/* A userspace thread makes the request */
 			if (k_current_get() != entry->thread &&
 			    !fast_get_domain_exists(k_current_get(), ret,