From: Tom Stellard <tstellar@redhat.com>
Date: Tue, 24 Jul 2018 02:19:20 +0000 (+0000)
Subject: AMDGPU/GlobalISel: Legalize G_INSERT
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=6779fccada2c280bdfe9d542aa837a7bfb434ea9;p=llvm

AMDGPU/GlobalISel: Legalize G_INSERT

Reviewers: arsenm

Subscribers: kzhuravl, wdng, nhaehnle, yaxunl, rovka, kristof.beyls, dstuttard, tpr, t-tye, llvm-commits

Differential Revision: https://reviews.llvm.org/D49601

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

diff --git a/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp b/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp
index e06d2b5372f..87b072c9ea2 100644
--- a/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp
+++ b/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp
@@ -172,7 +172,7 @@ AMDGPULegalizerInfo::AMDGPULegalizerInfo(const GCNSubtarget &ST,
   }
 
   // FIXME: Doesn't handle extract of illegal sizes.
-  getActionDefinitionsBuilder(G_EXTRACT)
+  getActionDefinitionsBuilder({G_EXTRACT, G_INSERT})
     .legalIf([=](const LegalityQuery &Query) {
         const LLT &Ty0 = Query.Types[0];
         const LLT &Ty1 = Query.Types[1];
diff --git a/test/CodeGen/AMDGPU/GlobalISel/legalize-insert.mir b/test/CodeGen/AMDGPU/GlobalISel/legalize-insert.mir
new file mode 100644
index 00000000000..569f6a3b260
--- /dev/null
+++ b/test/CodeGen/AMDGPU/GlobalISel/legalize-insert.mir
@@ -0,0 +1,123 @@
+# RUN: llc -mtriple=amdgcn-mesa-mesa3d -run-pass=legalizer %s -o - | FileCheck %s
+
+---
+name: test_insert_lo32_i64
+body: |
+  bb.0.entry:
+    liveins: $vgpr0_vgpr1, $vgpr2
+
+    ; CHECK-LABEL: name: test_insert_lo32_i64
+    ; CHECK: %2:_(s64) = G_INSERT %0, %1(s32), 0
+    %0:_(s64) = COPY $vgpr0_vgpr1
+    %1:_(s32) = COPY $vgpr2
+    %2:_(s64) = G_INSERT %0, %1, 0
+    S_ENDPGM implicit %2
+...
+---
+name: test_insert_hi32_i64
+body: |
+  bb.0.entry:
+    liveins: $vgpr0_vgpr1, $vgpr2
+
+    ; CHECK-LABEL: name: test_insert_hi32_i64
+    ; CHECK: %2:_(s64) = G_INSERT %0, %1(s32), 32
+    %0:_(s64) = COPY $vgpr0_vgpr1
+    %1:_(s32) = COPY $vgpr2
+    %2:_(s64) = G_INSERT %0, %1, 32
+    S_ENDPGM implicit %2
+...
+
+---
+name: test_insert_dword0_i128
+body: |
+  bb.0.entry:
+    liveins: $vgpr0_vgpr1_vgpr2_vgpr3, $vgpr4
+
+    ; CHECK-LABEL: name: test_insert_dword0_i128
+    ; CHECK: %2:_(s128) = G_INSERT %0, %1(s32), 0
+    %0:_(s128) = COPY $vgpr0_vgpr1_vgpr2_vgpr3
+    %1:_(s32) = COPY $vgpr4
+    %2:_(s128) = G_INSERT %0, %1, 0
+    S_ENDPGM implicit %2
+...
+
+---
+name: test_insert_dword1_i128
+body: |
+  bb.0.entry:
+    liveins: $vgpr0_vgpr1_vgpr2_vgpr3, $vgpr4
+
+    ; CHECK-LABEL: name: test_insert_dword1_i128
+    ; CHECK: %2:_(s128) = G_INSERT %0, %1(s32), 32
+    %0:_(s128) = COPY $vgpr0_vgpr1_vgpr2_vgpr3
+    %1:_(s32) = COPY $vgpr4
+    %2:_(s128) = G_INSERT %0, %1, 32
+    S_ENDPGM implicit %2
+...
+
+---
+name: test_insert_dword2_i128
+body: |
+  bb.0.entry:
+    liveins: $vgpr0_vgpr1_vgpr2_vgpr3, $vgpr4
+
+    ; CHECK-LABEL: name: test_insert_dword2_i128
+    ; CHECK: %2:_(s128) = G_INSERT %0, %1(s32), 64
+    %0:_(s128) = COPY $vgpr0_vgpr1_vgpr2_vgpr3
+    %1:_(s32) = COPY $vgpr4
+    %2:_(s128) = G_INSERT %0, %1, 64
+    S_ENDPGM implicit %2
+...
+
+---
+name: test_insert_dword3_i128
+body: |
+  bb.0.entry:
+    liveins: $vgpr0_vgpr1_vgpr2_vgpr3, $vgpr4
+
+    ; CHECK-LABEL: name: test_insert_dword3_i128
+    ; CHECK: %2:_(s128) = G_INSERT %0, %1(s32), 96
+    %0:_(s128) = COPY $vgpr0_vgpr1_vgpr2_vgpr3
+    %1:_(s32) = COPY $vgpr4
+    %2:_(s128) = G_INSERT %0, %1, 96
+    S_ENDPGM implicit %2
+...
+---
+name: test_insert_dword0_i96
+body: |
+  bb.0.entry:
+    liveins: $vgpr0_vgpr1_vgpr2, $vgpr3
+
+    ; CHECK-LABEL: name: test_insert_dword0_i96
+    ; CHECK: %2:_(s96) = G_INSERT %0, %1(s32), 0
+    %0:_(s96) = COPY $vgpr0_vgpr1_vgpr2
+    %1:_(s32) = COPY $vgpr3
+    %2:_(s96) = G_INSERT %0, %1, 0
+    S_ENDPGM implicit %2
+...
+---
+name: test_insert_dword1_i96
+body: |
+  bb.0.entry:
+    liveins: $vgpr0_vgpr1_vgpr2, $vgpr3
+
+    ; CHECK-LABEL: name: test_insert_dword1_i96
+    ; CHECK: %2:_(s96) = G_INSERT %0, %1(s32), 32
+    %0:_(s96) = COPY $vgpr0_vgpr1_vgpr2
+    %1:_(s32) = COPY $vgpr3
+    %2:_(s96) = G_INSERT %0, %1, 32
+    S_ENDPGM implicit %2
+...
+---
+name: test_insert_dword2_i96
+body: |
+  bb.0.entry:
+    liveins: $vgpr0_vgpr1_vgpr2, $vgpr3
+
+    ; CHECK-LABEL: name: test_insert_dword2_i96
+    ; CHECK: %2:_(s96) = G_INSERT %0, %1(s32), 64
+    %0:_(s96) = COPY $vgpr0_vgpr1_vgpr2
+    %1:_(s32) = COPY $vgpr3
+    %2:_(s96) = G_INSERT %0, %1, 64
+    S_ENDPGM implicit %2
+...