/* -----------------------------------------------------------------------------
- * $Id: Main.c,v 1.11 1999/09/16 08:29:01 sof Exp $
*
- * (c) The GHC Team 1998-1999
+ * (c) The GHC Team 1998-2000
*
* Main function for a standalone Haskell program.
*
#define COMPILING_RTS_MAIN
+#include "PosixSource.h"
#include "Rts.h"
#include "RtsAPI.h"
#include "SchedAPI.h"
+#include "Schedule.h"
#include "RtsFlags.h"
-#include "Schedule.h" /* for MainTSO */
#include "RtsUtils.h"
+#include "Prelude.h"
+#include "Task.h"
+#include <stdlib.h>
#ifdef DEBUG
-#include "Printer.h" /* for printing */
+# include "Printer.h" /* for printing */
#endif
-#ifdef INTERPRETER
-#include "Assembler.h"
+#ifdef PAR
+# include "Parallel.h"
+# include "ParallelRts.h"
+# include "LLC.h"
#endif
-#ifdef PAR
-#include "ParInit.h"
-#include "Parallel.h"
-#include "LLC.h"
+#if defined(GRAN) || defined(PAR)
+# include "GranSimRts.h"
#endif
#ifdef HAVE_WINDOWS_H
-#include <windows.h>
+# include <windows.h>
#endif
+extern void __stginit_ZCMain(void);
/* Hack: we assume that we're building a batch-mode system unless
* INTERPRETER is set
*/
-# ifndef INTERPRETER /* Hack */
+#ifndef INTERPRETER /* Hack */
int main(int argc, char *argv[])
{
+ int exit_status;
SchedulerStatus status;
- startupHaskell(argc,argv);
-
-# ifndef PAR
- MainTSO = createIOThread(stg_max(BLOCK_SIZE_W,
- RtsFlags.GcFlags.initialStkSize),
- (StgClosure *)&mainIO_closure);
- status = schedule(MainTSO,NULL);
-# else
+ /* all GranSim/GUM init is done in startupHaskell; sets IAmMainThread! */
+
+ startupHaskell(argc,argv,__stginit_ZCMain);
+
+ /* kick off the computation by creating the main thread with a pointer
+ to mainIO_closure representing the computation of the overall program;
+ then enter the scheduler with this thread and off we go;
+
+ the same for GranSim (we have only one instance of this code)
+
+ in a parallel setup, where we have many instances of this code
+ running on different PEs, we should do this only for the main PE
+ (IAmMainThread is set in startupHaskell)
+ */
+
+# if defined(PAR)
+
+# if defined(DEBUG)
+ { /* a wait loop to allow attachment of gdb to UNIX threads */
+ nat i, j, s;
+
+ for (i=0, s=0; i<(nat)RtsFlags.ParFlags.wait; i++)
+ for (j=0; j<1000000; j++)
+ s += j % 65536;
+ }
+ IF_PAR_DEBUG(verbose,
+ belch("Passed wait loop"));
+# endif
+
if (IAmMainThread == rtsTrue) {
- /*Just to show we're alive */
- fprintf(stderr, "Main Thread Started ...\n");
-
- MainTSO = createIOThread(stg_max(BLOCK_SIZE_W,
- RtsFlags.GcFlags.initialStkSize),
- (StgClosure *)&mainIO_closure);
- status = schedule(MainTSO,NULL);
+ IF_PAR_DEBUG(verbose,
+ debugBelch("==== [%x] Main Thread Started ...\n", mytid));
+
+ /* ToDo: Dump event for the main thread */
+ status = rts_mainLazyIO((HaskellObj)mainIO_closure, NULL);
} else {
- WaitForPEOp(PP_FINISH,SysManTask);
- exit(EXIT_SUCCESS);
+ /* Just to show we're alive */
+ IF_PAR_DEBUG(verbose,
+ debugBelch("== [%x] Non-Main PE enters scheduler via taskStart() without work ...\n",
+ mytid));
+
+ /* all non-main threads enter the scheduler without work */
+ taskStart();
+ status = Success; // declare victory (see shutdownParallelSystem)
}
-# endif /* PAR */
+
+# elif defined(GRAN)
+
+ /* ToDo: Dump event for the main thread */
+ status = rts_mainLazyIO(mainIO_closure, NULL);
+
+# else /* !PAR && !GRAN */
+
+ /* ToDo: want to start with a larger stack size */
+ {
+ void *cap = rts_lock();
+ cap = rts_evalLazyIO(cap,(HaskellObj)(void *)mainIO_closure, NULL);
+ status = rts_getSchedStatus(cap);
+ rts_unlock(cap);
+ }
+
+# endif /* !PAR && !GRAN */
+
+ /* check the status of the entire Haskell computation */
switch (status) {
- case AllBlocked:
- barf("Scheduler stopped, all threads blocked");
- case Deadlock:
- shutdownHaskell();
- barf("No threads to run! Deadlock?");
case Killed:
- belch("%s: warning: main thread killed", prog_argv[0]);
- case Success:
+ errorBelch("main thread exited (uncaught exception)");
+ exit_status = EXIT_KILLED;
+ break;
case Interrupted:
- /* carry on */
+ errorBelch("interrupted");
+ exit_status = EXIT_INTERRUPTED;
+ break;
+ case Success:
+ exit_status = EXIT_SUCCESS;
+ break;
+#if defined(PAR)
+ case NoStatus:
+ errorBelch("main thread PE killed; probably due to failure of another PE; check /tmp/pvml...");
+ exit_status = EXIT_KILLED;
+ break;
+#endif
+ default:
+ barf("main thread completed with invalid status");
}
- shutdownHaskellAndExit(EXIT_SUCCESS);
+ shutdownHaskellAndExit(exit_status);
+ return 0; /* never reached, keep gcc -Wall happy */
}
# endif /* BATCH_MODE */