Simplify the cublas helper functions by the means of the

starpu_execute_on_each_worker helper.

src/util/starpu_cublas.c

@@ -18,91 +18,29 @@
 #include <common/config.h>
 
 #ifdef USE_CUDA
-static void init_cublas_func(starpu_data_interface_t *descr __attribute__((unused)),
-					void *_args __attribute__((unused)))
+static void init_cublas_func(void *args __attribute__((unused)))
 {
 	cublasStatus cublasst = cublasInit();
 	if (STARPU_UNLIKELY(cublasst))
 		CUBLAS_REPORT_ERROR(cublasst);
 }
 
-static void shutdown_cublas_func(starpu_data_interface_t *descr __attribute__((unused)),
-					void *_args __attribute__((unused)))
+static void shutdown_cublas_func(void *args __attribute__((unused)))
 {
 	cublasShutdown();
 }
-
-static starpu_codelet init_cublas_codelet = {
-	.where = CUDA,
-	.cuda_func = init_cublas_func,
-	.model = NULL,
-	.nbuffers = 0
-};
-
-static starpu_codelet shutdown_cublas_codelet = {
-	.where = CUDA,
-	.cuda_func = shutdown_cublas_func,
-	.model = NULL,
-	.nbuffers = 0
-};
 #endif
 
 void starpu_helper_init_cublas(void)
 {
 #ifdef USE_CUDA
-	unsigned worker;
-	unsigned nworkers = starpu_get_worker_count();
-
-	struct starpu_task *tasks[STARPU_NMAXWORKERS];
-
-	for (worker = 0; worker < nworkers; worker++)
-	{
-		if (starpu_get_worker_type(worker) == STARPU_CUDA_WORKER)
-		{
-			tasks[worker] = starpu_task_create();
-
-			tasks[worker]->cl = &init_cublas_codelet;
-
-			tasks[worker]->execute_on_a_specific_worker = 1;
-			tasks[worker]->workerid = worker;
-
-			tasks[worker]->detach = 0;
-			tasks[worker]->destroy = 0;
-
-			starpu_submit_task(tasks[worker]);
-		}
-	}
-
-	for (worker = 0; worker < nworkers; worker++)
-	{
-		if (starpu_get_worker_type(worker) == STARPU_CUDA_WORKER)
-		{
-			starpu_wait_task(tasks[worker]);
-			starpu_task_destroy(tasks[worker]);
-		}
-	}
-
+	starpu_execute_on_each_worker(init_cublas_func, NULL, CUDA);
 #endif
 }
 
 void starpu_helper_shutdown_cublas(void)
 {
 #ifdef USE_CUDA
-	unsigned worker;
-	unsigned nworkers = starpu_get_worker_count();
-	for (worker = 0; worker < nworkers; worker++)
-	{
-		if (starpu_get_worker_type(worker) == STARPU_CUDA_WORKER)
-		{
-			struct starpu_task *task = starpu_task_create();
-			task->cl = &shutdown_cublas_codelet;
-
-			task->execute_on_a_specific_worker = 1;
-			task->workerid = worker;
-
-			task->synchronous = 1;
-			starpu_submit_task(task);
-		}
-	}
+	starpu_execute_on_each_worker(shutdown_cublas_func, NULL, CUDA);
 #endif
 }