[PATCH] D88433: [IRMover] Fix up "CG Profile" MDNode after RAUW

[Yuanfang Chen via Phabricator via llvm-commits]

ychen updated this revision to Diff 302329.
ychen added a comment.
Herald added a subscriber: dexonsmith.

- revert fixup in IRMover
- allow CGProfile metadata node representing edge to use bitcast.


Repository:
  rG LLVM Github Monorepo

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D88433/new/

https://reviews.llvm.org/D88433

Files:
  llvm/lib/CodeGen/TargetLoweringObjectFileImpl.cpp
  llvm/lib/IR/Verifier.cpp
  llvm/test/Transforms/FunctionImport/Inputs/cg_profile.ll
  llvm/test/Transforms/FunctionImport/cg_profile.ll


Index: llvm/test/Transforms/FunctionImport/cg_profile.ll
===================================================================
--- /dev/null
+++ llvm/test/Transforms/FunctionImport/cg_profile.ll
@@ -0,0 +1,25 @@
+; Check that RAUW performed by IRMover during function importing does not
+; generate bitcast in "CG Profile" related metadat nodes.
+; RUN: opt -cg-profile -module-summary %s -o %t.bc
+; RUN: opt -module-summary %p/Inputs/cg_profile.ll -o %t2.bc
+; RUN: llvm-lto -thinlto -o %t3 %t.bc %t2.bc
+; RUN: opt -function-import -print-imports -summary-file %t3.thinlto.bc %t.bc -o %t.o
+
+target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
+target triple = "x86_64-unknown-linux-gnu"
+
+; %class.A is defined differently in %p/Inputs/cg_profile.ll. This is to trigger
+; bitcast.
+%class.A = type { i8 }
+
+define void @foo() !prof !2 {
+  call void @bar(%class.A* null)
+  ret void
+}
+
+declare void @bar(%class.A*)
+
+!llvm.module.flags = !{!1}
+
+!1 = !{i32 1, !"EnableSplitLTOUnit", i32 0}
+!2 = !{!"function_entry_count", i64 2753}
Index: llvm/test/Transforms/FunctionImport/Inputs/cg_profile.ll
===================================================================
--- /dev/null
+++ llvm/test/Transforms/FunctionImport/Inputs/cg_profile.ll
@@ -0,0 +1,12 @@
+target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
+target triple = "x86_64-unknown-linux-gnu"
+
+%class.A = type { [16 x i8] }
+
+define void @bar(%class.A*) {
+  ret void
+}
+
+!llvm.module.flags = !{!1}
+
+!1 = !{i32 1, !"EnableSplitLTOUnit", i32 0}
Index: llvm/lib/IR/Verifier.cpp
===================================================================
--- llvm/lib/IR/Verifier.cpp
+++ llvm/lib/IR/Verifier.cpp
@@ -1548,8 +1548,8 @@
     if (!FuncMDO)
       return;
     auto F = dyn_cast<ValueAsMetadata>(FuncMDO);
-    Assert(F && isa<Function>(F->getValue()), "expected a Function or null",
-           FuncMDO);
+    Assert(F && isa<Function>(F->getValue()->stripPointerCasts()),
+           "expected a Function or null", FuncMDO);
   };
   auto Node = dyn_cast_or_null<MDNode>(MDO);
   Assert(Node && Node->getNumOperands() == 3, "expected a MDNode triple", MDO);
Index: llvm/lib/CodeGen/TargetLoweringObjectFileImpl.cpp
===================================================================
--- llvm/lib/CodeGen/TargetLoweringObjectFileImpl.cpp
+++ llvm/lib/CodeGen/TargetLoweringObjectFileImpl.cpp
@@ -344,7 +344,7 @@
     if (!MDO)
       return nullptr;
     auto V = cast<ValueAsMetadata>(MDO);
-    const Function *F = cast<Function>(V->getValue());
+    const Function *F = cast<Function>(V->getValue()->stripPointerCasts());
     return TM->getSymbol(F);
   };
 


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D88433.302329.patch
Type: text/x-patch
Size: 2727 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20201102/c147977d/attachment.bin>