Andra Hugo 13 gadi atpakaļ
vecāks
revīzija
c2641ee759
1 mainītis faili ar 36 papildinājumiem un 0 dzēšanām
  1. 36 0
      examples/reductions/dot_product_kernels.cu

+ 36 - 0
examples/reductions/dot_product_kernels.cu

@@ -0,0 +1,36 @@
+/* StarPU --- Runtime system for heterogeneous multicore architectures.
+ *
+ * Copyright (C) 2011  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.
+ */
+
+#include <starpu.h>
+#include <starpu_cuda.h>
+
+#define DOT_TYPE double
+
+static __global__ void cuda_redux(DOT_TYPE *dota, DOT_TYPE *dotb)
+{
+	*dota = *dota + *dotb;
+	return;
+}
+
+extern "C" void redux_cuda_func(void *descr[], void *_args)
+{
+	(void)_args;
+	DOT_TYPE *dota = (DOT_TYPE *)STARPU_VARIABLE_GET_PTR(descr[0]);
+	DOT_TYPE *dotb = (DOT_TYPE *)STARPU_VARIABLE_GET_PTR(descr[1]);
+
+	cuda_redux<<<1,1, 0, starpu_cuda_get_local_stream()>>>(dota, dotb);
+	cudaStreamSynchronize(starpu_cuda_get_local_stream());
+}