|
@@ -1,6 +1,6 @@
|
|
|
/* StarPU --- Runtime system for heterogeneous multicore architectures.
|
|
|
*
|
|
|
- * Copyright (C) 2011, 2012, 2013, 2015, 2016 CNRS
|
|
|
+ * Copyright (C) 2011, 2012, 2013, 2015, 2016, 2017 CNRS
|
|
|
*
|
|
|
* StarPU is free software; you can redistribute it and/or modify
|
|
|
* it under the terms of the GNU Lesser General Public License as published by
|
|
@@ -20,6 +20,19 @@
|
|
|
|
|
|
#define FPRINTF(ofile, fmt, ...) do { if (!getenv("STARPU_SSILENT")) {fprintf(ofile, fmt, ## __VA_ARGS__); }} while(0)
|
|
|
#define FPRINTF_MPI(ofile, fmt, ...) do { if (!getenv("STARPU_SSILENT")) { \
|
|
|
- int _disp_rank; starpu_mpi_comm_rank(MPI_COMM_WORLD, &_disp_rank); \
|
|
|
- fprintf(ofile, "[%d][starpu_mpi][%s] " fmt , _disp_rank, __starpu_func__ ,## __VA_ARGS__); \
|
|
|
- fflush(ofile); }} while(0);
|
|
|
+ int _disp_rank; starpu_mpi_comm_rank(MPI_COMM_WORLD, &_disp_rank); \
|
|
|
+ fprintf(ofile, "[%d][starpu_mpi][%s] " fmt , _disp_rank, __starpu_func__ ,## __VA_ARGS__); \
|
|
|
+ fflush(ofile); }} while(0);
|
|
|
+
|
|
|
+#define MPI_INIT_THREAD(argc, argv, required) do { \
|
|
|
+ int thread_support; \
|
|
|
+ if (MPI_Init_thread(argc, argv, required, &thread_support) != MPI_SUCCESS) \
|
|
|
+ { \
|
|
|
+ fprintf(stderr,"MPI_Init_thread failed\n"); \
|
|
|
+ exit(1); \
|
|
|
+ } \
|
|
|
+ if (thread_support == MPI_THREAD_FUNNELED) \
|
|
|
+ fprintf(stderr,"Warning: MPI only has funneled thread support, not serialized, hoping this will work\n"); \
|
|
|
+ if (thread_support < MPI_THREAD_FUNNELED) \
|
|
|
+ fprintf(stderr,"Warning: MPI does not have thread support!\n"); } while(0);
|
|
|
+
|