mirror of
https://github.com/RPCS3/llvm.git
synced 2026-07-01 21:04:04 -04:00
[GISel][CombinerHelper] Add concat_vectors(build_vector, build_vector) => build_vector
Teach the combiner helper how to flatten concat_vectors of build_vectors into a build_vector. Add this combine as part of AArch64 pre-legalizer combiner. Differential Revision: https://reviews.llvm.org/D69071 git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@375066 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Quentin Colombet
@@ -0,0 +1,141 @@
|
||||
# NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
|
||||
# RUN: llc -mtriple aarch64-apple-ios -run-pass=aarch64-prelegalizer-combiner %s -o - | FileCheck %s
|
||||
|
||||
# Check that we canonicalize concat_vectors(build_vector, build_vector)
|
||||
# into build_vector.
|
||||
---
|
||||
name: concat_to_build_vector
|
||||
tracksRegLiveness: true
|
||||
body: |
|
||||
bb.1:
|
||||
liveins: $x0, $x1, $x2, $x3
|
||||
|
||||
; CHECK-LABEL: name: concat_to_build_vector
|
||||
; CHECK: liveins: $x0, $x1, $x2, $x3
|
||||
; CHECK: [[COPY:%[0-9]+]]:_(s64) = COPY $x0
|
||||
; CHECK: [[COPY1:%[0-9]+]]:_(s64) = COPY $x1
|
||||
; CHECK: [[COPY2:%[0-9]+]]:_(s64) = COPY $x2
|
||||
; CHECK: [[COPY3:%[0-9]+]]:_(s64) = COPY $x3
|
||||
; CHECK: [[BUILD_VECTOR:%[0-9]+]]:_(<4 x s64>) = G_BUILD_VECTOR [[COPY]](s64), [[COPY1]](s64), [[COPY2]](s64), [[COPY3]](s64)
|
||||
; CHECK: RET_ReallyLR implicit [[BUILD_VECTOR]](<4 x s64>)
|
||||
%0:_(s64) = COPY $x0
|
||||
%1:_(s64) = COPY $x1
|
||||
%2:_(s64) = COPY $x2
|
||||
%3:_(s64) = COPY $x3
|
||||
%4:_(<2 x s64>) = G_BUILD_VECTOR %0(s64), %1
|
||||
%5:_(<2 x s64>) = G_BUILD_VECTOR %2(s64), %3
|
||||
%6:_(<4 x s64>) = G_CONCAT_VECTORS %4(<2 x s64>), %5
|
||||
RET_ReallyLR implicit %6
|
||||
...
|
||||
# Same test as concat_to_build_vector but with pointer types.
|
||||
---
|
||||
name: concat_to_build_vector_ptr
|
||||
tracksRegLiveness: true
|
||||
body: |
|
||||
bb.1:
|
||||
liveins: $x0, $x1, $x2, $x3
|
||||
|
||||
; CHECK-LABEL: name: concat_to_build_vector_ptr
|
||||
; CHECK: liveins: $x0, $x1, $x2, $x3
|
||||
; CHECK: [[COPY:%[0-9]+]]:_(p0) = COPY $x0
|
||||
; CHECK: [[COPY1:%[0-9]+]]:_(p0) = COPY $x1
|
||||
; CHECK: [[COPY2:%[0-9]+]]:_(p0) = COPY $x2
|
||||
; CHECK: [[COPY3:%[0-9]+]]:_(p0) = COPY $x3
|
||||
; CHECK: [[BUILD_VECTOR:%[0-9]+]]:_(<4 x p0>) = G_BUILD_VECTOR [[COPY]](p0), [[COPY1]](p0), [[COPY2]](p0), [[COPY3]](p0)
|
||||
; CHECK: RET_ReallyLR implicit [[BUILD_VECTOR]](<4 x p0>)
|
||||
%0:_(p0) = COPY $x0
|
||||
%1:_(p0) = COPY $x1
|
||||
%2:_(p0) = COPY $x2
|
||||
%3:_(p0) = COPY $x3
|
||||
%4:_(<2 x p0>) = G_BUILD_VECTOR %0(p0), %1
|
||||
%5:_(<2 x p0>) = G_BUILD_VECTOR %2(p0), %3
|
||||
%6:_(<4 x p0>) = G_CONCAT_VECTORS %4(<2 x p0>), %5
|
||||
RET_ReallyLR implicit %6
|
||||
...
|
||||
# Check that we canonicalize concat_vectors(undef, undef) into undef.
|
||||
---
|
||||
name: concat_to_undef
|
||||
tracksRegLiveness: true
|
||||
body: |
|
||||
bb.1:
|
||||
; CHECK-LABEL: name: concat_to_undef
|
||||
; CHECK: [[DEF:%[0-9]+]]:_(<4 x s64>) = G_IMPLICIT_DEF
|
||||
; CHECK: RET_ReallyLR implicit [[DEF]](<4 x s64>)
|
||||
%4:_(<2 x s64>) = G_IMPLICIT_DEF
|
||||
%5:_(<2 x s64>) = G_IMPLICIT_DEF
|
||||
%6:_(<4 x s64>) = G_CONCAT_VECTORS %4(<2 x s64>), %5
|
||||
RET_ReallyLR implicit %6
|
||||
...
|
||||
# Check that when combining concat_vectors(build_vector, undef) into
|
||||
# build_vector, we correctly break the undef vector into a sequence
|
||||
# of undef scalar.
|
||||
---
|
||||
name: concat_to_build_vector_with_undef
|
||||
tracksRegLiveness: true
|
||||
body: |
|
||||
bb.1:
|
||||
liveins: $x0, $x1
|
||||
|
||||
; CHECK-LABEL: name: concat_to_build_vector_with_undef
|
||||
; CHECK: liveins: $x0, $x1
|
||||
; CHECK: [[COPY:%[0-9]+]]:_(s64) = COPY $x0
|
||||
; CHECK: [[COPY1:%[0-9]+]]:_(s64) = COPY $x1
|
||||
; CHECK: [[DEF:%[0-9]+]]:_(s64) = G_IMPLICIT_DEF
|
||||
; CHECK: [[BUILD_VECTOR:%[0-9]+]]:_(<4 x s64>) = G_BUILD_VECTOR [[COPY]](s64), [[COPY1]](s64), [[DEF]](s64), [[DEF]](s64)
|
||||
; CHECK: RET_ReallyLR implicit [[BUILD_VECTOR]](<4 x s64>)
|
||||
%0:_(s64) = COPY $x0
|
||||
%1:_(s64) = COPY $x1
|
||||
%4:_(<2 x s64>) = G_BUILD_VECTOR %0(s64), %1
|
||||
%5:_(<2 x s64>) = G_IMPLICIT_DEF
|
||||
%6:_(<4 x s64>) = G_CONCAT_VECTORS %4(<2 x s64>), %5
|
||||
RET_ReallyLR implicit %6
|
||||
...
|
||||
# Same as concat_to_build_vector_with_undef but with pointer types.
|
||||
---
|
||||
name: concat_to_build_vector_with_undef_ptr
|
||||
tracksRegLiveness: true
|
||||
body: |
|
||||
bb.1:
|
||||
liveins: $x0, $x1
|
||||
|
||||
; CHECK-LABEL: name: concat_to_build_vector_with_undef_ptr
|
||||
; CHECK: liveins: $x0, $x1
|
||||
; CHECK: [[COPY:%[0-9]+]]:_(p0) = COPY $x0
|
||||
; CHECK: [[COPY1:%[0-9]+]]:_(p0) = COPY $x1
|
||||
; CHECK: [[DEF:%[0-9]+]]:_(p0) = G_IMPLICIT_DEF
|
||||
; CHECK: [[BUILD_VECTOR:%[0-9]+]]:_(<4 x p0>) = G_BUILD_VECTOR [[COPY]](p0), [[COPY1]](p0), [[DEF]](p0), [[DEF]](p0)
|
||||
; CHECK: RET_ReallyLR implicit [[BUILD_VECTOR]](<4 x p0>)
|
||||
%0:_(p0) = COPY $x0
|
||||
%1:_(p0) = COPY $x1
|
||||
%4:_(<2 x p0>) = G_BUILD_VECTOR %0(p0), %1
|
||||
%5:_(<2 x p0>) = G_IMPLICIT_DEF
|
||||
%6:_(<4 x p0>) = G_CONCAT_VECTORS %4(<2 x p0>), %5
|
||||
RET_ReallyLR implicit %6
|
||||
...
|
||||
# Check that we keep a concat_vectors as soon as one of the operand is
|
||||
# not undef or build_vector. I.e., we cannot flatten the concat_vectors.
|
||||
---
|
||||
name: concat_to_build_vector_negative_test
|
||||
tracksRegLiveness: true
|
||||
body: |
|
||||
bb.1:
|
||||
liveins: $x0, $x1
|
||||
|
||||
; CHECK-LABEL: name: concat_to_build_vector_negative_test
|
||||
; CHECK: liveins: $x0, $x1
|
||||
; CHECK: [[COPY:%[0-9]+]]:_(s64) = COPY $x0
|
||||
; CHECK: [[DEF:%[0-9]+]]:_(<2 x s64>) = G_IMPLICIT_DEF
|
||||
; CHECK: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 1
|
||||
; CHECK: [[IVEC:%[0-9]+]]:_(<2 x s64>) = G_INSERT_VECTOR_ELT [[DEF]], [[COPY]](s64), [[C]](s32)
|
||||
; CHECK: [[DEF1:%[0-9]+]]:_(<2 x s64>) = G_IMPLICIT_DEF
|
||||
; CHECK: [[CONCAT_VECTORS:%[0-9]+]]:_(<4 x s64>) = G_CONCAT_VECTORS [[IVEC]](<2 x s64>), [[DEF1]](<2 x s64>)
|
||||
; CHECK: RET_ReallyLR implicit [[CONCAT_VECTORS]](<4 x s64>)
|
||||
%0:_(s64) = COPY $x0
|
||||
%1:_(s64) = COPY $x1
|
||||
%2:_(<2 x s64>) = G_IMPLICIT_DEF
|
||||
%3:_(s32) = G_CONSTANT i32 1
|
||||
%4:_(<2 x s64>) = G_INSERT_VECTOR_ELT %2, %0(s64), %3(s32)
|
||||
%5:_(<2 x s64>) = G_IMPLICIT_DEF
|
||||
%6:_(<4 x s64>) = G_CONCAT_VECTORS %4(<2 x s64>), %5
|
||||
RET_ReallyLR implicit %6
|
||||
...
|
||||
Reference in New Issue
Block a user