AMPI: fix exit code handling for bigsim, multicore, ucx, uth (#2478)

Author: matthiasdiener

src/arch/ucx/machine.C

@@ -654,7 +654,7 @@ void LrtsExit(int exitcode)
 
         ret = runtime_fini();
         UCX_CHECK_PMI_RET(ret, "runtime_fini");
-        exit(0);
+        exit(exitcode);
     }
 }
 

src/arch/uth/machine.C

@@ -71,6 +71,7 @@ void CmiReleaseCommHandle(CmiCommHandle c)
 }
 
 /********************* CONTEXT-SWITCHING FUNCTIONS ******************/
+static int _exitcode = 0;
 
 static void CmiNext(void)
 {
@@ -81,7 +82,7 @@ static void CmiNext(void)
     t = CmiThreads[index];
     if ((t)&&(!CmiBarred[index])) break;
     index = (index+1) % CmiNumPes();
-    if (index == orig) exit(0);
+    if (index == orig) exit(_exitcode);
   }
   _Cmi_mype = index;
   CthResume(t);
@@ -314,6 +315,7 @@ void ConverseExit(int exitcode)
 {
   ConverseCommonExit();
   CmiThreads[CmiMyPe()] = 0;
+  _exitcode = exitcode; // Used in CmiNext()
   CmiNext();
 }
 

src/arch/util/machine-common-core.C

@@ -1716,7 +1716,7 @@ if (MSG_STATISTIC)
     if (CmiMyRank() == 0) CmiExitXpmem();
     CmiNodeAllBarrier();
 #endif
-    if (CmiMyRank() == 0) LrtsExit();
+    if (CmiMyRank() == 0) LrtsExit(exitcode);
 #endif
     CmiYield();
     if (!CharmLibInterOperate || userDrivenMode) {

tests/ampi/Makefile

@@ -18,6 +18,7 @@ endif
 NOBGDIRS = \
   intercomm_coll \
   privatization \
+  exit \
   datatype \
 
 FTDIRS = \