|
@@ -0,0 +1,84 @@
|
|
|
+/* StarPU --- Runtime system for heterogeneous multicore architectures.
|
|
|
+ *
|
|
|
+ * Copyright (C) 2015 Inria
|
|
|
+ *
|
|
|
+ * 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.
|
|
|
+ */
|
|
|
+
|
|
|
+#include <starpu.h>
|
|
|
+#include "../helper.h"
|
|
|
+
|
|
|
+static void sum_cpu(void * descr[], void *cl_arg)
|
|
|
+{
|
|
|
+ double * v_dst = (double *) STARPU_VARIABLE_GET_PTR(descr[0]);
|
|
|
+ double * v_src = (double *) STARPU_VARIABLE_GET_PTR(descr[1]);
|
|
|
+ *v_dst+=*v_src;
|
|
|
+}
|
|
|
+
|
|
|
+static void sum3_cpu(void * descr[], void *cl_arg)
|
|
|
+{
|
|
|
+ double * v_src1 = (double *) STARPU_VARIABLE_GET_PTR(descr[1]);
|
|
|
+ double * v_src2 = (double *) STARPU_VARIABLE_GET_PTR(descr[1]);
|
|
|
+ double * v_dst = (double *) STARPU_VARIABLE_GET_PTR(descr[0]);
|
|
|
+ *v_dst+=*v_src1+*v_src2;
|
|
|
+}
|
|
|
+
|
|
|
+static struct starpu_codelet sum_cl =
|
|
|
+{
|
|
|
+ .cpu_funcs = {sum_cpu, NULL},
|
|
|
+ .nbuffers = 2,
|
|
|
+ .modes={STARPU_RW,STARPU_R}
|
|
|
+};
|
|
|
+
|
|
|
+static struct starpu_codelet sum3_cl =
|
|
|
+{
|
|
|
+ .cpu_funcs = {sum3_cpu, NULL},
|
|
|
+ .nbuffers = 3,
|
|
|
+ .modes={STARPU_R,STARPU_R,STARPU_RW}
|
|
|
+};
|
|
|
+
|
|
|
+int main(int argc, char * argv[])
|
|
|
+{
|
|
|
+ starpu_data_handle_t handle;
|
|
|
+ int ret = 0;
|
|
|
+ double value=1.0;
|
|
|
+ int i;
|
|
|
+
|
|
|
+ ret=starpu_init(NULL);
|
|
|
+ if (ret == -ENODEV) return STARPU_TEST_SKIPPED;
|
|
|
+
|
|
|
+ starpu_variable_data_register(&handle,0,(uintptr_t)&value,sizeof(double));
|
|
|
+
|
|
|
+ for (i=0; i<2; i++)
|
|
|
+ {
|
|
|
+ starpu_insert_task(&sum_cl,
|
|
|
+ STARPU_RW, handle,
|
|
|
+ STARPU_R, handle,
|
|
|
+ 0);
|
|
|
+ starpu_insert_task(&sum3_cl,
|
|
|
+ STARPU_R, handle,
|
|
|
+ STARPU_R, handle,
|
|
|
+ STARPU_RW, handle,
|
|
|
+ 0);
|
|
|
+ }
|
|
|
+
|
|
|
+ starpu_task_wait_for_all();
|
|
|
+ starpu_data_unregister(handle);
|
|
|
+ if (value != 36)
|
|
|
+ {
|
|
|
+ FPRINTF(stderr, "value is %f instead of %f\n", value, 36);
|
|
|
+ ret = EXIT_FAILURE;
|
|
|
+ }
|
|
|
+
|
|
|
+ starpu_shutdown();
|
|
|
+ return ret;
|
|
|
+}
|