Add a new primop: getCCCS# :: State# s -> (# State# s, Addr# #)
authorSimon Marlow <marlowsd@gmail.com>
Tue, 29 Nov 2011 14:05:17 +0000 (14:05 +0000)
committerSimon Marlow <marlowsd@gmail.com>
Tue, 29 Nov 2011 14:22:28 +0000 (14:22 +0000)
Returns a pointer to the current cost-centre stack when profiling,
NULL otherwise.

compiler/codeGen/CgPrimOp.hs
compiler/codeGen/StgCmmPrim.hs
compiler/prelude/primops.txt.pp
rts/Capability.c

index 6fe934b..72bbf6c 100644 (file)
@@ -155,6 +155,8 @@ emitPrimOp [res] SparkOp [arg] live = do
   where
        newspark = CmmLit (CmmLabel (mkCmmCodeLabel rtsPackageId (fsLit "newSpark")))
 
+emitPrimOp [res] GetCCCSOp [] _live
+   = stmtC (CmmAssign (CmmLocal res) curCCS)
 
 emitPrimOp [res] ReadMutVarOp [mutv] _
    = stmtC (CmmAssign (CmmLocal res) (cmmLoadIndexW mutv fixedHdrSize gcWord))
index f8cc425..d546c38 100644 (file)
@@ -228,6 +228,9 @@ emitPrimOp [res] SparkOp [arg]
             [(CmmReg (CmmGlobal BaseReg), AddrHint), ((CmmReg (CmmLocal tmp)), AddrHint)]
         emit (mkAssign (CmmLocal res) (CmmReg (CmmLocal tmp)))
 
+emitPrimOp [res] GetCCCSOp []
+   = emit (mkAssign (CmmLocal res) curCCS)
+
 emitPrimOp [res] ReadMutVarOp [mutv]
    = emit (mkAssign (CmmLocal res) (cmmLoadIndexW mutv fixedHdrSize gcWord))
 
index 204dce2..ceb9226 100644 (file)
@@ -1799,6 +1799,9 @@ primop  TraceCcsOp "traceCcs#" GenPrimOp
    has_side_effects = True
    out_of_line = True
 
+primop  GetCCCSOp "getCCCS#" GenPrimOp
+   State# s -> (# State# s, Addr# #)
+
 ------------------------------------------------------------------------
 section "Etc" 
        {Miscellaneous built-ins}
index fd9f64f..6c84d1e 100644 (file)
@@ -267,6 +267,8 @@ initCapability( Capability *cap, nat i )
 
 #ifdef PROFILING
     cap->r.rCCCS = CCS_SYSTEM;
+#else
+    cap->r.rCCCS = NULL;
 #endif
 
     traceCapsetAssignCap(CAPSET_OSPROCESS_DEFAULT, i);
@@ -614,6 +616,7 @@ waitForReturnCapability (Capability **pCap, Task *task)
 #ifdef PROFILING
     cap->r.rCCCS = CCS_SYSTEM;
 #endif
+
     ASSERT_FULL_CAPABILITY_INVARIANTS(cap,task);
 
     debugTrace(DEBUG_sched, "resuming capability %d", cap->no);