Similar to D123386, this adds D-Movs to the AArch64 perfect shuffle tables, slightly lowering the costs a little more. This is a rough improvement in general, especially if you ignore mov v0.16b, v2.16b type moves that are often artefacts of the calling convention. The D register movs are encoded as (0x4 | LaneIdx), and to generate a D register move we are required to bitcast into a higher type, but it is otherwise very similar to the S-lane mov's already supported. Differential Revision: https://reviews.llvm.org/D125477
130 lines
4.3 KiB
LLVM
130 lines
4.3 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
|
|
; RUN: llc < %s -mtriple=aarch64--linux-gnu | FileCheck %s
|
|
|
|
define <16 x i32> @test_shuf1(<16 x i32> %x, <16 x i32> %y) {
|
|
; CHECK-LABEL: test_shuf1:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: ext v16.16b, v6.16b, v1.16b, #4
|
|
; CHECK-NEXT: dup v5.4s, v4.s[0]
|
|
; CHECK-NEXT: uzp1 v17.4s, v1.4s, v0.4s
|
|
; CHECK-NEXT: uzp2 v18.4s, v2.4s, v4.4s
|
|
; CHECK-NEXT: rev64 v3.4s, v7.4s
|
|
; CHECK-NEXT: trn2 v4.4s, v1.4s, v16.4s
|
|
; CHECK-NEXT: mov v5.s[0], v6.s[3]
|
|
; CHECK-NEXT: trn2 v1.4s, v17.4s, v1.4s
|
|
; CHECK-NEXT: trn1 v2.4s, v18.4s, v2.4s
|
|
; CHECK-NEXT: mov v4.s[0], v7.s[1]
|
|
; CHECK-NEXT: mov v3.d[0], v5.d[0]
|
|
; CHECK-NEXT: ext v1.16b, v0.16b, v1.16b, #12
|
|
; CHECK-NEXT: mov v2.s[3], v7.s[0]
|
|
; CHECK-NEXT: mov v0.16b, v4.16b
|
|
; CHECK-NEXT: ret
|
|
%s3 = shufflevector <16 x i32> %x, <16 x i32> %y, <16 x i32> <i32 29, i32 26, i32 7, i32 4, i32 3, i32 6, i32 5, i32 2, i32 9, i32 8, i32 17, i32 28, i32 27, i32 16, i32 31, i32 30>
|
|
ret <16 x i32> %s3
|
|
}
|
|
|
|
define <4 x i32> @test_shuf2(<16 x i32> %x, <16 x i32> %y) {
|
|
; CHECK-LABEL: test_shuf2:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: zip2 v2.4s, v7.4s, v6.4s
|
|
; CHECK-NEXT: ext v0.16b, v1.16b, v1.16b, #4
|
|
; CHECK-NEXT: trn2 v1.4s, v7.4s, v2.4s
|
|
; CHECK-NEXT: mov v0.d[0], v1.d[0]
|
|
; CHECK-NEXT: ret
|
|
%s3 = shufflevector <16 x i32> %x, <16 x i32> %y, <4 x i32> <i32 29, i32 26, i32 7, i32 4>
|
|
ret <4 x i32> %s3
|
|
}
|
|
|
|
define <4 x i32> @test_shuf3(<16 x i32> %x, <16 x i32> %y) {
|
|
; CHECK-LABEL: test_shuf3:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: uzp1 v2.4s, v1.4s, v0.4s
|
|
; CHECK-NEXT: trn2 v1.4s, v2.4s, v1.4s
|
|
; CHECK-NEXT: ext v0.16b, v0.16b, v1.16b, #12
|
|
; CHECK-NEXT: ret
|
|
%s3 = shufflevector <16 x i32> %x, <16 x i32> %y, <4 x i32> <i32 3, i32 6, i32 5, i32 2>
|
|
ret <4 x i32> %s3
|
|
}
|
|
|
|
define <4 x i32> @test_shuf4(<16 x i32> %x, <16 x i32> %y) {
|
|
; CHECK-LABEL: test_shuf4:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: uzp2 v0.4s, v2.4s, v4.4s
|
|
; CHECK-NEXT: trn1 v0.4s, v0.4s, v2.4s
|
|
; CHECK-NEXT: mov v0.s[3], v7.s[0]
|
|
; CHECK-NEXT: ret
|
|
%s3 = shufflevector <16 x i32> %x, <16 x i32> %y, <4 x i32> <i32 9, i32 8, i32 17, i32 28>
|
|
ret <4 x i32> %s3
|
|
}
|
|
|
|
define <4 x i32> @test_shuf5(<16 x i32> %x, <16 x i32> %y) {
|
|
; CHECK-LABEL: test_shuf5:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: rev64 v0.4s, v7.4s
|
|
; CHECK-NEXT: ext v1.16b, v6.16b, v4.16b, #12
|
|
; CHECK-NEXT: mov v0.d[0], v1.d[0]
|
|
; CHECK-NEXT: ret
|
|
%s3 = shufflevector <16 x i32> %x, <16 x i32> %y, <4 x i32> <i32 27, i32 16, i32 31, i32 30>
|
|
ret <4 x i32> %s3
|
|
}
|
|
|
|
define <4 x i32> @test1503(<4 x i32> %a, <4 x i32> %b)
|
|
; CHECK-LABEL: test1503:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: zip1 v1.4s, v0.4s, v1.4s
|
|
; CHECK-NEXT: ext v1.16b, v1.16b, v0.16b, #8
|
|
; CHECK-NEXT: mov v1.s[3], v0.s[3]
|
|
; CHECK-NEXT: mov v0.16b, v1.16b
|
|
; CHECK-NEXT: ret
|
|
{
|
|
%r = shufflevector <4 x i32> %a, <4 x i32> %b, <4 x i32> <i32 1, i32 5, i32 0, i32 3>
|
|
ret <4 x i32> %r
|
|
}
|
|
|
|
define <4 x i32> @test4366(<4 x i32> %a, <4 x i32> %b)
|
|
; CHECK-LABEL: test4366:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: trn1 v1.4s, v1.4s, v1.4s
|
|
; CHECK-NEXT: mov v1.s[1], v0.s[3]
|
|
; CHECK-NEXT: mov v0.16b, v1.16b
|
|
; CHECK-NEXT: ret
|
|
{
|
|
%r = shufflevector <4 x i32> %a, <4 x i32> %b, <4 x i32> <i32 4, i32 3, i32 6, i32 6>
|
|
ret <4 x i32> %r
|
|
}
|
|
|
|
define <4 x i32> @test7367(<4 x i32> %a, <4 x i32> %b)
|
|
; CHECK-LABEL: test7367:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: mov v2.16b, v1.16b
|
|
; CHECK-NEXT: mov v2.d[0], v0.d[1]
|
|
; CHECK-NEXT: mov v2.s[0], v1.s[3]
|
|
; CHECK-NEXT: mov v0.16b, v2.16b
|
|
; CHECK-NEXT: ret
|
|
{
|
|
%r = shufflevector <4 x i32> %a, <4 x i32> %b, <4 x i32> <i32 7, i32 3, i32 6, i32 7>
|
|
ret <4 x i32> %r
|
|
}
|
|
|
|
define <4 x i32> @test4045(<4 x i32> %a, <4 x i32> %b)
|
|
; CHECK-LABEL: test4045:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: trn1 v0.4s, v1.4s, v0.4s
|
|
; CHECK-NEXT: mov v0.d[1], v1.d[0]
|
|
; CHECK-NEXT: ret
|
|
{
|
|
%r = shufflevector <4 x i32> %a, <4 x i32> %b, <4 x i32> <i32 4, i32 0, i32 4, i32 5>
|
|
ret <4 x i32> %r
|
|
}
|
|
|
|
define <4 x i32> @test0067(<4 x i32> %a, <4 x i32> %b)
|
|
; CHECK-LABEL: test0067:
|
|
; CHECK: // %bb.0:
|
|
; CHECK-NEXT: trn1 v0.4s, v0.4s, v0.4s
|
|
; CHECK-NEXT: mov v0.d[1], v1.d[1]
|
|
; CHECK-NEXT: ret
|
|
{
|
|
%r = shufflevector <4 x i32> %a, <4 x i32> %b, <4 x i32> <i32 0, i32 0, i32 6, i32 7>
|
|
ret <4 x i32> %r
|
|
}
|