Add MPI_Comm_split_type

Author: wangvsa

include/recorder-log-format.h

@@ -127,7 +127,7 @@ static const char* func_list[] = {
     "fseeko",       "ftello",
 
 
-    // MPI 86 functions
+    // MPI 87 functions
     "PMPI_File_close",              "PMPI_File_set_size",       "PMPI_File_iread_at",
     "PMPI_File_iread",              "PMPI_File_iread_shared",   "PMPI_File_iwrite_at",
     "PMPI_File_iwrite",             "PMPI_File_iwrite_shared",  "PMPI_File_open",
@@ -170,7 +170,7 @@ static const char* func_list[] = {
     "PMPI_Testany",                 "PMPI_Ireduce",             "PMPI_Iscatter",
     "PMPI_Igather",                 "PMPI_Ialltoall",
     // Added 2021/01/25
-    "PMPI_Comm_free",               "PMPI_Cart_sub",
+    "PMPI_Comm_free",               "PMPI_Cart_sub",            "PMPI_Comm_split_type",
 
     // HDF5 I/O - 74 functions
     "H5Fcreate",            "H5Fopen",              "H5Fclose",     "H5Fflush", // File interface
@@ -228,7 +228,7 @@ static char filename_arg_pos[] = {
     0b00000001,  0b00000001,  0b00000001,  0b00000001,  0b00000000, // remove
     0b00000001,  0b00000001,
 
-    // MPI 86 functions
+    // MPI 87 functions
     // Only MPI_File_open has the filename argument
     0b00000000,  0b00000000,  0b00000000,
     0b00000000,  0b00000000,  0b00000000,
@@ -270,7 +270,7 @@ static char filename_arg_pos[] = {
     0b00000000,  0b00000000,  0b00000000,
     0b00000000,  0b00000000,
     // Added 2020/01/25
-    0b00000000,  0b00000000,
+    0b00000000,  0b00000000,  0b00000000,
 
 
     // HDF5 I/O - 74 functions

include/recorder.h

@@ -368,7 +368,7 @@ RECORDER_FORWARD_DECL(PMPI_Ialltoall, int, (const void *sendbuf, int sendcount,
 // Add on 2021/01/25
 RECORDER_FORWARD_DECL(PMPI_Comm_free, int, (MPI_Comm *comm));
 RECORDER_FORWARD_DECL(PMPI_Cart_sub, int, (MPI_Comm comm, const int remain_dims[], MPI_Comm *newcomm));
-
+RECORDER_FORWARD_DECL(PMPI_Comm_split_type, int, (MPI_Comm comm, int split_type, int key, MPI_Info info, MPI_Comm *newcomm));
 
 
 

lib/recorder-mpi.c

@@ -75,10 +75,6 @@ typedef struct MPICommHash_t {
 } MPICommHash;
 static MPICommHash *mpi_comm_table = NULL;
 
-
-static int split_times;         // how many time we see MPI_Comm_split
-static int dup_times;
-
 void add_mpi_comm(MPI_Comm *newcomm) {
     if(newcomm == NULL || *newcomm == MPI_COMM_NULL)
         return;
@@ -728,3 +724,10 @@ int RECORDER_MPI_DECL(MPI_Cart_sub) (MPI_Comm comm, const int remain_dims[], MPI
     RECORDER_INTERCEPTOR(3, args);
 }
 
+
+int RECORDER_MPI_DECL(MPI_Comm_split_type) (MPI_Comm comm, int split_type, int key, MPI_Info info, MPI_Comm *newcomm) {
+    RECORDER_INTERCEPTOR_NOIO(int, PMPI_Comm_split_type, (comm, split_type, key, info, newcomm));
+    add_mpi_comm(newcomm);
+    char **args = assemble_args_list(5, comm2name(&comm), itoa(split_type), itoa(key), ptoa(&info), comm2name(newcomm));
+    RECORDER_INTERCEPTOR(5, args);
+}