[llvm] r354113 - [ARM GlobalISel] Support branches for Thumb2

Diana Picus via llvm-commits llvm-commits at lists.llvm.org
Fri Feb 15 02:24:03 PST 2019


Author: rovka
Date: Fri Feb 15 02:24:03 2019
New Revision: 354113

URL: http://llvm.org/viewvc/llvm-project?rev=354113&view=rev
Log:
[ARM GlobalISel] Support branches for Thumb2

Just like arm mode, but with different opcodes.

Added:
    llvm/trunk/test/CodeGen/ARM/GlobalISel/arm-legalize-control-flow.mir
    llvm/trunk/test/CodeGen/ARM/GlobalISel/thumb-select-br.mir
Modified:
    llvm/trunk/lib/Target/ARM/ARMInstructionSelector.cpp
    llvm/trunk/lib/Target/ARM/ARMLegalizerInfo.cpp
    llvm/trunk/test/CodeGen/ARM/GlobalISel/arm-legalizer.mir

Modified: llvm/trunk/lib/Target/ARM/ARMInstructionSelector.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/ARM/ARMInstructionSelector.cpp?rev=354113&r1=354112&r2=354113&view=diff
==============================================================================
--- llvm/trunk/lib/Target/ARM/ARMInstructionSelector.cpp (original)
+++ llvm/trunk/lib/Target/ARM/ARMInstructionSelector.cpp Fri Feb 15 02:24:03 2019
@@ -106,6 +106,9 @@ private:
     unsigned CMPri;
     unsigned MOVCCr;
 
+    unsigned TSTri;
+    unsigned Bcc;
+
     OpcodeCache(const ARMSubtarget &STI);
   } const Opcodes;
 
@@ -300,6 +303,9 @@ ARMInstructionSelector::OpcodeCache::Opc
 
   STORE_OPCODE(CMPri, CMPri);
   STORE_OPCODE(MOVCCr, MOVCCr);
+
+  STORE_OPCODE(TSTri, TSTri);
+  STORE_OPCODE(Bcc, Bcc);
 #undef MAP_OPCODE
 }
 
@@ -1008,17 +1014,19 @@ bool ARMInstructionSelector::select(Mach
     }
 
     // Set the flags.
-    auto Test = BuildMI(*I.getParent(), I, I.getDebugLoc(), TII.get(ARM::TSTri))
-                    .addReg(I.getOperand(0).getReg())
-                    .addImm(1)
-                    .add(predOps(ARMCC::AL));
+    auto Test =
+        BuildMI(*I.getParent(), I, I.getDebugLoc(), TII.get(Opcodes.TSTri))
+            .addReg(I.getOperand(0).getReg())
+            .addImm(1)
+            .add(predOps(ARMCC::AL));
     if (!constrainSelectedInstRegOperands(*Test, TII, TRI, RBI))
       return false;
 
     // Branch conditionally.
-    auto Branch = BuildMI(*I.getParent(), I, I.getDebugLoc(), TII.get(ARM::Bcc))
-                      .add(I.getOperand(1))
-                      .add(predOps(ARMCC::NE, ARM::CPSR));
+    auto Branch =
+        BuildMI(*I.getParent(), I, I.getDebugLoc(), TII.get(Opcodes.Bcc))
+            .add(I.getOperand(1))
+            .add(predOps(ARMCC::NE, ARM::CPSR));
     if (!constrainSelectedInstRegOperands(*Branch, TII, TRI, RBI))
       return false;
     I.eraseFromParent();

Modified: llvm/trunk/lib/Target/ARM/ARMLegalizerInfo.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/ARM/ARMLegalizerInfo.cpp?rev=354113&r1=354112&r2=354113&view=diff
==============================================================================
--- llvm/trunk/lib/Target/ARM/ARMLegalizerInfo.cpp (original)
+++ llvm/trunk/lib/Target/ARM/ARMLegalizerInfo.cpp Fri Feb 15 02:24:03 2019
@@ -140,6 +140,8 @@ ARMLegalizerInfo::ARMLegalizerInfo(const
 
   getActionDefinitionsBuilder(G_GEP).legalFor({{p0, s32}});
 
+  getActionDefinitionsBuilder(G_BRCOND).legalFor({s1});
+
   if (ST.isThumb()) {
     // FIXME: merge with the code for non-Thumb.
     computeTables();
@@ -170,8 +172,6 @@ ARMLegalizerInfo::ARMLegalizerInfo(const
         .clampScalar(0, s32, s32);
   }
 
-  getActionDefinitionsBuilder(G_BRCOND).legalFor({s1});
-
   // We're keeping these builders around because we'll want to add support for
   // floating point to them.
   auto &PhiBuilder =

Added: llvm/trunk/test/CodeGen/ARM/GlobalISel/arm-legalize-control-flow.mir
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/ARM/GlobalISel/arm-legalize-control-flow.mir?rev=354113&view=auto
==============================================================================
--- llvm/trunk/test/CodeGen/ARM/GlobalISel/arm-legalize-control-flow.mir (added)
+++ llvm/trunk/test/CodeGen/ARM/GlobalISel/arm-legalize-control-flow.mir Fri Feb 15 02:24:03 2019
@@ -0,0 +1,39 @@
+# RUN: llc -mtriple arm-- -run-pass=legalizer %s -o - | FileCheck %s
+# RUN: llc -mtriple thumb-- -mattr=+v6t2 -run-pass=legalizer %s -o - | FileCheck %s
+--- |
+  define void @test_brcond() { ret void }
+...
+---
+name:            test_brcond
+# CHECK-LABEL: name: test_brcond
+legalized:       false
+# CHECK: legalized: true
+regBankSelected: false
+selected:        false
+tracksRegLiveness: true
+registers:
+  - { id: 0, class: _ }
+  - { id: 1, class: _ }
+  - { id: 2, class: _ }
+body:             |
+  bb.0:
+    successors: %bb.1(0x40000000), %bb.2(0x40000000)
+    liveins: $r0, $r1
+
+    %0(s32) = COPY $r0
+    %1(s32) = COPY $r1
+    %2(s1) = G_ICMP intpred(sgt), %0(s32), %1
+    G_BRCOND %2(s1), %bb.1
+    ; G_BRCOND with s1 is legal, so we should find it unchanged in the output
+    ; CHECK: G_BRCOND {{%[0-9]+}}(s1), %bb.1
+    G_BR %bb.2
+
+  bb.1:
+    $r0 = COPY %1(s32)
+    BX_RET 14, $noreg, implicit $r0
+
+  bb.2:
+    $r0 = COPY %0(s32)
+    BX_RET 14, $noreg, implicit $r0
+
+...

Modified: llvm/trunk/test/CodeGen/ARM/GlobalISel/arm-legalizer.mir
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/ARM/GlobalISel/arm-legalizer.mir?rev=354113&r1=354112&r2=354113&view=diff
==============================================================================
--- llvm/trunk/test/CodeGen/ARM/GlobalISel/arm-legalizer.mir (original)
+++ llvm/trunk/test/CodeGen/ARM/GlobalISel/arm-legalizer.mir Fri Feb 15 02:24:03 2019
@@ -5,7 +5,6 @@
 
   define void @test_constants_s64() { ret void }
 
-  define void @test_brcond() { ret void }
 
   define void @test_phi_s32() { ret void }
   define void @test_phi_p0() { ret void }
@@ -105,40 +104,6 @@ body:             |
     BX_RET 14, $noreg
 ...
 ---
-name:            test_brcond
-# CHECK-LABEL: name: test_brcond
-legalized:       false
-# CHECK: legalized: true
-regBankSelected: false
-selected:        false
-tracksRegLiveness: true
-registers:
-  - { id: 0, class: _ }
-  - { id: 1, class: _ }
-  - { id: 2, class: _ }
-body:             |
-  bb.0:
-    successors: %bb.1(0x40000000), %bb.2(0x40000000)
-    liveins: $r0, $r1
-
-    %0(s32) = COPY $r0
-    %1(s32) = COPY $r1
-    %2(s1) = G_ICMP intpred(sgt), %0(s32), %1
-    G_BRCOND %2(s1), %bb.1
-    ; G_BRCOND with s1 is legal, so we should find it unchanged in the output
-    ; CHECK: G_BRCOND {{%[0-9]+}}(s1), %bb.1
-    G_BR %bb.2
-
-  bb.1:
-    $r0 = COPY %1(s32)
-    BX_RET 14, $noreg, implicit $r0
-
-  bb.2:
-    $r0 = COPY %0(s32)
-    BX_RET 14, $noreg, implicit $r0
-
-...
----
 name:            test_phi_s32
 # CHECK-LABEL: name: test_phi_s32
 legalized:       false

Added: llvm/trunk/test/CodeGen/ARM/GlobalISel/thumb-select-br.mir
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/ARM/GlobalISel/thumb-select-br.mir?rev=354113&view=auto
==============================================================================
--- llvm/trunk/test/CodeGen/ARM/GlobalISel/thumb-select-br.mir (added)
+++ llvm/trunk/test/CodeGen/ARM/GlobalISel/thumb-select-br.mir Fri Feb 15 02:24:03 2019
@@ -0,0 +1,44 @@
+# RUN: llc -O0 -mtriple thumb-- -mattr=+v6t2 -run-pass=instruction-select -verify-machineinstrs %s -o - | FileCheck %s
+--- |
+  define void @test_br() { ret void }
+...
+---
+name:            test_br
+# CHECK-LABEL: name: test_br
+legalized:       true
+regBankSelected: true
+selected:        false
+# CHECK: selected: true
+registers:
+  - { id: 0, class: gprb }
+  - { id: 1, class: gprb }
+body:             |
+  bb.0:
+  ; CHECK: bb.0
+    successors: %bb.1(0x40000000), %bb.2(0x40000000)
+    liveins: $r0
+
+    %0(s32) = COPY $r0
+    ; CHECK: [[COND32:%[0-9]+]]:gpr = COPY $r0
+    %1(s1) = G_TRUNC %0(s32)
+    ; CHECK: [[COND:%[0-9]+]]:gprnopc = COPY [[COND32]]
+
+    G_BRCOND %1(s1), %bb.1
+    ; CHECK: t2TSTri [[COND]], 1, 14, $noreg, implicit-def $cpsr
+    ; CHECK: t2Bcc %bb.1, 1, $cpsr
+    G_BR %bb.2
+    ; CHECK: t2B %bb.2, 14, $noreg
+
+  bb.1:
+  ; CHECK: bb.1
+    successors: %bb.2(0x80000000)
+
+    G_BR %bb.2
+    ; CHECK: t2B %bb.2, 14, $noreg
+
+  bb.2:
+  ; CHECK: bb.2
+
+    tBX_RET 14, $noreg
+    ; CHECK: tBX_RET 14, $noreg
+...




More information about the llvm-commits mailing list