X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Frts%2FRtsStartup.c;h=305c6eb66d7097641e05f90b90cfd526051e42fb;hb=d381d2096a5386031a62d9826b9cd57901fce68d;hp=c8cc08420cf264c3bbbdb7f63b2fc3a66120cf87;hpb=50a70f642ca958cbb2dec99b0b0ae67120c9ffa9;p=ghc-hetmet.git diff --git a/ghc/rts/RtsStartup.c b/ghc/rts/RtsStartup.c index c8cc084..305c6eb 100644 --- a/ghc/rts/RtsStartup.c +++ b/ghc/rts/RtsStartup.c @@ -1,5 +1,5 @@ /* ----------------------------------------------------------------------------- - * $Id: RtsStartup.c,v 1.11 1999/05/04 10:19:19 sof Exp $ + * $Id: RtsStartup.c,v 1.28 2000/02/07 20:57:58 simonmar Exp $ * * (c) The GHC Team, 1998-1999 * @@ -15,16 +15,22 @@ #include "StablePriv.h" /* initStablePtrTable */ #include "Schedule.h" /* initScheduler */ #include "Stats.h" /* initStats */ +#include "Signals.h" +#include "Itimer.h" #include "Weak.h" #include "Ticky.h" -#if defined(PROFILING) +#if defined(PROFILING) || defined(DEBUG) # include "ProfRts.h" -#elif defined(DEBUG) -# include "DebugProf.h" +# include "ProfHeap.h" #endif -#ifdef PAR +#if defined(GRAN) +#include "GranSimRts.h" +#include "ParallelRts.h" +#endif + +#if defined(PAR) #include "ParInit.h" #include "Parallel.h" #include "LLC.h" @@ -35,10 +41,17 @@ */ struct RTS_FLAGS RtsFlags; -extern void startupHaskell(int argc, char *argv[]) +static int rts_has_started_up = 0; +#if defined(PAR) +static ullong startTime = 0; +#endif + +void +startupHaskell(int argc, char *argv[]) { - static int rts_has_started_up = 0; +#ifdef ENABLE_WIN32_DLL_SUPPORT int i; +#endif /* To avoid repeated initialisations of the RTS */ if (rts_has_started_up) @@ -46,10 +59,6 @@ extern void startupHaskell(int argc, char *argv[]) else rts_has_started_up=1; -#if defined(PAR) - int nPEs = 0; /* Number of PEs */ -#endif - /* The very first thing we do is grab the start time...just in case we're * collecting timing statistics. */ @@ -57,13 +66,15 @@ extern void startupHaskell(int argc, char *argv[]) #ifdef PAR /* - *The parallel system needs to be initialised and synchronised before - *the program is run. + * The parallel system needs to be initialised and synchronised before + * the program is run. */ + fprintf(stderr, "startupHaskell: argv[0]=%s\n", argv[0]); if (*argv[0] == '-') { /* Look to see whether we're the Main Thread */ IAmMainThread = rtsTrue; argv++; argc--; /* Strip off flag argument */ -/* fprintf(stderr, "I am Main Thread\n"); */ + // IF_PAR_DEBUG(verbose, + fprintf(stderr, "[%x] I am Main Thread\n", mytid); } /* * Grab the number of PEs out of the argument vector, and @@ -73,7 +84,6 @@ extern void startupHaskell(int argc, char *argv[]) argv[1] = argv[0]; argv++; argc--; initEachPEHook(); /* HWL: hook to be execed on each PE */ - SynchroniseSystem(); #endif /* Set the RTS flags to default values. */ @@ -88,14 +98,26 @@ extern void startupHaskell(int argc, char *argv[]) prog_argv = argv; #if defined(PAR) - /* Initialise the parallel system -- before initHeap! */ - initParallelSystem(); - /* And start GranSim profiling if required: omitted for now - *if (Rtsflags.ParFlags.granSimStats) - *init_gr_profiling(rts_argc, rts_argv, prog_argc, prog_argv); - */ + /* NB: this really must be done after processing the RTS flags */ + fprintf(stderr, "Synchronising system (%d PEs)\n", nPEs); + SynchroniseSystem(); // calls initParallelSystem etc #endif /* PAR */ + /* initialise scheduler data structures (needs to be done before + * initStorage()). + */ + initScheduler(); + +#if defined(GRAN) + /* And start GranSim profiling if required: */ + if (RtsFlags.GranFlags.GranSimStats.Full) + init_gr_simulation(rts_argc, rts_argv, prog_argc, prog_argv); +#elif defined(PAR) + /* And start GUM profiling if required: */ + if (RtsFlags.ParFlags.ParStats.Full) + init_gr_simulation(rts_argc, rts_argv, prog_argc, prog_argv); +#endif /* PAR || GRAN */ + /* initialize the storage manager */ initStorage(); @@ -106,14 +128,23 @@ extern void startupHaskell(int argc, char *argv[]) initProfiling(); #endif - /* Initialise the scheduler */ - initScheduler(); + /* start the ticker */ + install_vtalrm_handler(); + initialize_virtual_timer(TICK_MILLISECS); + + /* start our haskell execution tasks */ +#ifdef SMP + startTasks(); +#endif /* Initialise the stats department */ initStats(); -#if 0 +#if !defined(mingw32_TARGET_OS) && !defined(PAR) + /* Initialise the user signal handler set */ initUserSignals(); + /* Set up handler to run on SIGINT */ + init_shutdown_handler(); #endif /* When the RTS and Prelude live in separate DLLs, @@ -122,8 +153,6 @@ extern void startupHaskell(int argc, char *argv[]) filling in the tables with references to where the static info tables have been loaded inside the running process. - - Ditto for Bool closure tbl. */ #ifdef ENABLE_WIN32_DLL_SUPPORT for(i=0;i<=255;i++) @@ -137,19 +166,57 @@ extern void startupHaskell(int argc, char *argv[]) end_init(); } +/* + * Shutting down the RTS - two ways of doing this, one which + * calls exit(), one that doesn't. + * + * (shutdownHaskellAndExit() is called by System.exitWith). + */ +void +shutdownHaskellAndExit(int n) +{ + OnExitHook(); + shutdownHaskell(); + stg_exit(n); +} + void shutdownHaskell(void) { + if (!rts_has_started_up) + return; + + /* start timing the shutdown */ + stat_startExit(); + +#if !defined(GRAN) /* Finalize any remaining weak pointers */ finalizeWeakPointersNow(); +#endif #if defined(GRAN) - #error FixMe. - if (!RTSflags.GranFlags.granSimStats_suppressed) + /* end_gr_simulation prints global stats if requested -- HWL */ + if (!RtsFlags.GranFlags.GranSimStats.Suppressed) end_gr_simulation(); #endif - /* clean up things from the storage manager's point of view */ + /* stop all running tasks */ + exitScheduler(); + + /* stop the ticker */ + initialize_virtual_timer(0); + + /* reset the standard file descriptors to blocking mode */ + resetNonBlockingFd(0); + resetNonBlockingFd(1); + resetNonBlockingFd(2); + + /* stop timing the shutdown, we're about to print stats */ + stat_endExit(); + + /* clean up things from the storage manager's point of view. + * also outputs the stats (+RTS -s) info. + */ exitStorage(); #if defined(PROFILING) || defined(DEBUG) @@ -164,34 +231,25 @@ shutdownHaskell(void) if (RtsFlags.TickyFlags.showTickyStats) PrintTickyInfo(); #endif - /* - This fflush is important, because: if "main" just returns, - then we will end up in pre-supplied exit code that will close - streams and flush buffers. In particular we have seen: it - will close fd 0 (stdin), then flush fd 1 (stdout), then ... - - But if you're playing with sockets, that "close fd 0" might - suggest to the daemon that all is over, only to be presented - with more stuff on "fd 1" at the flush. - - The fflush avoids this sad possibility. - */ - fflush(stdout); -} + rts_has_started_up=0; + +#if defined(PAR) + shutdownParallelSystem(0); +#endif +} /* - * called from STG-land to exit the program cleanly + * called from STG-land to exit the program */ void stg_exit(I_ n) { -#ifdef PAR +#if 0 /* def PAR */ par_exit(n); #else - OnExitHook(); exit(n); #endif } +