From 404e4350dcc9690a5a48a66aaf32425651b84b7f Mon Sep 17 00:00:00 2001
From: Quentin Colombet <qcolombet@apple.com>
Date: Wed, 12 Oct 2016 03:57:43 +0000
Subject: [PATCH] [AArch64][MachineLegalizer] Mark more bitcasts as legal.

Those are copies, we do not have to do any legalization action for them.

llvm-svn: 283970
---
 llvm/lib/Target/AArch64/AArch64MachineLegalizer.cpp      |  3 +++
 llvm/test/CodeGen/AArch64/GlobalISel/legalize-simple.mir | 10 ++++++++++
 2 files changed, 13 insertions(+)

diff --git a/llvm/lib/Target/AArch64/AArch64MachineLegalizer.cpp b/llvm/lib/Target/AArch64/AArch64MachineLegalizer.cpp
index 4bb0f8d5..adda760 100644
--- a/llvm/lib/Target/AArch64/AArch64MachineLegalizer.cpp
+++ b/llvm/lib/Target/AArch64/AArch64MachineLegalizer.cpp
@@ -176,6 +176,7 @@ AArch64MachineLegalizer::AArch64MachineLegalizer() {
   setAction({G_INTTOPTR, 0, p0}, Legal);
   setAction({G_INTTOPTR, 1, s64}, Legal);
 
+  // Casts for 32 and 64-bit width type are just copies.
   for (auto Ty : {s1, s8, s16, s32, s64}) {
     setAction({G_BITCAST, 0, Ty}, Legal);
     setAction({G_BITCAST, 1, Ty}, Legal);
@@ -183,10 +184,12 @@ AArch64MachineLegalizer::AArch64MachineLegalizer() {
 
   for (int EltSize = 8; EltSize <= 64; EltSize *= 2) {
     setAction({G_BITCAST, 0, LLT::vector(128/EltSize, EltSize)}, Legal);
+    setAction({G_BITCAST, 1, LLT::vector(128/EltSize, EltSize)}, Legal);
     if (EltSize == 64)
       continue;
 
     setAction({G_BITCAST, 0, LLT::vector(64/EltSize, EltSize)}, Legal);
+    setAction({G_BITCAST, 1, LLT::vector(64/EltSize, EltSize)}, Legal);
   }
 
   computeTables();
diff --git a/llvm/test/CodeGen/AArch64/GlobalISel/legalize-simple.mir b/llvm/test/CodeGen/AArch64/GlobalISel/legalize-simple.mir
index 499d800..29cc49f 100644
--- a/llvm/test/CodeGen/AArch64/GlobalISel/legalize-simple.mir
+++ b/llvm/test/CodeGen/AArch64/GlobalISel/legalize-simple.mir
@@ -44,6 +44,9 @@ registers:
   - { id: 27, class: _ }
   - { id: 28, class: _ }
   - { id: 29, class: _ }
+  - { id: 30, class: _ }
+  - { id: 31, class: _ }
+  - { id: 32, class: _ }
 body: |
   bb.0.entry:
     liveins: %x0, %x1, %x2, %x3
@@ -117,4 +120,11 @@ body: |
     %27(s16) = G_SELECT %10, %12, %7
     %28(s32) = G_SELECT %10, %15, %16
     %29(s64) = G_SELECT %10, %9, %24
+
+    ; CHECK: %30(<2 x s32>) = G_BITCAST %9
+    ; CHECK: %31(s64) = G_BITCAST %30
+    ; CHECK: %32(s32) = G_BITCAST %15
+    %30(<2 x s32>) = G_BITCAST %9
+    %31(s64) = G_BITCAST %30
+    %32(s32) = G_BITCAST %15
 ...
-- 
2.7.4