[llvm-commits] [llvm] r142168 - in /llvm/trunk: include/llvm/Instructions.h lib/Transforms/InstCombine/InstructionCombining.cpp lib/VMCore/Instructions.cpp test/Transforms/InstCombine/canonicalize_branch.ll

Chandler Carruth chandlerc at gmail.com
Sun Oct 16 18:11:57 PDT 2011


Author: chandlerc
Date: Sun Oct 16 20:11:57 2011
New Revision: 142168

URL: http://llvm.org/viewvc/llvm-project?rev=142168&view=rev
Log:
Add a routine to swap branch instruction operands, and update any
profile metadata at the same time. Use it to preserve metadata attached
to a branch when re-writing it in InstCombine.

Add metadata to the canonicalize_branch InstCombine test, and check that
it is tranformed correctly.

Reviewed by Nick Lewycky!

Modified:
    llvm/trunk/include/llvm/Instructions.h
    llvm/trunk/lib/Transforms/InstCombine/InstructionCombining.cpp
    llvm/trunk/lib/VMCore/Instructions.cpp
    llvm/trunk/test/Transforms/InstCombine/canonicalize_branch.ll

Modified: llvm/trunk/include/llvm/Instructions.h
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/include/llvm/Instructions.h?rev=142168&r1=142167&r2=142168&view=diff
==============================================================================
--- llvm/trunk/include/llvm/Instructions.h (original)
+++ llvm/trunk/include/llvm/Instructions.h Sun Oct 16 20:11:57 2011
@@ -2367,6 +2367,13 @@
     *(&Op<-1>() - idx) = (Value*)NewSucc;
   }
 
+  /// \brief Swap the successors of this branch instruction.
+  ///
+  /// Swaps the successors of the branch instruction. This also swaps any
+  /// branch weight metadata associated with the instruction so that it
+  /// continues to map correctly to each operand.
+  void swapSuccessors();
+
   // Methods for support type inquiry through isa, cast, and dyn_cast:
   static inline bool classof(const BranchInst *) { return true; }
   static inline bool classof(const Instruction *I) {

Modified: llvm/trunk/lib/Transforms/InstCombine/InstructionCombining.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Transforms/InstCombine/InstructionCombining.cpp?rev=142168&r1=142167&r2=142168&view=diff
==============================================================================
--- llvm/trunk/lib/Transforms/InstCombine/InstructionCombining.cpp (original)
+++ llvm/trunk/lib/Transforms/InstCombine/InstructionCombining.cpp Sun Oct 16 20:11:57 2011
@@ -1190,8 +1190,7 @@
       !isa<Constant>(X)) {
     // Swap Destinations and condition...
     BI.setCondition(X);
-    BI.setSuccessor(0, FalseDest);
-    BI.setSuccessor(1, TrueDest);
+    BI.swapSuccessors();
     return &BI;
   }
 
@@ -1206,8 +1205,7 @@
       Cond->setPredicate(FCmpInst::getInversePredicate(FPred));
       
       // Swap Destinations and condition.
-      BI.setSuccessor(0, FalseDest);
-      BI.setSuccessor(1, TrueDest);
+      BI.swapSuccessors();
       Worklist.Add(Cond);
       return &BI;
     }
@@ -1223,8 +1221,7 @@
       ICmpInst *Cond = cast<ICmpInst>(BI.getCondition());
       Cond->setPredicate(ICmpInst::getInversePredicate(IPred));
       // Swap Destinations and condition.
-      BI.setSuccessor(0, FalseDest);
-      BI.setSuccessor(1, TrueDest);
+      BI.swapSuccessors();
       Worklist.Add(Cond);
       return &BI;
     }

Modified: llvm/trunk/lib/VMCore/Instructions.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/VMCore/Instructions.cpp?rev=142168&r1=142167&r2=142168&view=diff
==============================================================================
--- llvm/trunk/lib/VMCore/Instructions.cpp (original)
+++ llvm/trunk/lib/VMCore/Instructions.cpp Sun Oct 16 20:11:57 2011
@@ -785,6 +785,27 @@
   SubclassOptionalData = BI.SubclassOptionalData;
 }
 
+void BranchInst::swapSuccessors() {
+  assert(isConditional() &&
+         "Cannot swap successors of an unconditional branch");
+  Op<-1>().swap(Op<-2>());
+
+  // Update profile metadata if present and it matches our structural
+  // expectations.
+  MDNode *ProfileData = getMetadata(LLVMContext::MD_prof);
+  if (!ProfileData || ProfileData->getNumOperands() != 3)
+    return;
+
+  // The first operand is the name. Fetch them backwards and build a new one.
+  Value *Ops[] = {
+    ProfileData->getOperand(0),
+    ProfileData->getOperand(2),
+    ProfileData->getOperand(1)
+  };
+  setMetadata(LLVMContext::MD_prof,
+              MDNode::get(ProfileData->getContext(), Ops));
+}
+
 BasicBlock *BranchInst::getSuccessorV(unsigned idx) const {
   return getSuccessor(idx);
 }

Modified: llvm/trunk/test/Transforms/InstCombine/canonicalize_branch.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/Transforms/InstCombine/canonicalize_branch.ll?rev=142168&r1=142167&r2=142168&view=diff
==============================================================================
--- llvm/trunk/test/Transforms/InstCombine/canonicalize_branch.ll (original)
+++ llvm/trunk/test/Transforms/InstCombine/canonicalize_branch.ll Sun Oct 16 20:11:57 2011
@@ -1,8 +1,23 @@
 ; RUN: opt < %s -instcombine -S | FileCheck %s
 
+; Test an already canonical branch to make sure we don't flip those.
+define i32 @test0(i32 %X, i32 %Y) {
+        %C = icmp eq i32 %X, %Y
+        br i1 %C, label %T, label %F, !prof !0
+
+; CHECK: @test0
+; CHECK: %C = icmp eq i32 %X, %Y
+; CHECK: br i1 %C, label %T, label %F
+
+T:
+        ret i32 12
+F:
+        ret i32 123
+}
+
 define i32 @test1(i32 %X, i32 %Y) {
         %C = icmp ne i32 %X, %Y
-        br i1 %C, label %T, label %F
+        br i1 %C, label %T, label %F, !prof !1
 
 ; CHECK: @test1
 ; CHECK: %C = icmp eq i32 %X, %Y
@@ -16,7 +31,7 @@
 
 define i32 @test2(i32 %X, i32 %Y) {
         %C = icmp ule i32 %X, %Y
-        br i1 %C, label %T, label %F
+        br i1 %C, label %T, label %F, !prof !2
 
 ; CHECK: @test2
 ; CHECK: %C = icmp ugt i32 %X, %Y
@@ -30,7 +45,7 @@
 
 define i32 @test3(i32 %X, i32 %Y) {
         %C = icmp uge i32 %X, %Y
-        br i1 %C, label %T, label %F
+        br i1 %C, label %T, label %F, !prof !3
 
 ; CHECK: @test3
 ; CHECK: %C = icmp ult i32 %X, %Y
@@ -42,3 +57,13 @@
         ret i32 123
 }
 
+!0 = metadata !{metadata !"branch_weights", i32 1, i32 2}
+!1 = metadata !{metadata !"branch_weights", i32 3, i32 4}
+!2 = metadata !{metadata !"branch_weights", i32 5, i32 6}
+!3 = metadata !{metadata !"branch_weights", i32 7, i32 8}
+; Base case shouldn't change.
+; CHECK: !0 = {{.*}} i32 1, i32 2}
+; Ensure that the branch metadata is reversed to match the reversals above.
+; CHECK: !1 = {{.*}} i32 4, i32 3}
+; CHECK: !2 = {{.*}} i32 6, i32 5}
+; CHECK: !3 = {{.*}} i32 8, i32 7}





More information about the llvm-commits mailing list