Fixes the GAE fork issue with openmpi (#2820)

## Related issue number Fixes #2812 Signed-off-by: Tao He <linzhu.ht@alibaba-inc.com>
alibaba · Jun 6, 2023 · 99c9d15 · 99c9d15
1 parent ba35996
commit 99c9d15
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/analytical_engine/core/grape_engine.cc b/analytical_engine/core/grape_engine.cc
@@ -199,6 +199,9 @@ class RedirectLogSink : public google::LogSink {
 };
 
 int main(int argc, char* argv[]) {
+  // set RDMAV_FORK_SAFE to avoid the openmpi error, see also #2812
+  setenv("RDMAV_FORK_SAFE", "1", 0);
+
   int exit_code = 0;
   // not output any log to stderr by glog.
   FLAGS_stderrthreshold = std::numeric_limits<int>::max();