X-Git-Url: http://git.megacz.com/?p=ghc-hetmet.git;a=blobdiff_plain;f=rts%2FHpc.c;h=0f3f3b31e3498a225e4f0724e6001035d42c30ac;hp=b1b26082946f46529480d4b855624150d5b08170;hb=80578b7defea703ad24efa908534a13e2ebc880b;hpb=0df7dbb1b98d08521f177ade22615861752f8981 diff --git a/rts/Hpc.c b/rts/Hpc.c index b1b2608..0f3f3b3 100644 --- a/rts/Hpc.c +++ b/rts/Hpc.c @@ -2,14 +2,13 @@ * (c)2006 Galois Connections, Inc. */ -// #include "HsFFI.h" - #include #include #include #include #include #include "HsFFI.h" + #include "Rts.h" #include "Hpc.h" @@ -20,9 +19,12 @@ #define DEBUG_HPC 0 -static int hpc_inited = 0; // Have you started this component? -static FILE *tixFile; // file being read/written -static int tix_ch; // current char +static int hpc_inited = 0; // Have you started this component? +static FILE *tixFile; // file being read/written +static int tix_ch; // current char +static StgWord64 magicTixNumber; // Magic/Hash number to mark .tix files + +static FILE *rixFile = NULL; // The tracer file/pipe typedef struct _Info { char *modName; // name of module @@ -32,6 +34,13 @@ typedef struct _Info { struct _Info *next; } Info; +// This is a cruel hack, we should completely redesign the format specifier handling in the RTS. +#if SIZEOF_LONG == 8 +#define PRIuWORD64 "lu" +#else +#define PRIuWORD64 "llu" +#endif + Info *modules = 0; Info *nextModule = 0; StgWord64 *tixBoxes = 0; // local copy of tixBoxes array, from file. @@ -117,7 +126,7 @@ static void hpc_init(void) { expect('i'); expect('x'); ws(); - expectWord64(); + magicTixNumber = expectWord64(); ws(); expect('['); ws(); @@ -149,7 +158,8 @@ static void hpc_init(void) { if (tix_ch == ',') { expect(','); ws(); - }} + } + } expect(']'); ws(); tixBoxes = (StgWord64 *)calloc(totalTixes,sizeof(StgWord64)); @@ -166,6 +176,9 @@ static void hpc_init(void) { expect(']'); fclose(tixFile); + } else { + // later, we will find a binary specific + magicTixNumber = (StgWord64)0; } } @@ -174,10 +187,11 @@ static void hpc_init(void) { * of the tix file, or all zeros. */ -void +int hs_hpc_module(char *modName,int modCount,StgWord64 *tixArr) { Info *tmpModule, *lastModule; int i; + int offset = 0; #if DEBUG_HPC printf("hs_hpc_module(%s,%d)\n",modName,modCount); @@ -199,7 +213,7 @@ hs_hpc_module(char *modName,int modCount,StgWord64 *tixArr) { for(i=0;i < modCount;i++) { tixArr[i] = tixBoxes[i + tmpModule->tickOffset]; } - return; + return tmpModule->tickOffset; } lastModule = tmpModule; } @@ -227,6 +241,59 @@ hs_hpc_module(char *modName,int modCount,StgWord64 *tixArr) { #if DEBUG_HPC printf("end: hs_hpc_module\n"); #endif + return offset; +} + +static StgThreadID previous_tid = 0; + +static void +send_ThreadId(StgTSO *current_tso) { + // This assumes that there is no real thread 0. + StgThreadID tid = (current_tso == 0) ? 0 : current_tso->id; + if (tid != previous_tid) { + previous_tid = tid; + // How do we print StgWord32's without a cast? + fprintf(rixFile,"Thread Switch %d\n",(unsigned int)tid); + } +} + +/* + * Called on *every* exception thrown + */ +void +hs_hpc_event(char *msg,StgTSO *current_tso) { + // Assumes that we have had at least *one* tick first. + // All exceptions before the first tick are not reported. + // The only time this might be an issue is in bootstrapping code, + // so this is a feature. + + // This is called on *every* exception, even when Hpc is not enabled. + + if (rixFile != NULL) { + assert(hpc_inited != 0); + send_ThreadId(current_tso); + fprintf(rixFile,"%s\n",msg); + } +} + +/* Called on every tick, dynamically to our file record of program execution + */ + +void +hs_hpc_tick(int globIx, StgTSO *current_tso) { +#if DEBUG_HPC && DEBUG + printf("hs_hpc_tick(%d)\n",globIx); +#endif + assert(hpc_inited != 0); + if (rixFile != NULL) { + send_ThreadId(current_tso); + fprintf(rixFile,"%d\n",globIx); + } + +#if DEBUG_HPC + printf("end: hs_hpc_tick\n"); +#endif + } /* This is called after all the modules have registered their local tixboxes, @@ -236,6 +303,7 @@ hs_hpc_module(char *modName,int modCount,StgWord64 *tixArr) { void startupHpc(void) { Info *tmpModule; + char *hpcRix; #if DEBUG_HPC printf("startupHpc\n"); #endif @@ -256,8 +324,46 @@ startupHpc(void) { } } } + + // HPCRIX contains the name of the file to send our dynamic runtime output to. + // This might be a real file, or perhaps a named pipe. + hpcRix = getenv("HPCRIX"); + if (hpcRix) { + int comma; + Info *tmpModule; + + assert(hpc_inited); + + rixFile = fopen(hpcRix,"w"); + + comma = 0; + + fprintf(rixFile,"Starting %s\n",prog_name); + fprintf(rixFile,"["); + tmpModule = modules; + for(;tmpModule != 0;tmpModule = tmpModule->next) { + if (comma) { + fprintf(rixFile,","); + } else { + comma = 1; + } + fprintf(rixFile,"(\"%s\",%u)", + tmpModule->modName, + tmpModule->tickCount); +#if DEBUG_HPC + fprintf(stderr,"(tracer)%s: %u (offset=%u)\n", + tmpModule->modName, + tmpModule->tickCount, + tmpModule->tickOffset); +#endif + } + fprintf(rixFile,"]\n"); + fflush(rixFile); + } + } + /* Called at the end of execution, to write out the Hpc *.tix file * for this exection. Safe to call, even if coverage is not used. */ @@ -278,7 +384,7 @@ exitHpc(void) { comma = 0; - fprintf(f,"Tix 0 ["); + fprintf(f,"Tix %" PRIuWORD64 " [", magicTixNumber); tmpModule = modules; for(;tmpModule != 0;tmpModule = tmpModule->next) { if (comma) { @@ -286,11 +392,11 @@ exitHpc(void) { } else { comma = 1; } - fprintf(f,"(\"%s\",%d)", + fprintf(f,"(\"%s\",%u)", tmpModule->modName, tmpModule->tickCount); #if DEBUG_HPC - fprintf(stderr,"%s: %d (offset=%d)\n", + fprintf(stderr,"%s: %u (offset=%u)\n", tmpModule->modName, tmpModule->tickCount, tmpModule->tickOffset); @@ -314,7 +420,7 @@ exitHpc(void) { } if (tmpModule->tixArr) { - fprintf(f,"%lld",tmpModule->tixArr[i]); + fprintf(f,"%" PRIuWORD64,tmpModule->tixArr[i]); } else { fprintf(f,"0"); } @@ -324,6 +430,11 @@ exitHpc(void) { fprintf(f,"]\n"); fclose(f); + + if (rixFile != NULL) { + fprintf(rixFile,"Finished\n"); + fclose(rixFile); + } }