[PATCH] D43130: [ThinLTO] Teach ThinLTO about auto hide symbols

[Steven Wu via Phabricator via llvm-commits]

steven_wu updated this revision to Diff 133646.
steven_wu added a comment.

Move the check later so it is only check when needed.


Repository:
  rL LLVM

https://reviews.llvm.org/D43130

Files:
  lib/Transforms/IPO/FunctionImport.cpp
  test/ThinLTO/X86/linkonce_odr_unnamed_addr.ll
  test/ThinLTO/X86/linkonce_resolution_comdat.ll


Index: test/ThinLTO/X86/linkonce_resolution_comdat.ll
===================================================================
--- test/ThinLTO/X86/linkonce_resolution_comdat.ll
+++ test/ThinLTO/X86/linkonce_resolution_comdat.ll
@@ -10,7 +10,7 @@
 ; Copy from first module is prevailing and converted to weak_odr, copy
 ; from second module is preempted and converted to available_externally and
 ; removed from comdat.
-; IMPORT1: define weak_odr i32 @f(i8*) unnamed_addr comdat($c1) {
+; IMPORT1: define weak_odr hidden i32 @f(i8*) unnamed_addr comdat($c1) {
 ; IMPORT2: define available_externally i32 @f(i8*) unnamed_addr {
 
 ; RUN: llvm-nm -o - < %t1.bc.thinlto.o | FileCheck %s --check-prefix=NM1
Index: test/ThinLTO/X86/linkonce_odr_unnamed_addr.ll
===================================================================
--- /dev/null
+++ test/ThinLTO/X86/linkonce_odr_unnamed_addr.ll
@@ -0,0 +1,10 @@
+; This test ensures that when linkonce_odr + unnamed_addr symbols promoted to
+; weak symbols, it preserves the auto hide property.
+
+; RUN: opt -module-summary %s -o %t.bc
+; RUN: opt -module-summary %s -o %t2.bc
+; RUN: llvm-lto -thinlto-action=thinlink -o %t3.bc %t.bc %t2.bc
+; RUN: llvm-lto -thinlto-action=promote %t.bc -thinlto-index=%t3.bc -o - | llvm-dis -o - | FileCheck %s
+
+; CHECK: @linkonceodrunnamed = weak_odr hidden unnamed_addr constant i32 0
+ at linkonceodrunnamed = linkonce_odr unnamed_addr constant i32 0
Index: lib/Transforms/IPO/FunctionImport.cpp
===================================================================
--- lib/Transforms/IPO/FunctionImport.cpp
+++ lib/Transforms/IPO/FunctionImport.cpp
@@ -683,6 +683,13 @@
         // changed to enable this for aliases.
         llvm_unreachable("Expected GV to be converted");
     } else {
+      // If the original symbols has global unnamed addr and linkonce_odr linkage,
+      // it should be an auto hide symbol. Add hidden visibility to the symbol to
+      // preserve the property.
+      if (GV.hasLinkOnceODRLinkage() && GV.hasGlobalUnnamedAddr() &&
+          NewLinkage == GlobalValue::WeakODRLinkage)
+        GV.setVisibility(GlobalValue::HiddenVisibility);
+
       DEBUG(dbgs() << "ODR fixing up linkage for `" << GV.getName() << "` from "
                    << GV.getLinkage() << " to " << NewLinkage << "\n");
       GV.setLinkage(NewLinkage);


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D43130.133646.patch
Type: text/x-patch
Size: 2340 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20180209/bd86ca3f/attachment.bin>