[DAGCombiner] Add more rules to fold shuffles.
This patch adds two new rules to the DAGCombiner:
1. shuffle (shuffle A, Undef, M0), B, M1 -> shuffle A, B, M2
2. shuffle (shuffle A, Undef, M0), A, M1 -> shuffle A, Undef, M2
We only do this if the combined shuffle is legal for the target.
Example:
;;
define <4 x float> @test(<4 x float> %a, <4 x float> %b) {
%1 = shufflevector <4 x float> %a, <4 x float> undef, <4 x i32><i32 6, i32 0, i32 1, i32 7>
%2 = shufflevector <4 x float> %1, <4 x float> %b, <4 x i32><i32 1, i32 2, i32 4, i32 5>
ret <4 x i32> %2
}
;;
(using llc -mcpu=corei7 -march=x86-64)
Before, the x86 backend generated:
pshufd $120, %xmm0, %xmm0
shufps $-108, %xmm0, %xmm1
movaps %xmm1, %xmm0
Now the x86 backend generates:
movsd %xmm1, %xmm0
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@213069 91177308-0d34-0410-b5e6-96231b3b80d8
2014-07-15 13:26:28 +00:00
|
|
|
; RUN: llc < %s -mtriple=x86_64-unknown-linux-gnu -mcpu=corei7 | FileCheck %s
|
|
|
|
|
|
|
|
; Verify that we fold shuffles according to rule:
|
|
|
|
; (shuffle(shuffle A, Undef, M0), B, M1) -> (shuffle A, B, M2)
|
|
|
|
|
|
|
|
define <4 x float> @test1(<4 x float> %a, <4 x float> %b) {
|
|
|
|
%1 = shufflevector <4 x float> %a, <4 x float> undef, <4 x i32> <i32 4, i32 2, i32 3, i32 1>
|
|
|
|
%2 = shufflevector <4 x float> %1, <4 x float> %b, <4 x i32> <i32 4, i32 5, i32 1, i32 2>
|
|
|
|
ret <4 x float> %2
|
|
|
|
}
|
|
|
|
; CHECK-LABEL: test1
|
|
|
|
; Mask: [4,5,2,3]
|
|
|
|
; CHECK: movsd
|
|
|
|
; CHECK: ret
|
|
|
|
|
|
|
|
define <4 x float> @test2(<4 x float> %a, <4 x float> %b) {
|
|
|
|
%1 = shufflevector <4 x float> %a, <4 x float> undef, <4 x i32> <i32 6, i32 0, i32 1, i32 7>
|
|
|
|
%2 = shufflevector <4 x float> %1, <4 x float> %b, <4 x i32> <i32 1, i32 2, i32 4, i32 5>
|
|
|
|
ret <4 x float> %2
|
|
|
|
}
|
|
|
|
; CHECK-LABEL: test2
|
|
|
|
; Mask: [0,1,4,5]
|
|
|
|
; CHECK: movlhps
|
|
|
|
; CHECK: ret
|
|
|
|
|
|
|
|
define <4 x float> @test3(<4 x float> %a, <4 x float> %b) {
|
|
|
|
%1 = shufflevector <4 x float> %a, <4 x float> undef, <4 x i32> <i32 0, i32 5, i32 1, i32 7>
|
|
|
|
%2 = shufflevector <4 x float> %1, <4 x float> %b, <4 x i32> <i32 0, i32 2, i32 4, i32 1>
|
|
|
|
ret <4 x float> %2
|
|
|
|
}
|
|
|
|
; CHECK-LABEL: test3
|
|
|
|
; Mask: [0,1,4,u]
|
|
|
|
; CHECK: movlhps
|
|
|
|
; CHECK: ret
|
|
|
|
|
|
|
|
define <4 x float> @test4(<4 x float> %a, <4 x float> %b) {
|
|
|
|
%1 = shufflevector <4 x float> %a, <4 x float> undef, <4 x i32> <i32 2, i32 3, i32 5, i32 5>
|
|
|
|
%2 = shufflevector <4 x float> %1, <4 x float> %b, <4 x i32> <i32 6, i32 7, i32 0, i32 1>
|
|
|
|
ret <4 x float> %2
|
|
|
|
}
|
|
|
|
; CHECK-LABEL: test4
|
|
|
|
; Mask: [6,7,2,3]
|
|
|
|
; CHECK: movhlps
|
2014-07-16 11:29:39 +00:00
|
|
|
; CHECK-NEXT: ret
|
[DAGCombiner] Add more rules to fold shuffles.
This patch adds two new rules to the DAGCombiner:
1. shuffle (shuffle A, Undef, M0), B, M1 -> shuffle A, B, M2
2. shuffle (shuffle A, Undef, M0), A, M1 -> shuffle A, Undef, M2
We only do this if the combined shuffle is legal for the target.
Example:
;;
define <4 x float> @test(<4 x float> %a, <4 x float> %b) {
%1 = shufflevector <4 x float> %a, <4 x float> undef, <4 x i32><i32 6, i32 0, i32 1, i32 7>
%2 = shufflevector <4 x float> %1, <4 x float> %b, <4 x i32><i32 1, i32 2, i32 4, i32 5>
ret <4 x i32> %2
}
;;
(using llc -mcpu=corei7 -march=x86-64)
Before, the x86 backend generated:
pshufd $120, %xmm0, %xmm0
shufps $-108, %xmm0, %xmm1
movaps %xmm1, %xmm0
Now the x86 backend generates:
movsd %xmm1, %xmm0
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@213069 91177308-0d34-0410-b5e6-96231b3b80d8
2014-07-15 13:26:28 +00:00
|
|
|
|
|
|
|
define <4 x float> @test5(<4 x float> %a, <4 x float> %b) {
|
|
|
|
%1 = shufflevector <4 x float> %a, <4 x float> undef, <4 x i32> <i32 0, i32 4, i32 1, i32 3>
|
|
|
|
%2 = shufflevector <4 x float> %1, <4 x float> %b, <4 x i32> <i32 0, i32 2, i32 6, i32 7>
|
|
|
|
ret <4 x float> %2
|
|
|
|
}
|
|
|
|
; CHECK-LABEL: test5
|
|
|
|
; Mask: [0,1,6,7]
|
|
|
|
; CHECK: blendps $12
|
|
|
|
; CHECK: ret
|
|
|
|
|
|
|
|
; Verify that we fold shuffles according to rule:
|
|
|
|
; (shuffle(shuffle A, Undef, M0), A, M1) -> (shuffle A, Undef, M2)
|
|
|
|
|
|
|
|
define <4 x float> @test6(<4 x float> %a) {
|
|
|
|
%1 = shufflevector <4 x float> %a, <4 x float> undef, <4 x i32> <i32 4, i32 2, i32 3, i32 1>
|
|
|
|
%2 = shufflevector <4 x float> %1, <4 x float> %a, <4 x i32> <i32 4, i32 5, i32 1, i32 2>
|
|
|
|
ret <4 x float> %2
|
|
|
|
}
|
|
|
|
; CHECK-LABEL: test6
|
|
|
|
; Mask: [0,1,2,3]
|
|
|
|
; CHECK-NOT: pshufd
|
|
|
|
; CHECK-NOT: shufps
|
|
|
|
; CHECK-NOT: movlhps
|
|
|
|
; CHECK: ret
|
|
|
|
|
|
|
|
define <4 x float> @test7(<4 x float> %a) {
|
|
|
|
%1 = shufflevector <4 x float> %a, <4 x float> undef, <4 x i32> <i32 6, i32 0, i32 1, i32 7>
|
|
|
|
%2 = shufflevector <4 x float> %1, <4 x float> %a, <4 x i32> <i32 1, i32 2, i32 4, i32 5>
|
|
|
|
ret <4 x float> %2
|
|
|
|
}
|
|
|
|
; CHECK-LABEL: test7
|
|
|
|
; Mask: [0,1,0,1]
|
|
|
|
; CHECK-NOT: pshufd
|
|
|
|
; CHECK-NOT: shufps
|
|
|
|
; CHECK: movlhps
|
|
|
|
; CHECK-NEXT: ret
|
|
|
|
|
|
|
|
define <4 x float> @test8(<4 x float> %a) {
|
|
|
|
%1 = shufflevector <4 x float> %a, <4 x float> undef, <4 x i32> <i32 0, i32 5, i32 1, i32 7>
|
|
|
|
%2 = shufflevector <4 x float> %1, <4 x float> %a, <4 x i32> <i32 0, i32 2, i32 4, i32 1>
|
|
|
|
ret <4 x float> %2
|
|
|
|
}
|
|
|
|
; CHECK-LABEL: test8
|
|
|
|
; Mask: [0,1,0,u]
|
|
|
|
; CHECK-NOT: pshufd
|
|
|
|
; CHECK-NOT: shufps
|
|
|
|
; CHECK: movlhps
|
|
|
|
; CHECK-NEXT: ret
|
|
|
|
|
|
|
|
define <4 x float> @test9(<4 x float> %a) {
|
|
|
|
%1 = shufflevector <4 x float> %a, <4 x float> undef, <4 x i32> <i32 2, i32 3, i32 5, i32 5>
|
|
|
|
%2 = shufflevector <4 x float> %1, <4 x float> %a, <4 x i32> <i32 6, i32 7, i32 0, i32 1>
|
|
|
|
ret <4 x float> %2
|
|
|
|
}
|
|
|
|
; CHECK-LABEL: test9
|
|
|
|
; Mask: [2,3,2,3]
|
|
|
|
; CHECK-NOT: movlhps
|
|
|
|
; CHECK-NOT: palignr
|
|
|
|
; CHECK: movhlps
|
|
|
|
; CHECK-NEXT: ret
|
|
|
|
|
|
|
|
define <4 x float> @test10(<4 x float> %a) {
|
|
|
|
%1 = shufflevector <4 x float> %a, <4 x float> undef, <4 x i32> <i32 0, i32 4, i32 1, i32 3>
|
|
|
|
%2 = shufflevector <4 x float> %1, <4 x float> %a, <4 x i32> <i32 0, i32 2, i32 6, i32 7>
|
|
|
|
ret <4 x float> %2
|
|
|
|
}
|
|
|
|
; CHECK-LABEL: test10
|
|
|
|
; Mask: [0,1,2,3]
|
|
|
|
; CHECK-NOT: pshufd
|
|
|
|
; CHECK-NOT: shufps
|
|
|
|
; CHECK-NOT: movlhps
|
|
|
|
; CHECK: ret
|
|
|
|
|