Try again at implementing thread-safe lazy pass initialization, without depending on static local initialization

being threadsafe AND ensuring that initialization is complete by the time the initializeFooPass method returns.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@116492 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 72 insertions, 10 deletions

diff --git a/include/llvm/PassSupport.h b/include/llvm/PassSupport.h
index e0e4e8a613..58527264b5 100644
--- a/include/llvm/PassSupport.h
+++ b/include/llvm/PassSupport.h
@@ -24,7 +24,7 @@
 #include "Pass.h"
 #include "llvm/PassRegistry.h"
 #include "llvm/InitializePasses.h"
-#include "llvm/Support/Compiler.h"
+#include "llvm/System/Atomic.h"
 #include <vector>
 
 namespace llvm {
@@ -137,8 +137,20 @@ private:
     return PI; \
   } \
   void llvm::initialize##passName##Pass(PassRegistry &Registry) { \
-    ATTRIBUTE_USED \
-    static void* initialized = initialize##passName##PassOnce(Registry); \
+    static volatile sys::cas_flag initialized = 0; \
+    sys::cas_flag old_val = sys::CompareAndSwap(&initialized, 1, 0); \
+    if (old_val == 0) { \
+      initialize##passName##PassOnce(Registry); \
+      sys::MemoryFence(); \
+      initialized = 2; \
+    } else { \
+      sys::cas_flag tmp = initialized; \
+      sys::MemoryFence(); \
+      while (tmp != 2) { \
+        tmp = initialized; \
+        sys::MemoryFence(); \
+      } \
+    } \
   } \
   static RegisterPass<passName> passName ## _info(arg, name, cfg, analysis);
 
@@ -157,8 +169,20 @@ private:
     return PI; \
   } \
   void llvm::initialize##passName##Pass(PassRegistry &Registry) { \
-    ATTRIBUTE_USED \
-    static void* initialized = initialize##passName##PassOnce(Registry); \
+    static volatile sys::cas_flag initialized = 0; \
+    sys::cas_flag old_val = sys::CompareAndSwap(&initialized, 1, 0); \
+    if (old_val == 0) { \
+      initialize##passName##PassOnce(Registry); \
+      sys::MemoryFence(); \
+      initialized = 2; \
+    } else { \
+      sys::cas_flag tmp = initialized; \
+      sys::MemoryFence(); \
+      while (tmp != 2) { \
+        tmp = initialized; \
+        sys::MemoryFence(); \
+      } \
+    } \
   } \
   static RegisterPass<passName> passName ## _info(arg, name, cfg, analysis);
 
@@ -244,11 +268,24 @@ struct RegisterAnalysisGroup : public RegisterAGBase {
     return AI; \
   } \
   void llvm::initialize##agName##AnalysisGroup(PassRegistry &Registry) { \
-    ATTRIBUTE_USED static void* initialized = \
+    static volatile sys::cas_flag initialized = 0; \
+    sys::cas_flag old_val = sys::CompareAndSwap(&initialized, 1, 0); \
+    if (old_val == 0) { \
       initialize##agName##AnalysisGroupOnce(Registry); \
+      sys::MemoryFence(); \
+      initialized = 2; \
+    } else { \
+      sys::cas_flag tmp = initialized; \
+      sys::MemoryFence(); \
+      while (tmp != 2) { \
+        tmp = initialized; \
+        sys::MemoryFence(); \
+      } \
+    } \
   } \
   static RegisterAnalysisGroup<agName> agName##_info (name);
 
+
 #define INITIALIZE_AG_PASS(passName, agName, arg, name, cfg, analysis, def) \
   static void* initialize##passName##PassOnce(PassRegistry &Registry) { \
     PassInfo *PI = new PassInfo(name, arg, & passName ::ID, \
@@ -260,12 +297,25 @@ struct RegisterAnalysisGroup : public RegisterAGBase {
     return AI; \
   } \
   void llvm::initialize##passName##Pass(PassRegistry &Registry) { \
-    ATTRIBUTE_USED \
-    static void* initialized = initialize##passName##PassOnce(Registry); \
+    static volatile sys::cas_flag initialized = 0; \
+    sys::cas_flag old_val = sys::CompareAndSwap(&initialized, 1, 0); \
+    if (old_val == 0) { \
+      initialize##passName##PassOnce(Registry); \
+      sys::MemoryFence(); \
+      initialized = 2; \
+    } else { \
+      sys::cas_flag tmp = initialized; \
+      sys::MemoryFence(); \
+      while (tmp != 2) { \
+        tmp = initialized; \
+        sys::MemoryFence(); \
+      } \
+    } \
   } \
   static RegisterPass<passName> passName ## _info(arg, name, cfg, analysis); \
   static RegisterAnalysisGroup<agName, def> passName ## _ag(passName ## _info);
 
+
 #define INITIALIZE_AG_PASS_BEGIN(passName, agName, arg, n, cfg, analysis, def) \
   static void* initialize##passName##PassOnce(PassRegistry &Registry) {
 
@@ -279,8 +329,20 @@ struct RegisterAnalysisGroup : public RegisterAGBase {
     return AI; \
   } \
   void llvm::initialize##passName##Pass(PassRegistry &Registry) { \
-    ATTRIBUTE_USED \
-    static void* initialized = initialize##passName##PassOnce(Registry); \
+    static volatile sys::cas_flag initialized = 0; \
+    sys::cas_flag old_val = sys::CompareAndSwap(&initialized, 1, 0); \
+    if (old_val == 0) { \
+      initialize##passName##PassOnce(Registry); \
+      sys::MemoryFence(); \
+      initialized = 2; \
+    } else { \
+      sys::cas_flag tmp = initialized; \
+      sys::MemoryFence(); \
+      while (tmp != 2) { \
+        tmp = initialized; \
+        sys::MemoryFence(); \
+      } \
+    } \
   } \
   static RegisterPass<passName> passName ## _info(arg, n, cfg, analysis); \
   static RegisterAnalysisGroup<agName, def> passName ## _ag(passName ## _info);