[PATCH] D23637: [ThinLTO] Keep common symbols in ThinLTO modules

[Teresa Johnson via llvm-commits]

This revision was automatically updated to reflect the committed changes.
Closed by commit rL279023: [ThinLTO] Keep common symbols in ThinLTO modules (authored by tejohnson).

Changed prior to commit:
  https://reviews.llvm.org/D23637?vs=68462&id=68470#toc

Repository:
  rL LLVM

https://reviews.llvm.org/D23637

Files:
  llvm/trunk/include/llvm/LTO/LTO.h
  llvm/trunk/test/tools/gold/X86/Inputs/common_thinlto.ll
  llvm/trunk/test/tools/gold/X86/common_thinlto.ll
  llvm/trunk/tools/gold/gold-plugin.cpp

Index: llvm/trunk/include/llvm/LTO/LTO.h
===================================================================
--- llvm/trunk/include/llvm/LTO/LTO.h
+++ llvm/trunk/include/llvm/LTO/LTO.h
@@ -228,6 +228,10 @@
   StringRef getSourceFileName() const {
     return Obj->getModule().getSourceFileName();
   }
+
+  MemoryBufferRef getMemoryBufferRef() const {
+    return Obj->getMemoryBufferRef();
+  }
 };
 
 /// A ThinBackend defines what happens after the thin-link phase during ThinLTO.
Index: llvm/trunk/test/tools/gold/X86/common_thinlto.ll
===================================================================
--- llvm/trunk/test/tools/gold/X86/common_thinlto.ll
+++ llvm/trunk/test/tools/gold/X86/common_thinlto.ll
@@ -0,0 +1,38 @@
+; RUN: opt -module-summary %s -o %t1.o
+; RUN: opt -module-summary %p/Inputs/common_thinlto.ll -o %t2.o
+
+; RUN: %gold -plugin %llvmshlibdir/LLVMgold.so \
+; RUN:    --plugin-opt=save-temps \
+; RUN:    --plugin-opt=thinlto \
+; RUN:    -shared %t1.o %t2.o -o %t3
+
+; RUN: llvm-dis %t1.o.2.internalize.bc -o - | FileCheck %s --check-prefix=INTERNALIZE
+; We should not have internalized P, and it should still be common.
+; INTERNALIZE: @P = common global
+
+; RUN: llvm-dis %t1.o.4.opt.bc -o - | FileCheck %s --check-prefix=OPT
+; bar should still exist (if we had internalized P it would look dead).
+; OPT: @bar
+
+; RUN: llvm-nm %t3.o | FileCheck %s --check-prefix=NM
+; NM: bar
+
+source_filename = "common1.c"
+target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
+target triple = "x86_64-unknown-linux-gnu"
+
+ at P = common global i8* (...)* null, align 8
+
+define i32 @main() {
+entry:
+  store i8* (...)* bitcast (i8* ()* @bar to i8* (...)*), i8* (...)** @P, align 8
+  %call = call i32 (...) @foo()
+  ret i32 0
+}
+
+declare i32 @foo(...)
+
+define internal i8* @bar() {
+entry:
+  ret i8* null
+}
Index: llvm/trunk/test/tools/gold/X86/Inputs/common_thinlto.ll
===================================================================
--- llvm/trunk/test/tools/gold/X86/Inputs/common_thinlto.ll
+++ llvm/trunk/test/tools/gold/X86/Inputs/common_thinlto.ll
@@ -0,0 +1,13 @@
+source_filename = "common2.c"
+target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
+target triple = "x86_64-unknown-linux-gnu"
+
+ at P = external global i8* (...)*, align 8
+
+; Function Attrs: nounwind uwtable
+define void @foo() #0 {
+entry:
+  %0 = load i8* (...)*, i8* (...)** @P, align 8
+  %call = call i8* (...) %0()
+  ret void
+}
Index: llvm/trunk/tools/gold/gold-plugin.cpp
===================================================================
--- llvm/trunk/tools/gold/gold-plugin.cpp
+++ llvm/trunk/tools/gold/gold-plugin.cpp
@@ -572,6 +572,8 @@
             toString(ObjOrErr.takeError()).c_str());
 
   InputFile &Obj = **ObjOrErr;
+  bool HasThinLTOSummary =
+      hasGlobalValueSummary(Obj.getMemoryBufferRef(), diagnosticHandler);
 
   unsigned SymNum = 0;
   std::vector<SymbolResolution> Resols(F.syms.size());
@@ -617,7 +619,8 @@
         (IsExecutable || !Res.DefaultVisibility))
       R.FinalDefinitionInLinkageUnit = true;
 
-    if (ObjSym.getFlags() & object::BasicSymbolRef::SF_Common) {
+    if ((ObjSym.getFlags() & object::BasicSymbolRef::SF_Common) &&
+        !HasThinLTOSummary) {
       // We ignore gold's resolution for common symbols. A common symbol with
       // the correct size and alignment is added to the module by the pre-opt
       // module hook if any common symbol prevailed.


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D23637.68470.patch
Type: text/x-patch
Size: 3449 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20160818/94d2f05e/attachment.bin>