|
@@ -0,0 +1,143 @@
|
|
|
+/*
|
|
|
+ * StarPU
|
|
|
+ * Copyright (C) INRIA 2008-2009 (see AUTHORS file)
|
|
|
+ *
|
|
|
+ * This program 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.
|
|
|
+ *
|
|
|
+ * This program 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.
|
|
|
+ */
|
|
|
+
|
|
|
+#include <pthread.h>
|
|
|
+#include <stdio.h>
|
|
|
+#include <unistd.h>
|
|
|
+
|
|
|
+#include <starpu.h>
|
|
|
+
|
|
|
+static void *dummy_func(void *arg __attribute__ ((unused)))
|
|
|
+{
|
|
|
+ return NULL;
|
|
|
+}
|
|
|
+
|
|
|
+static starpu_codelet dummy_codelet =
|
|
|
+{
|
|
|
+ .where = ANY,
|
|
|
+ .core_func = dummy_func,
|
|
|
+ .cublas_func = dummy_func,
|
|
|
+ .model = NULL,
|
|
|
+ .nbuffers = 0
|
|
|
+};
|
|
|
+
|
|
|
+static void callback(void *tag)
|
|
|
+{
|
|
|
+ fflush(stderr);
|
|
|
+ fprintf(stderr, "Callback for tag %p\n", tag);
|
|
|
+ fflush(stderr);
|
|
|
+}
|
|
|
+
|
|
|
+static struct starpu_task *create_dummy_task(starpu_tag_t tag)
|
|
|
+{
|
|
|
+ struct starpu_task *task = starpu_task_create();
|
|
|
+
|
|
|
+ task->cl = &dummy_codelet;
|
|
|
+ task->cl_arg = NULL;
|
|
|
+ task->callback_func = callback;
|
|
|
+ task->callback_arg = tag;
|
|
|
+
|
|
|
+ task->use_tag = 1;
|
|
|
+ task->tag_id = tag;
|
|
|
+
|
|
|
+ task->cleanup = 0;
|
|
|
+
|
|
|
+ return task;
|
|
|
+}
|
|
|
+
|
|
|
+#define tagA 0x42
|
|
|
+#define tagB 0x12300
|
|
|
+
|
|
|
+#define tagC 0x32
|
|
|
+#define tagD 0x52
|
|
|
+#define tagE 0x19999
|
|
|
+#define tagF 0x2312
|
|
|
+#define tagG 0x1985
|
|
|
+
|
|
|
+#define tagH 0x32234
|
|
|
+#define tagI 0x5234
|
|
|
+#define tagJ 0x199
|
|
|
+#define tagK 0x231234
|
|
|
+#define tagL 0x2345
|
|
|
+
|
|
|
+
|
|
|
+int main(int argc, char **argv)
|
|
|
+{
|
|
|
+ starpu_init(NULL);
|
|
|
+
|
|
|
+ fprintf(stderr, "{ A } -> { B }\n");
|
|
|
+ fflush(stderr);
|
|
|
+
|
|
|
+ struct starpu_task *taskA, *taskB;
|
|
|
+
|
|
|
+ taskA = create_dummy_task(tagA);
|
|
|
+ taskB = create_dummy_task(tagB);
|
|
|
+
|
|
|
+ /* B depends on A */
|
|
|
+ starpu_tag_declare_deps(tagB, 1, tagA);
|
|
|
+
|
|
|
+ starpu_submit_task(taskB);
|
|
|
+ starpu_submit_task(taskA);
|
|
|
+
|
|
|
+ starpu_tag_wait(tagB);
|
|
|
+
|
|
|
+ fflush(stderr);
|
|
|
+ fprintf(stderr, "{ C, D, E, F } -> { G }\n");
|
|
|
+ fflush(stderr);
|
|
|
+
|
|
|
+ struct starpu_task *taskC, *taskD, *taskE, *taskF, *taskG;
|
|
|
+
|
|
|
+ taskC = create_dummy_task(tagC);
|
|
|
+ taskD = create_dummy_task(tagD);
|
|
|
+ taskE = create_dummy_task(tagE);
|
|
|
+ taskF = create_dummy_task(tagF);
|
|
|
+ taskG = create_dummy_task(tagG);
|
|
|
+
|
|
|
+ /* NB: we could have used starpu_tag_declare_deps_array instead */
|
|
|
+ starpu_tag_declare_deps(tagG, 4, tagC, tagD, tagE, tagF);
|
|
|
+
|
|
|
+ starpu_submit_task(taskC);
|
|
|
+ starpu_submit_task(taskD);
|
|
|
+ starpu_submit_task(taskG);
|
|
|
+ starpu_submit_task(taskE);
|
|
|
+ starpu_submit_task(taskF);
|
|
|
+
|
|
|
+ starpu_tag_wait(tagG);
|
|
|
+
|
|
|
+ fflush(stderr);
|
|
|
+ fprintf(stderr, "{ H, I } -> { J, K, L }\n");
|
|
|
+ fflush(stderr);
|
|
|
+
|
|
|
+ struct starpu_task *taskH, *taskI, *taskJ, *taskK, *taskL;
|
|
|
+
|
|
|
+ taskH = create_dummy_task(tagH);
|
|
|
+ taskI = create_dummy_task(tagI);
|
|
|
+ taskJ = create_dummy_task(tagJ);
|
|
|
+ taskK = create_dummy_task(tagK);
|
|
|
+ taskL = create_dummy_task(tagL);
|
|
|
+
|
|
|
+ starpu_tag_declare_deps(tagJ, 2, tagH, tagI);
|
|
|
+ starpu_tag_declare_deps(tagK, 2, tagH, tagI);
|
|
|
+ starpu_tag_declare_deps(tagL, 2, tagH, tagI);
|
|
|
+
|
|
|
+ starpu_tag_t tagJKL[3] = {tagJ, tagK, tagL};
|
|
|
+
|
|
|
+ starpu_tag_wait_array(3, tagJKL);
|
|
|
+
|
|
|
+ starpu_shutdown();
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|