properly initialize magma

examples/cholesky/cholesky_grain_tag.c

@@ -18,6 +18,10 @@
 
 #include "cholesky.h"
 
+#if defined(STARPU_USE_CUDA) && defined(STARPU_HAVE_MAGMA)
+#include "magma.h"
+#endif
+
 /*
  *	Some useful functions
  */
@@ -254,6 +258,10 @@ static void initialize_system(float **A, unsigned dim, unsigned pinned)
 {
 	int ret;
 
+#ifdef STARPU_HAVE_MAGMA
+	magma_init();
+#endif
+
 	ret = starpu_init(NULL);
 	if (ret == -ENODEV)
 		exit(77);

examples/cholesky/cholesky_implicit.c

@@ -19,6 +19,10 @@
 #include "cholesky.h"
 #include "../sched_ctx_utils/sched_ctx_utils.h"
 
+#if defined(STARPU_USE_CUDA) && defined(STARPU_HAVE_MAGMA)
+#include "magma.h"
+#endif
+
 /*
  *	code to bootstrap the factorization
  *	and construct the DAG
@@ -301,6 +305,10 @@ int main(int argc, char **argv)
 	if(with_ctxs || with_noctxs || chole1 || chole2)
 		parse_args_ctx(argc, argv);
 
+#ifdef STARPU_HAVE_MAGMA
+	magma_init();
+#endif
+
 	int ret;
 	ret = starpu_init(NULL);
 	starpu_fxt_stop_profiling();

examples/cholesky/cholesky_tag.c

@@ -19,6 +19,10 @@
 #include "cholesky.h"
 #include <starpu_perfmodel.h>
 
+#if defined(STARPU_USE_CUDA) && defined(STARPU_HAVE_MAGMA)
+#include "magma.h"
+#endif
+
 /*
  *	Some useful functions
  */
@@ -221,6 +225,10 @@ static int initialize_system(float **A, unsigned dim, unsigned pinned)
 {
 	int ret;
 
+#ifdef STARPU_HAVE_MAGMA
+	magma_init();
+#endif
+
 	ret = starpu_init(NULL);
 	if (ret == -ENODEV)
 		return 77;

examples/cholesky/cholesky_tile_tag.c

@@ -17,6 +17,10 @@
 
 #include "cholesky.h"
 
+#if defined(STARPU_USE_CUDA) && defined(STARPU_HAVE_MAGMA)
+#include "magma.h"
+#endif
+
 /* A [ y ] [ x ] */
 float *A[NMAXBLOCKS][NMAXBLOCKS];
 starpu_data_handle_t A_state[NMAXBLOCKS][NMAXBLOCKS];
@@ -216,6 +220,10 @@ int main(int argc, char **argv)
 
 	FPRINTF(stderr, "BLOCK SIZE = %d\n", size / nblocks);
 
+#ifdef STARPU_HAVE_MAGMA
+	magma_init();
+#endif
+
 	ret = starpu_init(NULL);
 	if (ret == -ENODEV)
 		return 77;