RPCS3/llvm-mirror
1
0
Fork 0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2024-11-22 18:54:02 +01:00
Code Issues Projects Releases Wiki Activity

[X86][AVX] lowerBuildVectorAsBroadcast - improve i64 BROADCASTM lowering on 32-bit targets

Browse Source 
We already handle the the cases where we have a 'zero extended splat' build vector (a, 0, 0, 0, a, 0, 0, 0, ...) but were missing the case where the 'a' scalar was zero-extended as well - such as i64 -> vXi64 splat cases on 32-bit targets.
This commit is contained in:
Simon Pilgrim 2020-09-18 16:35:39 +01:00
parent 975db31f6e
commit 39e440fde2
4 changed files with 42 additions and 101 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lib/Target/X86/X86ISelLowering.cpp
View File

@ -8655,10 +8655,14 @@ static SDValue lowerBuildVectorAsBroadcast(BuildVectorSDNode *BVOp,
SDValue BOperand;
SDValue ZeroExtended = isSplatZeroExtended(BVOp, NumElts, EltType);
if ((ZeroExtended && ZeroExtended.getOpcode() == ISD::BITCAST) ||
(ZeroExtended && ZeroExtended.getOpcode() == ISD::ZERO_EXTEND &&
ZeroExtended.getOperand(0).getOpcode() == ISD::BITCAST) ||
(Ld && Ld.getOpcode() == ISD::ZERO_EXTEND &&
Ld.getOperand(0).getOpcode() == ISD::BITCAST)) {
if (ZeroExtended)
if (ZeroExtended && ZeroExtended.getOpcode() == ISD::BITCAST)
BOperand = ZeroExtended.getOperand(0);
else if (ZeroExtended)
BOperand = ZeroExtended.getOperand(0).getOperand(0);
else
BOperand = Ld.getOperand(0).getOperand(0);
MVT MaskVT = BOperand.getSimpleValueType();

19
test/CodeGen/X86/avx512cd-intrinsics-fast-isel.ll
View File

@ -3,20 +3,11 @@
; RUN: llc < %s -fast-isel -mtriple=x86_64-unknown -mattr=+avx512cd | FileCheck %s --check-prefixes=CHECK,X64
define <8 x i64> @test_mm512_broadcastmb_epi64(<8 x i64> %a, <8 x i64> %b) {
; X86-LABEL: test_mm512_broadcastmb_epi64:
; X86: # %bb.0: # %entry
; X86-NEXT: vpcmpeqq %zmm1, %zmm0, %k0
; X86-NEXT: kmovw %k0, %eax
; X86-NEXT: movzbl %al, %eax
; X86-NEXT: vmovd %eax, %xmm0
; X86-NEXT: vpbroadcastq %xmm0, %zmm0
; X86-NEXT: retl
;
; X64-LABEL: test_mm512_broadcastmb_epi64:
; X64: # %bb.0: # %entry
; X64-NEXT: vpcmpeqq %zmm1, %zmm0, %k0
; X64-NEXT: vpbroadcastmb2q %k0, %zmm0
; X64-NEXT: retq
; CHECK-LABEL: test_mm512_broadcastmb_epi64:
; CHECK: # %bb.0: # %entry
; CHECK-NEXT: vpcmpeqq %zmm1, %zmm0, %k0
; CHECK-NEXT: vpbroadcastmb2q %k0, %zmm0
; CHECK-NEXT: ret{{[l|q]}}
entry:
%0 = icmp eq <8 x i64> %a, %b
%1 = bitcast <8 x i1> %0 to i8

36
test/CodeGen/X86/avx512vlcd-intrinsics-fast-isel.ll
View File

@ -3,19 +3,11 @@
; RUN: llc < %s -fast-isel -mtriple=x86_64-unknown-unknown -mattr=+avx512cd,+avx512vl | FileCheck %s --check-prefixes=CHECK,X64
define <2 x i64> @test_mm_broadcastmb_epi64(<2 x i64> %a, <2 x i64> %b) {
; X86-LABEL: test_mm_broadcastmb_epi64:
; X86: # %bb.0: # %entry
; X86-NEXT: vpcmpeqd %xmm1, %xmm0, %k0
; X86-NEXT: kmovw %k0, %eax
; X86-NEXT: vmovd %eax, %xmm0
; X86-NEXT: vpbroadcastq %xmm0, %xmm0
; X86-NEXT: retl
;
; X64-LABEL: test_mm_broadcastmb_epi64:
; X64: # %bb.0: # %entry
; X64-NEXT: vpcmpeqd %xmm1, %xmm0, %k0
; X64-NEXT: vpbroadcastmb2q %k0, %xmm0
; X64-NEXT: retq
; CHECK-LABEL: test_mm_broadcastmb_epi64:
; CHECK: # %bb.0: # %entry
; CHECK-NEXT: vpcmpeqd %xmm1, %xmm0, %k0
; CHECK-NEXT: vpbroadcastmb2q %k0, %xmm0
; CHECK-NEXT: ret{{[l|q]}}
entry:
%0 = bitcast <2 x i64> %a to <4 x i32>
%1 = bitcast <2 x i64> %b to <4 x i32>
@ -29,19 +21,11 @@ entry:
}
define <4 x i64> @test_mm256_broadcastmb_epi64(<4 x i64> %a, <4 x i64> %b) {
; X86-LABEL: test_mm256_broadcastmb_epi64:
; X86: # %bb.0: # %entry
; X86-NEXT: vpcmpeqq %ymm1, %ymm0, %k0
; X86-NEXT: kmovw %k0, %eax
; X86-NEXT: vmovd %eax, %xmm0
; X86-NEXT: vpbroadcastq %xmm0, %ymm0
; X86-NEXT: retl
;
; X64-LABEL: test_mm256_broadcastmb_epi64:
; X64: # %bb.0: # %entry
; X64-NEXT: vpcmpeqq %ymm1, %ymm0, %k0
; X64-NEXT: vpbroadcastmb2q %k0, %ymm0
; X64-NEXT: retq
; CHECK-LABEL: test_mm256_broadcastmb_epi64:
; CHECK: # %bb.0: # %entry
; CHECK-NEXT: vpcmpeqq %ymm1, %ymm0, %k0
; CHECK-NEXT: vpbroadcastmb2q %k0, %ymm0
; CHECK-NEXT: ret{{[l|q]}}
entry:
%0 = icmp eq <4 x i64> %a, %b
%1 = shufflevector <4 x i1> %0, <4 x i1> zeroinitializer, <8 x i32> <i32 0, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6, i32 7>

82
test/CodeGen/X86/broadcastm-lowering.ll
View File

@ -29,20 +29,11 @@ define <2 x i64> @test_mm_epi64(<8 x i16> %a, <8 x i16> %b) {
; X86-AVX512CD-NEXT: vzeroupper
; X86-AVX512CD-NEXT: retl
;
; X64-AVX512VLCDBW-LABEL: test_mm_epi64:
; X64-AVX512VLCDBW: # %bb.0: # %entry
; X64-AVX512VLCDBW-NEXT: vpcmpeqw %xmm1, %xmm0, %k0
; X64-AVX512VLCDBW-NEXT: vpbroadcastmb2q %k0, %xmm0
; X64-AVX512VLCDBW-NEXT: retq
;
; X86-AVX512VLCDBW-LABEL: test_mm_epi64:
; X86-AVX512VLCDBW: # %bb.0: # %entry
; X86-AVX512VLCDBW-NEXT: vpcmpeqw %xmm1, %xmm0, %k0
; X86-AVX512VLCDBW-NEXT: kmovd %k0, %eax
; X86-AVX512VLCDBW-NEXT: movzbl %al, %eax
; X86-AVX512VLCDBW-NEXT: vmovd %eax, %xmm0
; X86-AVX512VLCDBW-NEXT: vpshufb {{.*#+}} xmm0 = xmm0[0,1,2,3],zero,zero,zero,zero,xmm0[0,1,2,3],zero,zero,zero,zero
; X86-AVX512VLCDBW-NEXT: retl
; AVX512VLCDBW-LABEL: test_mm_epi64:
; AVX512VLCDBW: # %bb.0: # %entry
; AVX512VLCDBW-NEXT: vpcmpeqw %xmm1, %xmm0, %k0
; AVX512VLCDBW-NEXT: vpbroadcastmb2q %k0, %xmm0
; AVX512VLCDBW-NEXT: ret{{[l|q]}}
entry:
%0 = icmp eq <8 x i16> %a, %b
%1 = bitcast <8 x i1> %0 to i8
@ -91,39 +82,19 @@ entry:
}
define <8 x i64> @test_mm512_epi64(<8 x i32> %a, <8 x i32> %b) {
; X64-AVX512CD-LABEL: test_mm512_epi64:
; X64-AVX512CD: # %bb.0: # %entry
; X64-AVX512CD-NEXT: # kill: def $ymm1 killed $ymm1 def $zmm1
; X64-AVX512CD-NEXT: # kill: def $ymm0 killed $ymm0 def $zmm0
; X64-AVX512CD-NEXT: vpcmpeqd %zmm1, %zmm0, %k0
; X64-AVX512CD-NEXT: vpbroadcastmb2q %k0, %zmm0
; X64-AVX512CD-NEXT: retq
; AVX512CD-LABEL: test_mm512_epi64:
; AVX512CD: # %bb.0: # %entry
; AVX512CD-NEXT: # kill: def $ymm1 killed $ymm1 def $zmm1
; AVX512CD-NEXT: # kill: def $ymm0 killed $ymm0 def $zmm0
; AVX512CD-NEXT: vpcmpeqd %zmm1, %zmm0, %k0
; AVX512CD-NEXT: vpbroadcastmb2q %k0, %zmm0
; AVX512CD-NEXT: ret{{[l|q]}}
;
; X86-AVX512CD-LABEL: test_mm512_epi64:
; X86-AVX512CD: # %bb.0: # %entry
; X86-AVX512CD-NEXT: # kill: def $ymm1 killed $ymm1 def $zmm1
; X86-AVX512CD-NEXT: # kill: def $ymm0 killed $ymm0 def $zmm0
; X86-AVX512CD-NEXT: vpcmpeqd %zmm1, %zmm0, %k0
; X86-AVX512CD-NEXT: kmovw %k0, %eax
; X86-AVX512CD-NEXT: movzbl %al, %eax
; X86-AVX512CD-NEXT: vmovd %eax, %xmm0
; X86-AVX512CD-NEXT: vpbroadcastq %xmm0, %zmm0
; X86-AVX512CD-NEXT: retl
;
; X64-AVX512VLCDBW-LABEL: test_mm512_epi64:
; X64-AVX512VLCDBW: # %bb.0: # %entry
; X64-AVX512VLCDBW-NEXT: vpcmpeqd %ymm1, %ymm0, %k0
; X64-AVX512VLCDBW-NEXT: vpbroadcastmb2q %k0, %zmm0
; X64-AVX512VLCDBW-NEXT: retq
;
; X86-AVX512VLCDBW-LABEL: test_mm512_epi64:
; X86-AVX512VLCDBW: # %bb.0: # %entry
; X86-AVX512VLCDBW-NEXT: vpcmpeqd %ymm1, %ymm0, %k0
; X86-AVX512VLCDBW-NEXT: kmovd %k0, %eax
; X86-AVX512VLCDBW-NEXT: movzbl %al, %eax
; X86-AVX512VLCDBW-NEXT: vmovd %eax, %xmm0
; X86-AVX512VLCDBW-NEXT: vpbroadcastq %xmm0, %zmm0
; X86-AVX512VLCDBW-NEXT: retl
; AVX512VLCDBW-LABEL: test_mm512_epi64:
; AVX512VLCDBW: # %bb.0: # %entry
; AVX512VLCDBW-NEXT: vpcmpeqd %ymm1, %ymm0, %k0
; AVX512VLCDBW-NEXT: vpbroadcastmb2q %k0, %zmm0
; AVX512VLCDBW-NEXT: ret{{[l|q]}}
entry:
%0 = icmp eq <8 x i32> %a, %b
%1 = bitcast <8 x i1> %0 to i8
@ -156,20 +127,11 @@ define <4 x i64> @test_mm256_epi64(<8 x i32> %a, <8 x i32> %b) {
; X86-AVX512CD-NEXT: vpbroadcastq %xmm0, %ymm0
; X86-AVX512CD-NEXT: retl
;
; X64-AVX512VLCDBW-LABEL: test_mm256_epi64:
; X64-AVX512VLCDBW: # %bb.0: # %entry
; X64-AVX512VLCDBW-NEXT: vpcmpeqd %ymm1, %ymm0, %k0
; X64-AVX512VLCDBW-NEXT: vpbroadcastmb2q %k0, %ymm0
; X64-AVX512VLCDBW-NEXT: retq
;
; X86-AVX512VLCDBW-LABEL: test_mm256_epi64:
; X86-AVX512VLCDBW: # %bb.0: # %entry
; X86-AVX512VLCDBW-NEXT: vpcmpeqd %ymm1, %ymm0, %k0
; X86-AVX512VLCDBW-NEXT: kmovd %k0, %eax
; X86-AVX512VLCDBW-NEXT: movzbl %al, %eax
; X86-AVX512VLCDBW-NEXT: vmovd %eax, %xmm0
; X86-AVX512VLCDBW-NEXT: vpbroadcastq %xmm0, %ymm0
; X86-AVX512VLCDBW-NEXT: retl
; AVX512VLCDBW-LABEL: test_mm256_epi64:
; AVX512VLCDBW: # %bb.0: # %entry
; AVX512VLCDBW-NEXT: vpcmpeqd %ymm1, %ymm0, %k0
; AVX512VLCDBW-NEXT: vpbroadcastmb2q %k0, %ymm0
; AVX512VLCDBW-NEXT: ret{{[l|q]}}
entry:
%0 = icmp eq <8 x i32> %a, %b
%1 = bitcast <8 x i1> %0 to i8