PMI: allow user overriding KEY VAL length

Fixes #6924.
pmodels · May 22, 2024 · d5adec2 · d5adec2
1 parent 2f98250
commit d5adec2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/util/mpir_pmi.c b/src/util/mpir_pmi.c
@@ -148,6 +148,10 @@ int MPIR_pmi_init(void)
                mpi_errno = pmix_init(&has_parent, &rank, &size, &appnum));
     MPIR_ERR_CHECK(mpi_errno);
 
+    /* allowing user to override default KEY VAL size for certain system setups */
+    MPL_env2int("PMI_MAX_KEY_SIZE", &pmi_max_key_size);
+    MPL_env2int("PMI_MAX_VAL_SIZE", &pmi_max_val_size);
+
     unsigned world_id = 0;
     if (pmi_kvs_name) {
         HASH_FNV(pmi_kvs_name, strlen(pmi_kvs_name), world_id);