Browse Source

Move the opt directory into tests/opt/

Cédric Augonnet 14 years ago
parent
commit
6674168898
6 changed files with 141 additions and 19 deletions
  1. 1 1
      Makefile.am
  2. 1 1
      configure.ac
  3. 16 17
      opt/datawizard/wt_host.c
  4. 0 0
      tests/opt/Makefile.am
  5. 0 0
      tests/opt/datawizard/wt_broadcast.c
  6. 123 0
      tests/opt/datawizard/wt_host.c

+ 1 - 1
Makefile.am

@@ -23,7 +23,7 @@ SUBDIRS += mpi
 endif
 SUBDIRS += tools examples tests doc
 if COND_OPT
-SUBDIRS += opt
+SUBDIRS += tests/opt
 endif
 
 if BUILD_GCC_PLUGIN

+ 1 - 1
configure.ac

@@ -1212,7 +1212,7 @@ AC_OUTPUT([
 	examples/starpufft/Makefile
 	examples/stencil/Makefile
 	tests/Makefile
-        opt/Makefile
+        tests/opt/Makefile
 	doc/Makefile
 	mpi/Makefile
 ])

+ 16 - 17
opt/datawizard/wt_host.c

@@ -92,31 +92,30 @@ int main(int argc, char **argv)
 	uint32_t wt_mask = (1<<0);
 	starpu_data_set_wt_mask(handle, wt_mask);
 
-	unsigned ntasks = 1024;
-	unsigned nloops = 16;
+	unsigned ntasks = 2;
 
-	unsigned loop;
 	unsigned t;
 
-	for (loop = 0; loop < nloops; loop++)
+	for (t = 0; t < ntasks; t++)
 	{
-		for (t = 0; t < ntasks; t++)
-		{
-			struct starpu_task *task = starpu_task_create();
-	
-			task->cl = &increment_cl;
-	
-			task->buffers[0].mode = STARPU_RW;
-			task->buffers[0].handle = handle;
-	
-			int ret = starpu_task_submit(task);
-			STARPU_ASSERT(!ret);
+		struct starpu_task *task = starpu_task_create();
+
+		task->cl = &increment_cl;
+
+		task->buffers[0].mode = STARPU_RW;
+		task->buffers[0].handle = handle;
+
+		int ret = starpu_task_submit(task);
+		STARPU_ASSERT(!ret);
 
-		}
 	}
 
 	starpu_data_unregister(handle);
-	STARPU_ASSERT(var == ntasks*nloops);
+
+	if (var != ntasks)
+		fprintf(stderr, "VAR is %d should be %d\n", var, ntasks);
+
+	STARPU_ASSERT(var == ntasks);
 	
 	starpu_shutdown();
 

opt/Makefile.am → tests/opt/Makefile.am


opt/datawizard/wt_broadcast.c → tests/opt/datawizard/wt_broadcast.c


+ 123 - 0
tests/opt/datawizard/wt_host.c

@@ -0,0 +1,123 @@
+/* StarPU --- Runtime system for heterogeneous multicore architectures.
+ *
+ * Copyright (C) 2011  Université de Bordeaux 1
+ *
+ * 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>
+
+#ifdef STARPU_USE_CUDA
+#include <starpu_cuda.h>
+#endif
+#ifdef STARPU_USE_OPENCL
+#include <starpu_opencl.h>
+#endif
+
+
+static unsigned var = 0;
+static starpu_data_handle handle;
+/*
+ *	Increment codelet
+ */
+
+#ifdef STARPU_USE_OPENCL
+/* dummy OpenCL implementation */
+static void increment_opencl_kernel(void *descr[], void *cl_arg __attribute__((unused)))
+{
+	cl_mem d_token = (cl_mem)STARPU_VARIABLE_GET_PTR(descr[0]);
+	unsigned h_token;
+
+	cl_command_queue queue;
+	starpu_opencl_get_current_queue(&queue);
+
+	clEnqueueReadBuffer(queue, d_token, CL_TRUE, 0, sizeof(unsigned), (void *)&h_token, 0, NULL, NULL);
+	h_token++;
+	clEnqueueWriteBuffer(queue, d_token, CL_TRUE, 0, sizeof(unsigned), (void *)&h_token, 0, NULL, NULL); 
+}
+#endif
+
+
+#ifdef STARPU_USE_CUDA
+static void increment_cuda_kernel(void *descr[], void *arg)
+{
+	unsigned *tokenptr = (unsigned *)STARPU_VARIABLE_GET_PTR(descr[0]);
+	unsigned host_token;
+
+	/* This is a dummy technique of course */
+	cudaMemcpy(&host_token, tokenptr, sizeof(unsigned), cudaMemcpyDeviceToHost);
+	cudaThreadSynchronize();
+
+	host_token++;
+
+	cudaMemcpy(tokenptr, &host_token, sizeof(unsigned), cudaMemcpyHostToDevice);
+	cudaThreadSynchronize();
+}
+#endif
+
+static void increment_cpu_kernel(void *descr[], void *arg)
+{
+	unsigned *tokenptr = (unsigned *)STARPU_VARIABLE_GET_PTR(descr[0]);
+	*tokenptr = *tokenptr + 1;
+}
+
+static starpu_codelet increment_cl = {
+	.where = STARPU_CPU|STARPU_CUDA|STARPU_OPENCL,
+#ifdef STARPU_USE_CUDA
+	.cuda_func = increment_cuda_kernel,
+#endif
+#ifdef STARPU_USE_OPENCL
+	.opencl_func = increment_opencl_kernel,
+#endif
+	.cpu_func = increment_cpu_kernel,
+	.nbuffers = 1
+};
+
+int main(int argc, char **argv)
+{
+	starpu_init(NULL);
+
+	starpu_variable_data_register(&handle, 0, (uintptr_t)&var, sizeof(unsigned));
+
+	/* Copy the handle in main memory every time it is modified */
+	uint32_t wt_mask = (1<<0);
+	starpu_data_set_wt_mask(handle, wt_mask);
+
+	unsigned ntasks = 2;
+
+	unsigned t;
+
+	for (t = 0; t < ntasks; t++)
+	{
+		struct starpu_task *task = starpu_task_create();
+
+		task->cl = &increment_cl;
+
+		task->buffers[0].mode = STARPU_RW;
+		task->buffers[0].handle = handle;
+
+		int ret = starpu_task_submit(task);
+		STARPU_ASSERT(!ret);
+
+	}
+
+	starpu_data_unregister(handle);
+
+	if (var != ntasks)
+		fprintf(stderr, "VAR is %d should be %d\n", var, ntasks);
+
+	STARPU_ASSERT(var == ntasks);
+	
+	starpu_shutdown();
+
+	return 0;
+}