Quellcode durchsuchen

Use a different model name when magma support is enabled

Samuel Thibault vor 10 Jahren
Ursprung
Commit
751c44cd97

+ 6 - 0
examples/cholesky/cholesky_grain_tag.c

@@ -268,9 +268,15 @@ static void initialize_system(float **A, unsigned dim, unsigned pinned)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 
 #ifdef STARPU_USE_CUDA
+#ifdef STARPU_HAVE_MAGMA
+	initialize_chol_model(&chol_model_11,"chol_model_11_magma",cpu_chol_task_11_cost,cuda_chol_task_11_cost);
+	initialize_chol_model(&chol_model_21,"chol_model_21_magma",cpu_chol_task_21_cost,cuda_chol_task_21_cost);
+	initialize_chol_model(&chol_model_22,"chol_model_22_magma",cpu_chol_task_22_cost,cuda_chol_task_22_cost);
+#else
 	initialize_chol_model(&chol_model_11,"chol_model_11",cpu_chol_task_11_cost,cuda_chol_task_11_cost);
 	initialize_chol_model(&chol_model_21,"chol_model_21",cpu_chol_task_21_cost,cuda_chol_task_21_cost);
 	initialize_chol_model(&chol_model_22,"chol_model_22",cpu_chol_task_22_cost,cuda_chol_task_22_cost);
+#endif
 #else
 	initialize_chol_model(&chol_model_11,"chol_model_11",cpu_chol_task_11_cost,NULL);
 	initialize_chol_model(&chol_model_21,"chol_model_21",cpu_chol_task_21_cost,NULL);

+ 6 - 0
examples/cholesky/cholesky_implicit.c

@@ -318,9 +318,15 @@ int main(int argc, char **argv)
         STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 
 #ifdef STARPU_USE_CUDA
+#ifdef STARPU_HAVE_MAGMA
+	initialize_chol_model(&chol_model_11,"chol_model_11_magma",cpu_chol_task_11_cost,cuda_chol_task_11_cost);
+	initialize_chol_model(&chol_model_21,"chol_model_21_magma",cpu_chol_task_21_cost,cuda_chol_task_21_cost);
+	initialize_chol_model(&chol_model_22,"chol_model_22_magma",cpu_chol_task_22_cost,cuda_chol_task_22_cost);
+#else
 	initialize_chol_model(&chol_model_11,"chol_model_11",cpu_chol_task_11_cost,cuda_chol_task_11_cost);
 	initialize_chol_model(&chol_model_21,"chol_model_21",cpu_chol_task_21_cost,cuda_chol_task_21_cost);
 	initialize_chol_model(&chol_model_22,"chol_model_22",cpu_chol_task_22_cost,cuda_chol_task_22_cost);
+#endif
 #else
 	initialize_chol_model(&chol_model_11,"chol_model_11",cpu_chol_task_11_cost,NULL);
 	initialize_chol_model(&chol_model_21,"chol_model_21",cpu_chol_task_21_cost,NULL);

+ 0 - 3
examples/cholesky/cholesky_kernels.c

@@ -252,9 +252,6 @@ struct starpu_codelet cl11 =
 #elif defined(STARPU_SIMGRID)
 	.cuda_funcs = {(void*)1},
 #endif
-#ifdef STARPU_HAVE_MAGMA
-	.cuda_flags = {STARPU_CUDA_ASYNC},
-#endif
 	.nbuffers = 1,
 	.modes = { STARPU_RW },
 	.model = &chol_model_11

+ 6 - 0
examples/cholesky/cholesky_tag.c

@@ -235,9 +235,15 @@ static int initialize_system(float **A, unsigned dim, unsigned pinned)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 
 #ifdef STARPU_USE_CUDA
+#ifdef STARPU_HAVE_MAGMA
+	initialize_chol_model(&chol_model_11,"chol_model_11_magma",cpu_chol_task_11_cost,cuda_chol_task_11_cost);
+	initialize_chol_model(&chol_model_21,"chol_model_21_magma",cpu_chol_task_21_cost,cuda_chol_task_21_cost);
+	initialize_chol_model(&chol_model_22,"chol_model_22_magma",cpu_chol_task_22_cost,cuda_chol_task_22_cost);
+#else
 	initialize_chol_model(&chol_model_11,"chol_model_11",cpu_chol_task_11_cost,cuda_chol_task_11_cost);
 	initialize_chol_model(&chol_model_21,"chol_model_21",cpu_chol_task_21_cost,cuda_chol_task_21_cost);
 	initialize_chol_model(&chol_model_22,"chol_model_22",cpu_chol_task_22_cost,cuda_chol_task_22_cost);
+#endif
 #else
 	initialize_chol_model(&chol_model_11,"chol_model_11",cpu_chol_task_11_cost,NULL);
 	initialize_chol_model(&chol_model_21,"chol_model_21",cpu_chol_task_21_cost,NULL);

+ 6 - 0
examples/cholesky/cholesky_tile_tag.c

@@ -230,9 +230,15 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 
 #ifdef STARPU_USE_CUDA
+#ifdef STARPU_HAVE_MAGMA
+	initialize_chol_model(&chol_model_11,"chol_model_11_magma",cpu_chol_task_11_cost,cuda_chol_task_11_cost);
+	initialize_chol_model(&chol_model_21,"chol_model_21_magma",cpu_chol_task_21_cost,cuda_chol_task_21_cost);
+	initialize_chol_model(&chol_model_22,"chol_model_22_magma",cpu_chol_task_22_cost,cuda_chol_task_22_cost);
+#else
 	initialize_chol_model(&chol_model_11,"chol_model_11",cpu_chol_task_11_cost,cuda_chol_task_11_cost);
 	initialize_chol_model(&chol_model_21,"chol_model_21",cpu_chol_task_21_cost,cuda_chol_task_21_cost);
 	initialize_chol_model(&chol_model_22,"chol_model_22",cpu_chol_task_22_cost,cuda_chol_task_22_cost);
+#endif
 #else
 	initialize_chol_model(&chol_model_11,"chol_model_11",cpu_chol_task_11_cost,NULL);
 	initialize_chol_model(&chol_model_21,"chol_model_21",cpu_chol_task_21_cost,NULL);