Add -prot/-protlazy to OMPI CLI

Signed-off-by: Joshua Hursey <jhursey@us.ibm.com>

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,16 @@ static int parse_env(prte_cmd_line_t *cmd_line,
         }
     }
 
+    if (prte_cmd_line_is_taken(cmd_line, "prot") && prte_cmd_line_is_taken(cmd_line, "protlazy")) {
+        prte_setenv("OMPI_MCA_ompi_comm_method", "mpi_init,mpi_finalize", true, dstenv);
+    }
+    else if (prte_cmd_line_is_taken(cmd_line, "prot")) {
+        prte_setenv("OMPI_MCA_ompi_comm_method", "mpi_init", true, dstenv);
+    }
+    else if (prte_cmd_line_is_taken(cmd_line, "protlazy")) {
+        prte_setenv("OMPI_MCA_ompi_comm_method", "mpi_finalize", 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 */