From 4e66b4b3837ac5e511d359dada54ff6c9e42ed58 Mon Sep 17 00:00:00 2001
From: Artem Polyakov <artpol84@gmail.com>
Date: Wed, 9 Aug 2017 02:02:27 -0700
Subject: [PATCH] mpi/pmix: Addittional UCX settings to ensure safe fork()

Signed-off-by: Artem Polyakov <artpol84@gmail.com>
---
 src/plugins/mpi/pmix/pmixp_dconn_ucx.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/plugins/mpi/pmix/pmixp_dconn_ucx.c b/src/plugins/mpi/pmix/pmixp_dconn_ucx.c
index 783f845cd53..d0800f32ae6 100644
--- a/src/plugins/mpi/pmix/pmixp_dconn_ucx.c
+++ b/src/plugins/mpi/pmix/pmixp_dconn_ucx.c
@@ -156,7 +156,11 @@ static int _load_ucx_lib()
 	 * To avoid that we need to disable memory hooks before
 	 * loading UCX library.
 	 */
+	setenv("UCX_MEM_MMAP_RELOC", "no", 1);
 	setenv("UCX_MEM_MALLOC_HOOKS", "no", 1);
+	setenv("UCX_MEM_MALLOC_RELOC", "no", 1);
+	setenv("UCX_MEM_EVENTS", "no", 1);
+	setenv("UCX_IB_FORK_INIT", "no", 1);
 
 #ifdef PMIXP_UCX_LIBPATH
 	/* If this SLURM build doesn't allow RPATH's
-- 
GitLab