Browse Source

examples/cg/cg.c: skip test when no required workers are available

Nathalie Furmento 4 years ago
parent
commit
0748a190d0
1 changed files with 5 additions and 5 deletions
  1. 5 5
      examples/cg/cg.c

+ 5 - 5
examples/cg/cg.c

@@ -76,12 +76,9 @@ static int copy_handle(starpu_data_handle_t dst, starpu_data_handle_t src, unsig
 
 #include "cg_kernels.c"
 
-
-
 static TYPE *A, *b, *x;
 static TYPE *r, *d, *q;
 
-
 static int copy_handle(starpu_data_handle_t dst, starpu_data_handle_t src, unsigned nblocks)
 {
 	unsigned b;
@@ -91,11 +88,9 @@ static int copy_handle(starpu_data_handle_t dst, starpu_data_handle_t src, unsig
 	return 0;
 }
 
-
 /*
  *	Generate Input data
  */
-
 static void generate_random_problem(void)
 {
 	int i, j;
@@ -314,6 +309,11 @@ int main(int argc, char **argv)
 	if (ret == -ENODEV)
 		return 77;
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
+	if (starpu_cpu_worker_get_count() + starpu_cuda_worker_get_count() + starpu_opencl_worker_get_count() == 0)
+	{
+		starpu_shutdown();
+		return 77;
+	}
 
 	starpu_cublas_init();