[llvm] r337750 - [ThinLTO] Ensure the TargetLibraryInfo is constructed early enough

[Teresa Johnson via llvm-commits]

Author: tejohnson
Date: Mon Jul 23 14:58:19 2018
New Revision: 337750

URL: http://llvm.org/viewvc/llvm-project?rev=337750&view=rev
Log:
[ThinLTO] Ensure the TargetLibraryInfo is constructed early enough

Summary:
Without this change, the WholeProgramDevirt pass, which requires the
TargetLibraryInfo, will construct one from the default triple.

Fixes PR38139.

Reviewers: pcc

Subscribers: mehdi_amini, inglorion, steven_wu, dexonsmith, llvm-commits

Differential Revision: https://reviews.llvm.org/D49278

Added:
    llvm/trunk/test/LTO/X86/triple-init2.ll
Modified:
    llvm/trunk/lib/Transforms/IPO/PassManagerBuilder.cpp
    llvm/trunk/test/LTO/X86/triple-init.ll

Modified: llvm/trunk/lib/Transforms/IPO/PassManagerBuilder.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Transforms/IPO/PassManagerBuilder.cpp?rev=337750&r1=337749&r2=337750&view=diff
==============================================================================
--- llvm/trunk/lib/Transforms/IPO/PassManagerBuilder.cpp (original)
+++ llvm/trunk/lib/Transforms/IPO/PassManagerBuilder.cpp Mon Jul 23 14:58:19 2018
@@ -903,6 +903,8 @@ void PassManagerBuilder::addLateLTOOptim
 void PassManagerBuilder::populateThinLTOPassManager(
     legacy::PassManagerBase &PM) {
   PerformThinLTO = true;
+  if (LibraryInfo)
+    PM.add(new TargetLibraryInfoWrapperPass(*LibraryInfo));
 
   if (VerifyInput)
     PM.add(createVerifierPass());

Modified: llvm/trunk/test/LTO/X86/triple-init.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/LTO/X86/triple-init.ll?rev=337750&r1=337749&r2=337750&view=diff
==============================================================================
--- llvm/trunk/test/LTO/X86/triple-init.ll (original)
+++ llvm/trunk/test/LTO/X86/triple-init.ll Mon Jul 23 14:58:19 2018
@@ -1,6 +1,15 @@
+; Test to ensure that the LTO pipelines add pass to build the TargetLibraryInfo
+; using the specified target triple.
+
+; Check with regular LTO
 ; RUN: llvm-as < %s >%t1
 ; RUN: llvm-lto -exported-symbol=_main  -o %t2 %t1
-; RUN: llvm-nm %t2 | FileCheck %s 
+; RUN: llvm-nm %t2 | FileCheck %s
+; Check with ThinLTO. Use llvm-lto2 since this adds earlier passes requiring
+; the TargetLibraryInfo with ThinLTO (WholeProgramDevirt).
+; RUN: opt -module-summary -o %t1 %s
+; RUN: llvm-lto2 run -r %t1,_pow, -r %t1,_main,plx -o %t2 %t1
+; RUN: llvm-nm %t2.1 | FileCheck %s
 
 target datalayout = "e-m:o-i64:64-f80:128-n8:16:32:64-S128"
 target triple = "x86_64-apple-macosx10.9"

Added: llvm/trunk/test/LTO/X86/triple-init2.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/LTO/X86/triple-init2.ll?rev=337750&view=auto
==============================================================================
--- llvm/trunk/test/LTO/X86/triple-init2.ll (added)
+++ llvm/trunk/test/LTO/X86/triple-init2.ll Mon Jul 23 14:58:19 2018
@@ -0,0 +1,31 @@
+; Test to ensure that the LTO pipelines add pass to build the TargetLibraryInfo
+; using the specified target triple.
+
+; Check with regular LTO
+; RUN: llvm-as < %s >%t1
+; RUN: llvm-lto -exported-symbol=main -o %t2 %t1
+; RUN: llvm-nm %t2 | FileCheck %s
+; Check with ThinLTO. Use llvm-lto2 since this adds earlier passes requiring
+; the TargetLibraryInfo with ThinLTO (WholeProgramDevirt).
+; RUN: opt -module-summary -o %t1 %s
+; RUN: llvm-lto2 run -r %t1,main,plx -o %t2 %t1
+; RUN: llvm-nm %t2.1 | FileCheck %s
+
+; We check that LTO will be aware of target triple and prevent exp2 to ldexpf
+; transformation on Windows.
+; CHECK: U exp2f
+
+target datalayout = "e-m:w-i64:64-f80:128-n8:16:32:64-S128"
+target triple = "x86_64-pc-windows-msvc19.11.0"
+
+define dso_local i32 @main(i32 %argc, i8** nocapture readnone %argv) local_unnamed_addr {
+entry:
+  %conv = sitofp i32 %argc to float
+  %exp2 = tail call float @llvm.exp2.f32(float %conv)
+  %conv1 = fptosi float %exp2 to i32
+  ret i32 %conv1
+}
+
+; Function Attrs: nounwind readnone speculatable
+declare float @llvm.exp2.f32(float)
+