[project @ 2003-08-19 16:32:23 by simonmar]

[ghc-hetmet.git] / ghc / rts / Capability.h

diff --git a/ghc/rts/Capability.h b/ghc/rts/Capability.h
index 35f1ad4..dd6a7be 100644 (file)
--- a/ghc/rts/Capability.h
+++ b/ghc/rts/Capability.h
@@ -24,28 +24,49 @@
 #include "RtsFlags.h"
 /* ToDo: assume that RtsFlags.h has been included at usage sites of Capability.h? */
 
-extern void initCapability(Capability* cap);
-#if defined(SMP)
-extern nat rts_n_free_capabilities;  /* total number of available capabilities */
+#if !defined(SMP)
+extern Capability MainCapability;
+#endif
 
-static inline nat getFreeCapabilities()
+extern void initCapabilities(void);
+extern void grabCapability(Capability** pCap);
+extern void releaseCapability(Capability* cap);
+
+extern nat rts_n_free_capabilities;  
+#if defined(RTS_SUPPORTS_THREADS)
+/* number of worker threads waiting for a return capability
+ */
+extern nat rts_n_waiting_workers;
+
+extern void grabReturnCapability(Mutex* pMutex, Capability** pCap);
+extern void yieldToReturningWorker(Mutex* pMutex, Capability** pCap);
+extern void waitForWorkCapability(Mutex* pMutex, Capability** pCap, rtsBool runnable);
+
+
+static inline rtsBool needToYieldToReturningWorker(void)
+{
+       return rts_n_waiting_workers > 0;
+}
+
+static inline nat getFreeCapabilities (void)
 {
   return rts_n_free_capabilities;
 }
 
-static inline rtsBool noFreeCapabilities()
+static inline rtsBool noCapabilities (void)
 {
   return (rts_n_free_capabilities == 0);
 }
 
-static inline rtsBool allFreeCapabilities()
+static inline rtsBool allFreeCapabilities (void)
 {
+# if defined(SMP)
   return (rts_n_free_capabilities == RtsFlags.ParFlags.nNodes);
+# else
+  return (rts_n_free_capabilities == 1);
+# endif 
 }
 
-extern void initCapabilities(nat n);
-extern void grabCapability(Capability** cap);
-extern void releaseCapability(Capability** cap);
-#endif /* SMP */
+#endif /* RTS_SUPPORTS_THREADS */
 
 #endif /* __CAPABILITY_H__ */