[AArch64][RegisterBankInfo] Use static mapping for same bank G_BITCAST.

NFC.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@284090 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/lib/Target/AArch64/AArch64RegisterBankInfo.cpp b/lib/Target/AArch64/AArch64RegisterBankInfo.cpp
index 298e697df909c848b63d8fe506cec65925b1ce3f..35d62c46909e984d6fd6bf2008a2ac5742fbba0d 100644 (file)
--- a/lib/Target/AArch64/AArch64RegisterBankInfo.cpp
+++ b/lib/Target/AArch64/AArch64RegisterBankInfo.cpp
@@ -378,6 +378,14 @@ AArch64RegisterBankInfo::getInstrMapping(const MachineInstr &MI) const {
   case TargetOpcode::G_FMUL:
   case TargetOpcode::G_FDIV:
     return getSameKindOfOperandsMapping(MI);
+  case TargetOpcode::G_BITCAST: {
+    LLT DstTy = MRI.getType(MI.getOperand(0).getReg());
+    LLT SrcTy = MRI.getType(MI.getOperand(1).getReg());
+    // If we are on the same bank, we can use the "same kind" mapping.
+    if (DstTy.isVector() == SrcTy.isVector())
+      return getSameKindOfOperandsMapping(MI);
+    break;
+  }
   default:
     break;
   }

diff --git a/test/CodeGen/AArch64/GlobalISel/arm64-regbankselect.mir b/test/CodeGen/AArch64/GlobalISel/arm64-regbankselect.mir
index 90d526114ee50ff9e0dbbaa5f042b8fe9b48391c..b8245d3515610fc4027d6afee3d6c9b3ed33ce20 100644 (file)
--- a/test/CodeGen/AArch64/GlobalISel/arm64-regbankselect.mir
+++ b/test/CodeGen/AArch64/GlobalISel/arm64-regbankselect.mir
@@ -57,6 +57,15 @@
   define void @ignoreTargetSpecificInst() { ret void }
 
   define void @regBankSelected_property() { ret void }
+
+  define void @bitcast_s32_gpr() { ret void }
+  define void @bitcast_s32_fpr() { ret void }
+  define void @bitcast_s32_gpr_fpr() { ret void }
+  define void @bitcast_s32_fpr_gpr() { ret void }
+  define void @bitcast_s64_gpr() { ret void }
+  define void @bitcast_s64_fpr() { ret void }
+  define void @bitcast_s64_gpr_fpr() { ret void }
+  define void @bitcast_s64_fpr_gpr() { ret void }
 ...
 
 ---
@@ -395,3 +404,186 @@ regBankSelected: false
 body:             |
   bb.0:
 ...
+
+---
+# CHECK-LABEL: name: bitcast_s32_gpr
+name:            bitcast_s32_gpr
+legalized:       true
+
+# CHECK:      registers:
+# CHECK-NEXT:  - { id: 0, class: gpr }
+# CHECK-NEXT:  - { id: 1, class: gpr }
+registers:
+  - { id: 0, class: _ }
+  - { id: 1, class: _ }
+
+# CHECK:  body:
+# CHECK:    %0(s32) = COPY %w0
+# CHECK:    %1(s32) = G_BITCAST %0
+body:             |
+  bb.0:
+    liveins: %w0
+
+    %0(s32) = COPY %w0
+    %1(s32) = G_BITCAST %0
+...
+
+---
+# CHECK-LABEL: name: bitcast_s32_fpr
+name:            bitcast_s32_fpr
+legalized:       true
+
+# CHECK:      registers:
+# CHECK-NEXT:  - { id: 0, class: fpr }
+# CHECK-NEXT:  - { id: 1, class: fpr }
+registers:
+  - { id: 0, class: _ }
+  - { id: 1, class: _ }
+
+# CHECK:  body:
+# CHECK:    %0(<2 x s16>) = COPY %s0
+# CHECK:    %1(<2 x s16>) = G_BITCAST %0
+body:             |
+  bb.0:
+    liveins: %s0
+
+    %0(<2 x s16>) = COPY %s0
+    %1(<2 x s16>) = G_BITCAST %0
+...
+
+---
+# CHECK-LABEL: name: bitcast_s32_gpr_fpr
+name:            bitcast_s32_gpr_fpr
+legalized:       true
+
+# CHECK:      registers:
+# CHECK-NEXT:  - { id: 0, class: gpr }
+# CHECK-NEXT:  - { id: 1, class: fpr }
+registers:
+  - { id: 0, class: _ }
+  - { id: 1, class: _ }
+
+# CHECK:  body:
+# CHECK:    %0(s32) = COPY %w0
+# CHECK:    %1(<2 x s16>) = G_BITCAST %0
+body:             |
+  bb.0:
+    liveins: %w0
+
+    %0(s32) = COPY %w0
+    %1(<2 x s16>) = G_BITCAST %0
+...
+
+---
+# CHECK-LABEL: name: bitcast_s32_fpr_gpr
+name:            bitcast_s32_fpr_gpr
+legalized:       true
+
+# CHECK:      registers:
+# CHECK-NEXT:  - { id: 0, class: fpr }
+# CHECK-NEXT:  - { id: 1, class: gpr }
+registers:
+  - { id: 0, class: _ }
+  - { id: 1, class: _ }
+
+# CHECK:  body:
+# CHECK:    %0(<2 x s16>) = COPY %s0
+# CHECK:    %1(s32) = G_BITCAST %0
+body:             |
+  bb.0:
+    liveins: %s0
+
+    %0(<2 x s16>) = COPY %s0
+    %1(s32) = G_BITCAST %0
+...
+
+---
+# CHECK-LABEL: name: bitcast_s64_gpr
+name:            bitcast_s64_gpr
+legalized:       true
+
+# CHECK:      registers:
+# CHECK-NEXT:  - { id: 0, class: gpr }
+# CHECK-NEXT:  - { id: 1, class: gpr }
+registers:
+  - { id: 0, class: _ }
+  - { id: 1, class: _ }
+
+# CHECK:  body:
+# CHECK:    %0(s64) = COPY %x0
+# CHECK:    %1(s64) = G_BITCAST %0
+body:             |
+  bb.0:
+    liveins: %x0
+
+    %0(s64) = COPY %x0
+    %1(s64) = G_BITCAST %0
+...
+
+---
+# CHECK-LABEL: name: bitcast_s64_fpr
+name:            bitcast_s64_fpr
+legalized:       true
+
+# CHECK:      registers:
+# CHECK-NEXT:  - { id: 0, class: fpr }
+# CHECK-NEXT:  - { id: 1, class: fpr }
+registers:
+  - { id: 0, class: _ }
+  - { id: 1, class: _ }
+
+# CHECK:  body:
+# CHECK:    %0(<2 x s32>) = COPY %d0
+# CHECK:    %1(<2 x s32>) = G_BITCAST %0
+body:             |
+  bb.0:
+    liveins: %d0
+
+    %0(<2 x s32>) = COPY %d0
+    %1(<2 x s32>) = G_BITCAST %0
+...
+
+---
+# CHECK-LABEL: name: bitcast_s64_gpr_fpr
+name:            bitcast_s64_gpr_fpr
+legalized:       true
+
+# CHECK:      registers:
+# CHECK-NEXT:  - { id: 0, class: gpr }
+# CHECK-NEXT:  - { id: 1, class: fpr }
+registers:
+  - { id: 0, class: _ }
+  - { id: 1, class: _ }
+# CHECK:  body:
+# CHECK:    %0(s64) = COPY %x0
+# CHECK:    %1(<2 x s32>) = G_BITCAST %0
+body:             |
+  bb.0:
+    liveins: %x0
+
+    %0(s64) = COPY %x0
+    %1(<2 x s32>) = G_BITCAST %0
+...
+
+---
+# CHECK-LABEL: name: bitcast_s64_fpr_gpr
+name:            bitcast_s64_fpr_gpr
+legalized:       true
+
+# CHECK:      registers:
+# CHECK-NEXT:  - { id: 0, class: fpr }
+# CHECK-NEXT:  - { id: 1, class: gpr }
+registers:
+  - { id: 0, class: _ }
+  - { id: 1, class: _ }
+
+# CHECK:  body:
+# CHECK:    %0(<2 x s32>) = COPY %d0
+# CHECK:    %1(s64) = G_BITCAST %0
+body:             |
+  bb.0:
+    liveins: %d0
+
+    %0(<2 x s32>) = COPY %d0
+    %1(s64) = G_BITCAST %0
+...