diff --git a/src/mca/schizo/ompi/schizo_ompi.c b/src/mca/schizo/ompi/schizo_ompi.c index 9a810ebf72..f4e9abbf41 100644 --- a/src/mca/schizo/ompi/schizo_ompi.c +++ b/src/mca/schizo/ompi/schizo_ompi.c @@ -162,6 +162,16 @@ static prte_cmd_line_init_t cmd_line_init[] = { "Forward mpirun port to compute node daemons so all will use it", PRTE_CMD_LINE_OTYPE_DVM }, + /* Protocol : MPI_Init */ + { '\0', "prot", 0, PRTE_CMD_LINE_TYPE_BOOL, + "Display table of communication methods between ranks during MPI_Init", + PRTE_CMD_LINE_OTYPE_GENERAL }, + + /* Protocol : MPI_Finalize */ + { '\0', "protlazy", 0, PRTE_CMD_LINE_TYPE_BOOL, + "Display table of communication methods between ranks during MPI_Finalize", + PRTE_CMD_LINE_OTYPE_GENERAL }, + /* End of list */ { '\0', NULL, 0, PRTE_CMD_LINE_TYPE_NULL, NULL } @@ -890,6 +900,13 @@ static int parse_env(prte_cmd_line_t *cmd_line, } } + if (prte_cmd_line_is_taken(cmd_line, "prot")) { + prte_setenv("OMPI_MCA_hook_comm_method_enable_mpi_init", "1", true, dstenv); + } + if (prte_cmd_line_is_taken(cmd_line, "protlazy")) { + prte_setenv("OMPI_MCA_hook_comm_method_enable_mpi_finalize", "1", true, dstenv); + } + /* now look for any "--mca" options - note that it is an error * for the same MCA param to be given more than once if the * values differ */