wip: vregion: add DP module support for vregion.

Make DP modules use their own vregion for allocations.

TODO: abstract usage.

Signed-off-by: Liam Girdwood <liam.r.girdwood@linux.intel.com>

Author: lrgirdwo

src/schedule/zephyr_dp_schedule.c

@@ -411,7 +411,13 @@ static int scheduler_dp_task_free(void *data, struct task *task)
 #endif
 
 	/* free task stack */
+#if CONFIG_SOF_VREGIONS
+	struct vregion *vregion = module_get_vregion(pdata->mod);
+	vregion_free(vregion, (__sparse_force void *)pdata->p_stack);
+	ret = 0;
+#else
 	ret = user_stack_free((__sparse_force void *)pdata->p_stack);
+#endif
 	pdata->p_stack = NULL;
 
 	/* all other memory has been allocated as a single malloc, will be freed later by caller */
@@ -599,6 +605,7 @@ int scheduler_dp_task_init(struct task **task,
 {
 	void __sparse_cache *p_stack = NULL;
 	struct sys_heap *const user_heap = mod->dev->drv->user_heap;
+	struct vregion *vregion = module_get_vregion(mod);
 
 	/* memory allocation helper structure */
 	struct {
@@ -611,6 +618,21 @@ int scheduler_dp_task_init(struct task **task,
 	/* must be called on the same core the task will be binded to */
 	assert(cpu_get_id() == core);
 
+#if CONFIG_SOF_VREGIONS
+	//TODO: add check if vregion is in correct memory domain/coherent
+	task_memory = vregion_alloc_align(vregion, VREGION_MEM_TYPE_LIFETIME_SHARED,
+		sizeof(*task_memory), CONFIG_DCACHE_LINE_SIZE);
+	if (!task_memory) {
+		tr_err(&dp_tr, "vregion task memory alloc failed");
+		return -ENOMEM;
+	}
+	p_stack = vregion_alloc_align(vregion, VREGION_MEM_TYPE_LIFETIME,
+		stack_size, CONFIG_DCACHE_LINE_SIZE);
+	if (!p_stack) {
+		tr_err(&dp_tr, "vregion stack alloc failed");
+		return -ENOMEM;
+	}
+#else
 	/*
 	 * allocate memory
 	 * to avoid multiple malloc operations allocate all required memory as a single structure
@@ -632,7 +654,7 @@ int scheduler_dp_task_init(struct task **task,
 		ret = -ENOMEM;
 		goto err;
 	}
-
+#endif /* CONFIG_SOF_VREGION */
 	/* internal SOF task init */
 	ret = schedule_task_init(&task_memory->task, uid, SOF_SCHEDULE_DP, 0, ops->run,
 				 mod, core, options);