Changeset 1503 for XIOS/dev/branch_openmp/extern/ep_dev/main.cpp

Timestamp:

05/28/18 17:28:07 (6 years ago)

Author:

yushan

Message:

rank_map is passed from vector to map, in order to have more flexibility in comm_split

File:

• XIOS/dev/branch_openmp/extern/ep_dev/main.cpp (modified) (5 diffs)

XIOS/dev/branch_openmp/extern/ep_dev/main.cpp

@@ -52 +52 @@
 
 
+    MPI_Comm comm_for_dup; // this should act as EP_COMM_WORLD
     MPI_Comm comm; // this should act as EP_COMM_WORLD
-    comm = passage[omp_get_thread_num()];
+        
+    comm_for_dup = passage[omp_get_thread_num()];
+    MPI_Comm_dup(comm_for_dup, &comm);
+    
+    MPI_Comm_free(&comm_for_dup);
+    
+    MPI_Barrier(comm);
 
     int rank, size;
     MPI_Comm_rank(comm, &rank);
     MPI_Comm_size(comm, &size);
+    
+    if(rank == 0) printf("           \t test MPI_Comm_dup \t OK \n");
 /*
     // TIMING SYCHRONIZATION
@@ -109 +118 @@
       MPI_Barrier(comm);
       
+      MPI_Comm equal_comm = comm;
+      
+      
+      
+      
       double sendbuf[10];
       double recvbuf[20];
       
-      int sender = 1;
-      int receiver = 2;
+      int sender;
+      if(rank == 0) sender = rand() % size;
+      MPI_Bcast(&sender, 1, MPI_INT, 0, comm);
+      
+      int receiver = sender;
+      if(rank == 0) while(sender == receiver) {receiver = rand() % size;}
+      MPI_Bcast(&receiver, 1, MPI_INT, 0, comm);
+      
+      
+      
       
       if(rank == sender)
       {
         for(int i=0; i<10; i++) sendbuf[i] = 99.99;
-        MPI_Send(sendbuf, 10, MPI_DOUBLE, receiver, 99, comm);
+        MPI_Send(sendbuf, 10, MPI_DOUBLE, receiver, 99, equal_comm);
         for(int i=0; i<10; i++) sendbuf[i] = -99.99;
-        MPI_Send(sendbuf, 10, MPI_DOUBLE, receiver, 11, comm);
+        MPI_Send(sendbuf, 10, MPI_DOUBLE, receiver, 11, equal_comm);
       }
       
@@ -131 +153 @@
         
         for(int i=0; i<20; i++) std::cout << "recvbuf["<< i <<"] = "<< recvbuf[i] << std::endl;
+        printf("sender = %d\nreceiver = %d \tTEST of p2p blocking communication\tOK\n", sender, receiver);
       }
       
@@ -145 +168 @@
       double recvbuf[20];
       
-      int sender = 3;
-      int receiver = 7;
-      
+      int sender;
+      if(rank == 0) sender = rand() % size;
+      MPI_Bcast(&sender, 1, MPI_INT, 0, comm);
+      
+      int receiver = sender;
+      if(rank == 0) receiver = rand() % size;
+      MPI_Bcast(&receiver, 1, MPI_INT, 0, comm);
+
+
+            
       MPI_Request request[2];
       
@@ -179 +209 @@
       {
         for(int i=0; i<20; i++) std::cout << "recvbuf["<< i <<"] = "<< recvbuf[i] << std::endl;
+        printf("sender = %d\nreceiver = %d \tTEST of p2p non-blocking communication\tOK\n", sender, receiver);
       }