[llvm] r325686 - [IRMover] Implement name based structure type mapping

Eugene Leviant via llvm-commits llvm-commits at lists.llvm.org
Wed Feb 21 07:13:48 PST 2018


Author: evgeny777
Date: Wed Feb 21 07:13:48 2018
New Revision: 325686

URL: http://llvm.org/viewvc/llvm-project?rev=325686&view=rev
Log:
[IRMover] Implement name based structure type mapping

Differential revision: https://reviews.llvm.org/D43199

Added:
    llvm/trunk/test/Linker/Inputs/struct-mapping.ll
    llvm/trunk/test/Linker/struct-mapping.ll
Modified:
    llvm/trunk/include/llvm/Linker/IRMover.h
    llvm/trunk/lib/Linker/IRMover.cpp
    llvm/trunk/tools/llvm-link/llvm-link.cpp

Modified: llvm/trunk/include/llvm/Linker/IRMover.h
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/include/llvm/Linker/IRMover.h?rev=325686&r1=325685&r2=325686&view=diff
==============================================================================
--- llvm/trunk/include/llvm/Linker/IRMover.h (original)
+++ llvm/trunk/include/llvm/Linker/IRMover.h Wed Feb 21 07:13:48 2018
@@ -49,17 +49,23 @@ public:
     // The set of opaque types is the composite module.
     DenseSet<StructType *> OpaqueStructTypes;
 
-    // The set of identified but non opaque structures in the composite module.
-    DenseSet<StructType *, StructTypeKeyInfo> NonOpaqueStructTypes;
-
-  public:
-    void addNonOpaque(StructType *Ty);
-    void switchToNonOpaque(StructType *Ty);
-    void addOpaque(StructType *Ty);
-    StructType *findNonOpaque(ArrayRef<Type *> ETypes, bool IsPacked);
-    bool hasType(StructType *Ty);
-  };
-
+    // The set of identified but non opaque structures in the composite module.
+    DenseSet<StructType *, StructTypeKeyInfo> NonOpaqueStructTypes;
+
+    // Map between structure type name and instance. Used in findNonOpaque
+    // to correctly map imported global variable type during ThinLTO import
+    // phase.
+    DenseMap<StringRef, StructType *> NonOpaqueStructNameMap;
+
+  public:
+    void addNonOpaque(StructType *Ty);
+    void switchToNonOpaque(StructType *Ty);
+    void addOpaque(StructType *Ty);
+    StructType *findNonOpaque(ArrayRef<Type *> ETypes, bool IsPacked,
+                              StringRef Name);
+    bool hasType(StructType *Ty);
+  };
+
   IRMover(Module &M);
 
   typedef std::function<void(GlobalValue &)> ValueAdder;

Modified: llvm/trunk/lib/Linker/IRMover.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Linker/IRMover.cpp?rev=325686&r1=325685&r2=325686&view=diff
==============================================================================
--- llvm/trunk/lib/Linker/IRMover.cpp (original)
+++ llvm/trunk/lib/Linker/IRMover.cpp Wed Feb 21 07:13:48 2018
@@ -318,8 +318,8 @@ Type *TypeMapTy::get(Type *Ty, SmallPtrS
       return *Entry = Ty;
     }
 
-    if (StructType *OldT =
-            DstStructTypesSet.findNonOpaque(ElementTypes, IsPacked)) {
+    if (StructType *OldT = DstStructTypesSet.findNonOpaque(
+            ElementTypes, IsPacked, STy->getName())) {
       STy->setName("");
       return *Entry = OldT;
     }
@@ -906,7 +906,6 @@ bool IRLinker::shouldLink(GlobalValue *D
 Expected<Constant *> IRLinker::linkGlobalValueProto(GlobalValue *SGV,
                                                     bool ForAlias) {
   GlobalValue *DGV = getLinkedToGlobal(SGV);
-
   bool ShouldLink = shouldLink(DGV, *SGV);
 
   // just missing from map
@@ -1410,12 +1409,14 @@ bool IRMover::StructTypeKeyInfo::isEqual
 
 void IRMover::IdentifiedStructTypeSet::addNonOpaque(StructType *Ty) {
   assert(!Ty->isOpaque());
+  if (Ty->hasName())
+    NonOpaqueStructNameMap.insert({getTypeNamePrefix(Ty->getName()), Ty});
+
   NonOpaqueStructTypes.insert(Ty);
 }
 
 void IRMover::IdentifiedStructTypeSet::switchToNonOpaque(StructType *Ty) {
-  assert(!Ty->isOpaque());
-  NonOpaqueStructTypes.insert(Ty);
+  addNonOpaque(Ty);
   bool Removed = OpaqueStructTypes.erase(Ty);
   (void)Removed;
   assert(Removed);
@@ -1428,10 +1429,16 @@ void IRMover::IdentifiedStructTypeSet::a
 
 StructType *
 IRMover::IdentifiedStructTypeSet::findNonOpaque(ArrayRef<Type *> ETypes,
-                                                bool IsPacked) {
+                                                bool IsPacked, StringRef Name) {
   IRMover::StructTypeKeyInfo::KeyTy Key(ETypes, IsPacked);
   auto I = NonOpaqueStructTypes.find_as(Key);
-  return I == NonOpaqueStructTypes.end() ? nullptr : *I;
+  if (I == NonOpaqueStructTypes.end())
+    return nullptr;
+  auto NI = NonOpaqueStructNameMap.find(getTypeNamePrefix(Name));
+  if (NI != NonOpaqueStructNameMap.end() &&
+      IRMover::StructTypeKeyInfo::KeyTy((*NI).second) == Key)
+    return (*NI).second;
+  return *I;
 }
 
 bool IRMover::IdentifiedStructTypeSet::hasType(StructType *Ty) {

Added: llvm/trunk/test/Linker/Inputs/struct-mapping.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/Linker/Inputs/struct-mapping.ll?rev=325686&view=auto
==============================================================================
--- llvm/trunk/test/Linker/Inputs/struct-mapping.ll (added)
+++ llvm/trunk/test/Linker/Inputs/struct-mapping.ll Wed Feb 21 07:13:48 2018
@@ -0,0 +1,4 @@
+%struct.Baz = type { i64, i64, %struct.Foo }
+%struct.Foo = type { i64, i64 }
+
+ at baz = global %struct.Baz zeroinitializer

Added: llvm/trunk/test/Linker/struct-mapping.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/Linker/struct-mapping.ll?rev=325686&view=auto
==============================================================================
--- llvm/trunk/test/Linker/struct-mapping.ll (added)
+++ llvm/trunk/test/Linker/struct-mapping.ll Wed Feb 21 07:13:48 2018
@@ -0,0 +1,12 @@
+; RUN: llvm-link --initial-module=%s %p/Inputs/struct-mapping.ll -S -o - | FileCheck %s
+
+; Here we check that new type mapping algorithm correctly mapped type of internal
+; member of struct.Baz to struct.Foo. Without it we'd map that type to struct.Bar, because
+; it is recursively isomorphic to struct.Foo and is defined first in source file.
+; CHECK: %struct.Baz = type { i64, i64, %struct.Foo }
+
+%struct.Bar = type { i64, i64 }
+%struct.Foo = type { i64, i64 }
+
+ at bar = global %struct.Bar zeroinitializer
+ at foo = global %struct.Foo zeroinitializer

Modified: llvm/trunk/tools/llvm-link/llvm-link.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/llvm-link/llvm-link.cpp?rev=325686&r1=325685&r2=325686&view=diff
==============================================================================
--- llvm/trunk/tools/llvm-link/llvm-link.cpp (original)
+++ llvm/trunk/tools/llvm-link/llvm-link.cpp Wed Feb 21 07:13:48 2018
@@ -70,6 +70,11 @@ static cl::opt<std::string>
 OutputFilename("o", cl::desc("Override output filename"), cl::init("-"),
                cl::value_desc("filename"));
 
+static cl::opt<std::string>
+    InitialModule("initial-module",
+                  cl::desc("Link to existing destination module"), cl::init(""),
+                  cl::value_desc("filename"));
+
 static cl::opt<bool>
 Internalize("internalize", cl::desc("Internalize linked symbols"));
 
@@ -360,7 +365,9 @@ int main(int argc, char **argv) {
   if (!DisableDITypeMap)
     Context.enableDebugTypeODRUniquing();
 
-  auto Composite = make_unique<Module>("llvm-link", Context);
+  auto Composite = InitialModule.empty()
+                       ? make_unique<Module>("llvm-link", Context)
+                       : loadFile(argv[0], InitialModule, Context);
   Linker L(*Composite);
 
   unsigned Flags = Linker::Flags::None;




More information about the llvm-commits mailing list