|
@@ -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);
|