|
@@ -0,0 +1,62 @@
|
|
|
+/* StarPU --- Runtime system for heterogeneous multicore architectures.
|
|
|
+ *
|
|
|
+ * Copyright (C) 2010-2012 Université de Bordeaux 1
|
|
|
+ * Copyright (C) 2010, 2011, 2012 Centre National de la Recherche Scientifique
|
|
|
+ *
|
|
|
+ * StarPU is free software; you can redistribute it and/or modify
|
|
|
+ * it under the terms of the GNU Lesser General Public License as published by
|
|
|
+ * the Free Software Foundation; either version 2.1 of the License, or (at
|
|
|
+ * your option) any later version.
|
|
|
+ *
|
|
|
+ * StarPU is distributed in the hope that it will be useful, but
|
|
|
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
|
|
+ *
|
|
|
+ * See the GNU Lesser General Public License in COPYING.LGPL for more details.
|
|
|
+ */
|
|
|
+
|
|
|
+#ifndef __STARPU_TASK_UTIL_H__
|
|
|
+#define __STARPU_TASK_UTIL_H__
|
|
|
+
|
|
|
+#include <stdio.h>
|
|
|
+#include <stdlib.h>
|
|
|
+#include <string.h>
|
|
|
+#include <assert.h>
|
|
|
+#include <starpu.h>
|
|
|
+
|
|
|
+#ifdef __cplusplus
|
|
|
+extern "C"
|
|
|
+{
|
|
|
+#endif
|
|
|
+
|
|
|
+/* This creates (and submits) an empty task that unlocks a tag once all its
|
|
|
+ * dependencies are fulfilled. */
|
|
|
+void starpu_create_sync_task(starpu_tag_t sync_tag, unsigned ndeps, starpu_tag_t *deps,
|
|
|
+ void (*callback)(void *), void *callback_arg);
|
|
|
+
|
|
|
+/* Constants used by the starpu_insert_task helper to determine the different types of argument */
|
|
|
+#define STARPU_VALUE (1<<4) /* Pointer to a constant value */
|
|
|
+#define STARPU_CALLBACK (1<<5) /* Callback function */
|
|
|
+#define STARPU_CALLBACK_WITH_ARG (1<<6) /* Callback function */
|
|
|
+#define STARPU_CALLBACK_ARG (1<<7) /* Argument of the callback function (of type void *) */
|
|
|
+#define STARPU_PRIORITY (1<<8) /* Priority associated to the task */
|
|
|
+#define STARPU_EXECUTE_ON_NODE (1<<9) /* Used by MPI to define which task is going to execute the codelet */
|
|
|
+#define STARPU_EXECUTE_ON_DATA (1<<10) /* Used by MPI to define which task is going to execute the codelet */
|
|
|
+#define STARPU_DATA_ARRAY (1<<11) /* Array of data handles */
|
|
|
+
|
|
|
+/* Wrapper to create a task. */
|
|
|
+int starpu_insert_task(struct starpu_codelet *cl, ...);
|
|
|
+
|
|
|
+/* Retrieve the arguments of type STARPU_VALUE associated to a task
|
|
|
+ * automatically created using starpu_insert_task. */
|
|
|
+void starpu_codelet_unpack_args(void *cl_arg, ...);
|
|
|
+
|
|
|
+/* Pack arguments of type STARPU_VALUE into a buffer which can be
|
|
|
+ * given to a codelet and later unpacked with starpu_codelet_unpack_args */
|
|
|
+void starpu_codelet_pack_args(char **arg_buffer, size_t *arg_buffer_size, ...);
|
|
|
+
|
|
|
+#ifdef __cplusplus
|
|
|
+}
|
|
|
+#endif
|
|
|
+
|
|
|
+#endif /* __STARPU_TASK_UTIL_H__ */
|