[PATCH] D101226: [SimplifyCFG] Preserve metadata when unconditionalizing branches (same target).

[Michael Kruse via Phabricator via llvm-commits]

Meinersbur updated this revision to Diff 340417.
Meinersbur added a comment.

- Transfer only selected metadata kinds.


Repository:
  rG LLVM Github Monorepo

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

https://reviews.llvm.org/D101226

Files:
  llvm/lib/Transforms/Utils/Local.cpp
  llvm/test/Transforms/SimplifyCFG/commondest-loopid.ll


Index: llvm/test/Transforms/SimplifyCFG/commondest-loopid.ll
===================================================================
--- /dev/null
+++ llvm/test/Transforms/SimplifyCFG/commondest-loopid.ll
@@ -0,0 +1,19 @@
+; RUN: opt < %s -simplifycfg -S | FileCheck %s
+;
+; Ensure that the loop metadata is preserved when converting the
+; conditional branch to an unconditional.
+
+define void @commondest_loopid(i1 %T) {
+; CHECK-LABEL: @commondest_loopid(
+; CHECK: !llvm.loop !0
+; CHECK: !0 = distinct !{!0, !1}
+; CHECK: !1 = !{!"loopprop"}
+entry:
+        br label %loop
+
+loop:
+        br i1 %T, label %loop, label %loop, !llvm.loop !0
+}
+
+!0 = distinct !{!0, !1}
+!1 = !{!"loopprop"}
Index: llvm/lib/Transforms/Utils/Local.cpp
===================================================================
--- llvm/lib/Transforms/Utils/Local.cpp
+++ llvm/lib/Transforms/Utils/Local.cpp
@@ -148,7 +148,11 @@
       Dest1->removePredecessor(BI->getParent());
 
       // Replace the conditional branch with an unconditional one.
-      Builder.CreateBr(Dest1);
+      BranchInst *NewBI = Builder.CreateBr(Dest1);
+
+      // Transfer the metadata to the new branch instruction.
+      NewBI->copyMetadata(*BI, {LLVMContext::MD_loop, LLVMContext::MD_dbg,        LLVMContext::MD_annotation});
+
       Value *Cond = BI->getCondition();
       BI->eraseFromParent();
       if (DeleteDeadConditions)


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D101226.340417.patch
Type: text/x-patch
Size: 1396 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20210426/3321d231/attachment.bin>