From: simonmar <unknown>
Date: Mon, 25 Apr 2005 14:13:23 +0000 (+0000)
Subject: [project @ 2005-04-25 14:13:23 by simonmar]
X-Git-Tag: Initial_conversion_from_CVS_complete~675
X-Git-Url: http://git.megacz.com/?a=commitdiff_plain;h=e343c124fcf6a76e2b85020a330f23b63506a162;p=ghc-hetmet.git

[project @ 2005-04-25 14:13:23 by simonmar]
atomicModifyMutVarzh_fast: move the sm_mutex acquisition *after* the
heap check
---

diff --git a/ghc/rts/PrimOps.cmm b/ghc/rts/PrimOps.cmm
index 2d306f6..7cb1aaf 100644
--- a/ghc/rts/PrimOps.cmm
+++ b/ghc/rts/PrimOps.cmm
@@ -166,10 +166,6 @@ atomicModifyMutVarzh_fast
     W_ mv, z, x, y, r;
     /* Args: R1 :: MutVar#,  R2 :: a -> (a,b) */
 
-#if defined(SMP)
-    foreign "C" ACQUIRE_LOCK(sm_mutex "ptr");
-#endif
-
     /* If x is the current contents of the MutVar#, then 
        We want to make the new contents point to
 
@@ -206,6 +202,10 @@ atomicModifyMutVarzh_fast
 
    HP_CHK_GEN_TICKY(SIZE, R1_PTR & R2_PTR, atomicModifyMutVarzh_fast);
 
+#if defined(SMP)
+    foreign "C" ACQUIRE_LOCK(sm_mutex "ptr");
+#endif
+
    x = StgMutVar_var(R1);
 
    TICK_ALLOC_THUNK_2();
@@ -1429,6 +1429,10 @@ takeMVarzh_fast
 {
     W_ mvar, val, info, tso;
 
+#if defined(SMP)
+    foreign "C" ACQUIRE_LOCK(sm_mutex "ptr");
+#endif
+
     /* args: R1 = MVar closure */
     mvar = R1;
 
@@ -1448,6 +1452,10 @@ takeMVarzh_fast
 	StgTSO_block_info(CurrentTSO)  = mvar;
 	StgMVar_tail(mvar) = CurrentTSO;
 	
+#if defined(SMP)
+	foreign "C" RELEASE_LOCK(sm_mutex "ptr");
+#endif
+
 	jump stg_block_takemvar;
   }
 
@@ -1476,6 +1484,11 @@ takeMVarzh_fast
       if (StgMVar_head(mvar) == stg_END_TSO_QUEUE_closure) {
 	  StgMVar_tail(mvar) = stg_END_TSO_QUEUE_closure;
       }
+
+#if defined(SMP)
+      foreign "C" RELEASE_LOCK(sm_mutex "ptr");
+#endif
+
       RET_P(val);
   } 
   else
@@ -1487,6 +1500,11 @@ takeMVarzh_fast
        */
       SET_INFO(mvar,stg_EMPTY_MVAR_info);
       StgMVar_value(mvar) = stg_END_TSO_QUEUE_closure;
+
+#if defined(SMP)
+      foreign "C" RELEASE_LOCK(sm_mutex "ptr");
+#endif
+
       RET_P(val);
   }
 }
@@ -1496,6 +1514,10 @@ tryTakeMVarzh_fast
 {
     W_ mvar, val, info, tso;
 
+#if defined(SMP)
+    foreign "C" ACQUIRE_LOCK(sm_mutex "ptr");
+#endif
+
     /* args: R1 = MVar closure */
 
     mvar = R1;
@@ -1506,6 +1528,9 @@ tryTakeMVarzh_fast
 	/* HACK: we need a pointer to pass back, 
 	 * so we abuse NO_FINALIZER_closure
 	 */
+#if defined(SMP)
+       foreign "C" RELEASE_LOCK(sm_mutex "ptr");
+#endif
 	RET_NP(0, stg_NO_FINALIZER_closure);
     }
 
@@ -1539,13 +1564,13 @@ tryTakeMVarzh_fast
     {
 	/* No further putMVars, MVar is now empty */
 	StgMVar_value(mvar) = stg_END_TSO_QUEUE_closure;
-	
-	/* do this last... we might have locked the MVar in the SMP case,
-	 * and writing the info pointer will unlock it.
-	 */
 	SET_INFO(mvar,stg_EMPTY_MVAR_info);
     }
     
+#if defined(SMP)
+    foreign "C" RELEASE_LOCK(sm_mutex "ptr");
+#endif
+
     RET_NP(1, val);
 }
 
@@ -1554,6 +1579,10 @@ putMVarzh_fast
 {
     W_ mvar, info, tso;
 
+#if defined(SMP)
+    foreign "C" ACQUIRE_LOCK(sm_mutex "ptr");
+#endif
+
     /* args: R1 = MVar, R2 = value */
     mvar = R1;
 
@@ -1570,6 +1599,9 @@ putMVarzh_fast
 	StgTSO_block_info(CurrentTSO)  = mvar;
 	StgMVar_tail(mvar) = CurrentTSO;
 	
+#if defined(SMP)
+	foreign "C" RELEASE_LOCK(sm_mutex "ptr");
+#endif
 	jump stg_block_putmvar;
     }
   
@@ -1595,6 +1627,9 @@ putMVarzh_fast
 	    StgMVar_tail(mvar) = stg_END_TSO_QUEUE_closure;
 	}
 
+#if defined(SMP)
+	foreign "C" RELEASE_LOCK(sm_mutex "ptr");
+#endif
 	jump %ENTRY_CODE(Sp(0));
     }
     else
@@ -1603,6 +1638,10 @@ putMVarzh_fast
 	StgMVar_value(mvar) = R2;
 	/* unlocks the MVar in the SMP case */
 	SET_INFO(mvar,stg_FULL_MVAR_info);
+
+#if defined(SMP)
+	foreign "C" RELEASE_LOCK(sm_mutex "ptr");
+#endif
 	jump %ENTRY_CODE(Sp(0));
     }
     
@@ -1614,12 +1653,19 @@ tryPutMVarzh_fast
 {
     W_ mvar, info, tso;
 
+#if defined(SMP)
+    foreign "C" ACQUIRE_LOCK(sm_mutex "ptr");
+#endif
+
     /* args: R1 = MVar, R2 = value */
     mvar = R1;
 
     info = GET_INFO(mvar);
 
     if (info == stg_FULL_MVAR_info) {
+#if defined(SMP)
+	foreign "C" RELEASE_LOCK(sm_mutex "ptr");
+#endif
 	RET_N(0);
     }
   
@@ -1645,6 +1691,9 @@ tryPutMVarzh_fast
 	    StgMVar_tail(mvar) = stg_END_TSO_QUEUE_closure;
 	}
 
+#if defined(SMP)
+	foreign "C" RELEASE_LOCK(sm_mutex "ptr");
+#endif
 	jump %ENTRY_CODE(Sp(0));
     }
     else
@@ -1653,6 +1702,9 @@ tryPutMVarzh_fast
 	StgMVar_value(mvar) = R2;
 	/* unlocks the MVar in the SMP case */
 	SET_INFO(mvar,stg_FULL_MVAR_info);
+#if defined(SMP)
+	foreign "C" RELEASE_LOCK(sm_mutex "ptr");
+#endif
 	jump %ENTRY_CODE(Sp(0));
     }