Add _starpu_prio_deque_pop_this_task

src/sched_policies/prio_deque.h

@@ -16,6 +16,7 @@
 #ifndef __PRIO_DEQUE_H__
 #define __PRIO_DEQUE_H__
 #include <starpu.h>
+#include <starpu_scheduler.h>
 #include <core/task.h>
 
 
@@ -85,6 +86,24 @@ static inline struct starpu_task * _starpu_prio_deque_pop_task(struct _starpu_pr
 	return task;
 }
 
+static inline int _starpu_prio_deque_pop_this_task(struct _starpu_prio_deque *pdeque, int workerid, struct starpu_task *task)
+{
+	unsigned nimpl = 0;
+#ifdef STARPU_DEBUG
+	STARPU_ASSERT(&starpu_task_prio_list_ismember(&pdeque->list, task));
+#endif
+
+	if (workerid < 0 || starpu_worker_can_execute_task_first_impl(workerid, task, &nimpl))
+	{
+		starpu_task_set_implementation(task, nimpl);
+		starpu_task_prio_list_erase(&pdeque->list, task);
+		pdeque->ntasks--;
+		return 1;
+	}
+
+	return 0;
+}
+
 /* return a task that can be executed by workerid
  */
 struct starpu_task * _starpu_prio_deque_pop_task_for_worker(struct _starpu_prio_deque *, int workerid, int *skipped);