瀏覽代碼

Use starpu_mpi_comm_rank/size instead of MPI_Comm_rank/size in mpi applications

Samuel Thibault 10 年之前
父節點
當前提交
b8dd6df73a
共有 57 個文件被更改,包括 115 次插入113 次删除
  1. 2 2
      mpi/examples/complex/mpi_complex.c
  2. 2 1
      mpi/examples/helper.h
  3. 2 2
      mpi/examples/matrix_decomposition/mpi_cholesky.c
  4. 2 2
      mpi/examples/matrix_decomposition/mpi_cholesky_distributed.c
  5. 2 2
      mpi/examples/mpi_lu/plu_example.c
  6. 2 2
      mpi/examples/mpi_lu/plu_implicit_example.c
  7. 2 2
      mpi/examples/mpi_lu/plu_outofcore_example.c
  8. 2 2
      mpi/examples/mpi_lu/plu_solve.c
  9. 4 4
      mpi/examples/mpi_lu/pxlu_kernels.c
  10. 3 3
      mpi/examples/stencil/stencil5.c
  11. 2 2
      mpi/src/starpu_mpi_datatype.c
  12. 2 2
      mpi/tests/block_interface.c
  13. 2 2
      mpi/tests/block_interface_pinned.c
  14. 1 1
      mpi/tests/cache.c
  15. 1 1
      mpi/tests/cache_disable.c
  16. 2 2
      mpi/tests/callback.c
  17. 2 2
      mpi/tests/datatypes.c
  18. 3 3
      mpi/tests/early_request.c
  19. 2 2
      mpi/tests/gather.c
  20. 2 2
      mpi/tests/gather2.c
  21. 2 1
      mpi/tests/helper.h
  22. 2 2
      mpi/tests/insert_task.c
  23. 2 2
      mpi/tests/insert_task_block.c
  24. 1 1
      mpi/tests/insert_task_compute.c
  25. 2 2
      mpi/tests/insert_task_count.c
  26. 1 1
      mpi/tests/insert_task_dyn_handles.c
  27. 3 3
      mpi/tests/insert_task_node_choice.c
  28. 3 3
      mpi/tests/insert_task_owner.c
  29. 2 2
      mpi/tests/insert_task_owner2.c
  30. 2 2
      mpi/tests/insert_task_owner_data.c
  31. 2 2
      mpi/tests/insert_task_recv_cache.c
  32. 2 2
      mpi/tests/insert_task_sent_cache.c
  33. 1 1
      mpi/tests/matrix.c
  34. 2 2
      mpi/tests/matrix2.c
  35. 2 2
      mpi/tests/mpi_detached_tag.c
  36. 2 2
      mpi/tests/mpi_earlyrecv.c
  37. 2 2
      mpi/tests/mpi_earlyrecv2.c
  38. 2 2
      mpi/tests/mpi_earlyrecv2_sync.c
  39. 2 2
      mpi/tests/mpi_irecv.c
  40. 2 2
      mpi/tests/mpi_irecv_detached.c
  41. 2 2
      mpi/tests/mpi_isend.c
  42. 2 2
      mpi/tests/mpi_isend_detached.c
  43. 2 2
      mpi/tests/mpi_reduction.c
  44. 2 2
      mpi/tests/mpi_redux.c
  45. 2 2
      mpi/tests/mpi_scatter_gather.c
  46. 2 2
      mpi/tests/mpi_test.c
  47. 2 2
      mpi/tests/multiple_send.c
  48. 2 2
      mpi/tests/pingpong.c
  49. 2 2
      mpi/tests/policy_register.c
  50. 2 2
      mpi/tests/ring.c
  51. 2 2
      mpi/tests/ring_async.c
  52. 2 2
      mpi/tests/ring_async_implicit.c
  53. 2 2
      mpi/tests/ring_sync.c
  54. 2 2
      mpi/tests/ring_sync_detached.c
  55. 2 2
      mpi/tests/sync.c
  56. 2 2
      mpi/tests/tags_checking.c
  57. 2 2
      mpi/tests/user_defined_datatype.c

+ 2 - 2
mpi/examples/complex/mpi_complex.c

@@ -43,8 +43,8 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &nodes);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &nodes);
 
 	if (nodes < 2)
 	{

+ 2 - 1
mpi/examples/helper.h

@@ -15,12 +15,13 @@
  */
 
 #include <errno.h>
+#include <starpu_mpi.h>
 
 #define STARPU_TEST_SKIPPED 77
 
 #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; MPI_Comm_rank(MPI_COMM_WORLD, &_disp_rank);       \
+    						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);
 

+ 2 - 2
mpi/examples/matrix_decomposition/mpi_cholesky.c

@@ -35,8 +35,8 @@ int main(int argc, char **argv)
 
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &nodes);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &nodes);
 	starpu_cublas_init();
 
 	parse_args(argc, argv, nodes);

+ 2 - 2
mpi/examples/matrix_decomposition/mpi_cholesky_distributed.c

@@ -34,8 +34,8 @@ int main(int argc, char **argv)
 
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &nodes);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &nodes);
 	starpu_cublas_init();
 
 	parse_args(argc, argv, nodes);

+ 2 - 2
mpi/examples/mpi_lu/plu_example.c

@@ -419,8 +419,8 @@ int main(int argc, char **argv)
 	if (thread_support < MPI_THREAD_FUNNELED)
 		fprintf(stderr,"Warning: MPI does not have thread support!\n");
 
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &world_size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &world_size);
 
 	starpu_srand48((long int)time(NULL));
 

+ 2 - 2
mpi/examples/mpi_lu/plu_implicit_example.c

@@ -237,8 +237,8 @@ int main(int argc, char **argv)
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
 
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &world_size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &world_size);
 
 	STARPU_ASSERT(p*q == world_size);
 

+ 2 - 2
mpi/examples/mpi_lu/plu_outofcore_example.c

@@ -254,8 +254,8 @@ int main(int argc, char **argv)
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
 
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &world_size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &world_size);
 
 	STARPU_ASSERT(p*q == world_size);
 

+ 2 - 2
mpi/examples/mpi_lu/plu_solve.c

@@ -233,7 +233,7 @@ TYPE *STARPU_PLU(reconstruct_matrix)(unsigned size, unsigned nblocks)
 	unsigned block_size = size/nblocks;
 
 	int rank;
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 
 	unsigned bi, bj;
 	for (bj = 0; bj < nblocks; bj++)
@@ -331,7 +331,7 @@ void STARPU_PLU(compute_lu_matrix)(unsigned size, unsigned nblocks, TYPE *Asaved
 	unsigned display = STARPU_PLU(display_flag)();
 
 	int rank;
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 
 	if (rank == 0)
 	{

+ 4 - 4
mpi/examples/mpi_lu/pxlu_kernels.c

@@ -44,7 +44,7 @@ static inline void STARPU_PLU(common_u22)(void *descr[],
 	struct debug_info *info = _args;
 
 	int rank;
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 	fprintf(stderr, "KERNEL 22 %d - k = %d i = %d j = %d\n", rank, info->k, info->i, info->j);
 #endif
 
@@ -142,7 +142,7 @@ static inline void STARPU_PLU(common_u12)(void *descr[],
 	struct debug_info *info = _args;
 
 	int rank;
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 #warning fixed debugging according to other tweak
 	//fprintf(stderr, "KERNEL 12 %d - k = %d i %d\n", rank, info->k, info->i);
 	fprintf(stderr, "KERNEL 21 %d - k = %d i %d\n", rank, info->k, info->j);
@@ -250,7 +250,7 @@ static inline void STARPU_PLU(common_u21)(void *descr[],
 	struct debug_info *info = _args;
 
 	int rank;
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 #warning fixed debugging according to other tweak
 	//fprintf(stderr, "KERNEL 21 %d (k = %d, i = %d)\n", rank, info->k, info->i);
 	fprintf(stderr, "KERNEL 12 %d (k = %d, j = %d)\n", rank, info->k, info->j);
@@ -356,7 +356,7 @@ static inline void STARPU_PLU(common_u11)(void *descr[],
 	struct debug_info *info = _args;
 
 	int rank;
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 	fprintf(stderr, "KERNEL 11 %d - k = %d\n", rank, info->k);
 #endif
 

+ 3 - 3
mpi/examples/stencil/stencil5.c

@@ -20,7 +20,7 @@
 
 #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; MPI_Comm_rank(MPI_COMM_WORLD, &_disp_rank);       \
+    						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);
 
@@ -97,8 +97,8 @@ int main(int argc, char **argv)
 	int ret = starpu_init(NULL);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	starpu_mpi_init(&argc, &argv, 1);
-	MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &my_rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	parse_args(argc, argv);
 

+ 2 - 2
mpi/src/starpu_mpi_datatype.c

@@ -1,6 +1,6 @@
 /* StarPU --- Runtime system for heterogeneous multicore architectures.
  *
- * Copyright (C) 2009-2011  Université de Bordeaux
+ * Copyright (C) 2009-2011, 2015  Université de Bordeaux
  * Copyright (C) 2010, 2011, 2012, 2013, 2014  CNRS
  *
  * StarPU is free software; you can redistribute it and/or modify
@@ -234,7 +234,7 @@ char *_starpu_mpi_datatype(MPI_Datatype datatype)
      if (datatype == MPI_COMPLEX) return "MPI_COMPLEX";
      if (datatype == MPI_DOUBLE_COMPLEX) return "MPI_DOUBLE_COMPLEX";
      if (datatype == MPI_LOGICAL) return "MPI_LOGICAL";
-     if (datatype == MPI_REAL) return "MPI_REAL";
+     //if (datatype == MPI_REAL) return "MPI_REAL";
      if (datatype == MPI_REAL4) return "MPI_REAL4";
      if (datatype == MPI_REAL8) return "MPI_REAL8";
      if (datatype == MPI_DOUBLE_PRECISION) return "MPI_DOUBLE_PRECISION";

+ 2 - 2
mpi/tests/block_interface.c

@@ -29,8 +29,8 @@ int main(int argc, char **argv)
 	int ret, rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size < 2)
 	{

+ 2 - 2
mpi/tests/block_interface_pinned.c

@@ -29,8 +29,8 @@ int main(int argc, char **argv)
 	int ret, rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size < 2)
 	{

+ 1 - 1
mpi/tests/cache.c

@@ -78,7 +78,7 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 
 	if (starpu_mpi_cache_is_enabled() == 0) goto skip;
 

+ 1 - 1
mpi/tests/cache_disable.c

@@ -43,7 +43,7 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 
 	cache = starpu_mpi_cache_is_enabled();
 	if (cache == 0) goto skip;

+ 2 - 2
mpi/tests/callback.c

@@ -65,8 +65,8 @@ int main(int argc, char **argv)
 
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	ret = starpu_mpi_task_insert(MPI_COMM_WORLD,
 				     NULL,

+ 2 - 2
mpi/tests/datatypes.c

@@ -176,8 +176,8 @@ int main(int argc, char **argv)
 	int nz=4;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size < 2)
 	{

+ 3 - 3
mpi/tests/early_request.c

@@ -113,7 +113,7 @@ void init_element(struct element *el, int size, int foreign_domain)
 	el->foreign_domain=foreign_domain;
 
 	int mpi_rank, mpi_size;
-	MPI_Comm_rank(MPI_COMM_WORLD, &mpi_rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &mpi_rank);
 
 	starpu_vector_data_register(&el->recv, 0, (uintptr_t)el->array_recv, size, sizeof(int));
 	starpu_vector_data_register(&el->send, 0, (uintptr_t)el->array_send, size, sizeof(int));
@@ -170,8 +170,8 @@ int main(int argc, char * argv[])
 	int ret;
 	int mpi_rank, mpi_size;
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &mpi_rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &mpi_size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &mpi_rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &mpi_size);
 
 	ret = starpu_init(NULL);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");

+ 2 - 2
mpi/tests/gather.c

@@ -24,8 +24,8 @@ int main(int argc, char **argv)
 	int var;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size<3)
 	{

+ 2 - 2
mpi/tests/gather2.c

@@ -22,8 +22,8 @@ int main(int argc, char **argv)
 	int ret, rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size<3)
 	{

+ 2 - 1
mpi/tests/helper.h

@@ -15,12 +15,13 @@
  */
 
 #include <errno.h>
+#include <starpu_mpi.h>
 
 #define STARPU_TEST_SKIPPED 77
 
 #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; MPI_Comm_rank(MPI_COMM_WORLD, &_disp_rank);       \
+    						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);
 

+ 2 - 2
mpi/tests/insert_task.c

@@ -55,8 +55,8 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	for(x = 0; x < X; x++)
 	{

+ 2 - 2
mpi/tests/insert_task_block.c

@@ -72,8 +72,8 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_initialize_extended");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	for(x = 0; x < SIZE; x++)
 	{

+ 1 - 1
mpi/tests/insert_task_compute.c

@@ -209,7 +209,7 @@ int main(int argc, char **argv)
 	int node, insert_task, data_array;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 
 	global_ret = 0;
 	for(node=0 ; node<=1 ; node++)

+ 2 - 2
mpi/tests/insert_task_count.c

@@ -51,8 +51,8 @@ int main(int argc, char **argv)
 	starpu_data_handle_t token_handle;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size < 2)
 	{

+ 1 - 1
mpi/tests/insert_task_dyn_handles.c

@@ -59,7 +59,7 @@ int main(int argc, char **argv)
 	struct starpu_data_descr *descrs;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 
 	ret = starpu_init(NULL);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");

+ 3 - 3
mpi/tests/insert_task_node_choice.c

@@ -24,7 +24,7 @@ void func_cpu(void *descr[], STARPU_ATTRIBUTE_UNUSED void *_args)
 	int rank;
 
 	starpu_codelet_unpack_args(_args, &node);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 	FPRINTF_MPI(stderr, "Expected node: %d - Actual node: %d\n", node, rank);
 
 	assert(node == rank);
@@ -50,8 +50,8 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (rank != 0 && rank != 1) goto end;
 

+ 3 - 3
mpi/tests/insert_task_owner.c

@@ -24,7 +24,7 @@ void func_cpu(void *descr[], STARPU_ATTRIBUTE_UNUSED void *_args)
 	int rank;
 
 	starpu_codelet_unpack_args(_args, &node);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 	FPRINTF_MPI(stderr, "Expected node: %d - Actual node: %d\n", node, rank);
 
 	assert(node == rank);
@@ -77,8 +77,8 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (rank != 0 && rank != 1) goto end;
 

+ 2 - 2
mpi/tests/insert_task_owner2.c

@@ -56,8 +56,8 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (rank == 0)
 	{

+ 2 - 2
mpi/tests/insert_task_owner_data.c

@@ -46,8 +46,8 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (rank == 0)
 	{

+ 2 - 2
mpi/tests/insert_task_recv_cache.c

@@ -115,8 +115,8 @@ int main(int argc, char **argv)
 	size_t *comm_amount_without_cache;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	setenv("STARPU_COMM_STATS", "1", 1);
 	setenv("STARPU_MPI_CACHE_STATS", "1", 1);

+ 2 - 2
mpi/tests/insert_task_sent_cache.c

@@ -121,8 +121,8 @@ int main(int argc, char **argv)
 	size_t *comm_amount_without_cache;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	setenv("STARPU_COMM_STATS", "1", 1);
 

+ 1 - 1
mpi/tests/matrix.c

@@ -52,7 +52,7 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
 
 	for(n = 0; n < N; n++)
 	{

+ 2 - 2
mpi/tests/matrix2.c

@@ -48,8 +48,8 @@ int main(int argc, char **argv)
 	starpu_data_handle_t data_X[N];
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size < 3)
 	{

+ 2 - 2
mpi/tests/mpi_detached_tag.c

@@ -33,8 +33,8 @@ int main(int argc, char **argv)
 	int ret, rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size%2 != 0)
 	{

+ 2 - 2
mpi/tests/mpi_earlyrecv.c

@@ -27,8 +27,8 @@ int main(int argc, char **argv)
 	starpu_mpi_req request[2] = {NULL, NULL};
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size%2 != 0)
 	{

+ 2 - 2
mpi/tests/mpi_earlyrecv2.c

@@ -206,8 +206,8 @@ int main(int argc, char **argv)
 	int rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size%2 != 0)
 	{

+ 2 - 2
mpi/tests/mpi_earlyrecv2_sync.c

@@ -211,8 +211,8 @@ int main(int argc, char **argv)
 	int rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size%2 != 0)
 	{

+ 2 - 2
mpi/tests/mpi_irecv.c

@@ -33,8 +33,8 @@ int main(int argc, char **argv)
 	int ret, rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size%2 != 0)
 	{

+ 2 - 2
mpi/tests/mpi_irecv_detached.c

@@ -48,8 +48,8 @@ int main(int argc, char **argv)
 	int ret, rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size%2 != 0)
 	{

+ 2 - 2
mpi/tests/mpi_isend.c

@@ -33,8 +33,8 @@ int main(int argc, char **argv)
 	int ret, rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size%2 != 0)
 	{

+ 2 - 2
mpi/tests/mpi_isend_detached.c

@@ -47,8 +47,8 @@ int main(int argc, char **argv)
 	starpu_data_handle_t tab_handle;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size%2 != 0)
 	{

+ 2 - 2
mpi/tests/mpi_reduction.c

@@ -80,8 +80,8 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &my_rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	nb_elements = size*8000;
 	step = 4;

+ 2 - 2
mpi/tests/mpi_redux.c

@@ -38,8 +38,8 @@ int main(int argc, char **argv)
 	starpu_data_handle_t *handles;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	sum = ((size-1) * (size) / 2);
 

+ 2 - 2
mpi/tests/mpi_scatter_gather.c

@@ -68,8 +68,8 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &nodes);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &nodes);
 
 	if (rank == 0)
 	{

+ 2 - 2
mpi/tests/mpi_test.c

@@ -34,8 +34,8 @@ int main(int argc, char **argv)
 	int ret, rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size%2 != 0)
 	{

+ 2 - 2
mpi/tests/multiple_send.c

@@ -30,8 +30,8 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size < 2)
 	{

+ 2 - 2
mpi/tests/pingpong.c

@@ -34,8 +34,8 @@ int main(int argc, char **argv)
 	int ret, rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size%2 != 0)
 	{

+ 2 - 2
mpi/tests/policy_register.c

@@ -59,8 +59,8 @@ int main(int argc, char **argv)
 	starpu_data_handle_t handles[2];
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size < 2)
 	{

+ 2 - 2
mpi/tests/ring.c

@@ -64,8 +64,8 @@ int main(int argc, char **argv)
 	int ret, rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size < 2)
 	{

+ 2 - 2
mpi/tests/ring_async.c

@@ -64,8 +64,8 @@ int main(int argc, char **argv)
 	int ret, rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size < 2)
 	{

+ 2 - 2
mpi/tests/ring_async_implicit.c

@@ -66,8 +66,8 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(NULL, NULL, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size < 2)
 	{

+ 2 - 2
mpi/tests/ring_sync.c

@@ -64,8 +64,8 @@ int main(int argc, char **argv)
 	int ret, rank, size;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size < 2)
 	{

+ 2 - 2
mpi/tests/ring_sync_detached.c

@@ -76,8 +76,8 @@ int main(int argc, char **argv)
 	starpu_data_handle_t token_handle;
 
 	MPI_Init(&argc, &argv);
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &size);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
 	if (size < 2)
 	{

+ 2 - 2
mpi/tests/sync.c

@@ -25,8 +25,8 @@ int main(int argc, char **argv)
 	starpu_data_handle_t data[2];
 
         MPI_Init(&argc, &argv);
-        MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-        MPI_Comm_size(MPI_COMM_WORLD, &size);
+        starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+        starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
         if (size % 2)
         {

+ 2 - 2
mpi/tests/tags_checking.c

@@ -125,8 +125,8 @@ int main(int argc, char **argv)
 	int sdetached, rdetached;
 
         MPI_Init(&argc, &argv);
-        MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-        MPI_Comm_size(MPI_COMM_WORLD, &size);
+        starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+        starpu_mpi_comm_size(MPI_COMM_WORLD, &size);
 
         if (size < 2)
         {

+ 2 - 2
mpi/tests/user_defined_datatype.c

@@ -69,8 +69,8 @@ int main(int argc, char **argv)
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 	ret = starpu_mpi_init(&argc, &argv, 1);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
-	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
-	MPI_Comm_size(MPI_COMM_WORLD, &nodes);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &rank);
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &nodes);
 
 	if (nodes < 2)
 	{