; RUN: llc < %s -mtriple=x86_64-unknown-linux-gnu -mcpu=core-avx2 -mattr=+avx2 | FileCheck %s ; AVX2 Logical Shift Left define <16 x i16> @test_sllw_1(<16 x i16> %InVec) { entry: %shl = shl <16 x i16> %InVec, ret <16 x i16> %shl } ; CHECK-LABEL: test_sllw_1: ; CHECK-NOT: vpsllw $0, %ymm0, %ymm0 ; CHECK: ret define <16 x i16> @test_sllw_2(<16 x i16> %InVec) { entry: %shl = shl <16 x i16> %InVec, ret <16 x i16> %shl } ; CHECK-LABEL: test_sllw_2: ; CHECK: vpaddw %ymm0, %ymm0, %ymm0 ; CHECK: ret define <16 x i16> @test_sllw_3(<16 x i16> %InVec) { entry: %shl = shl <16 x i16> %InVec, ret <16 x i16> %shl } ; CHECK-LABEL: test_sllw_3: ; CHECK: vpsllw $15, %ymm0, %ymm0 ; CHECK: ret define <8 x i32> @test_slld_1(<8 x i32> %InVec) { entry: %shl = shl <8 x i32> %InVec, ret <8 x i32> %shl } ; CHECK-LABEL: test_slld_1: ; CHECK-NOT: vpslld $0, %ymm0, %ymm0 ; CHECK: ret define <8 x i32> @test_slld_2(<8 x i32> %InVec) { entry: %shl = shl <8 x i32> %InVec, ret <8 x i32> %shl } ; CHECK-LABEL: test_slld_2: ; CHECK: vpaddd %ymm0, %ymm0, %ymm0 ; CHECK: ret define <8 x i32> @test_vpslld_var(i32 %shift) { %amt = insertelement <8 x i32> undef, i32 %shift, i32 0 %tmp = shl <8 x i32> , %amt ret <8 x i32> %tmp } ; CHECK-LABEL: test_vpslld_var: ; CHECK: vpslld %xmm0, %ymm1, %ymm0 ; CHECK: ret define <8 x i32> @test_slld_3(<8 x i32> %InVec) { entry: %shl = shl <8 x i32> %InVec, ret <8 x i32> %shl } ; CHECK-LABEL: test_slld_3: ; CHECK: vpslld $31, %ymm0, %ymm0 ; CHECK: ret define <4 x i64> @test_sllq_1(<4 x i64> %InVec) { entry: %shl = shl <4 x i64> %InVec, ret <4 x i64> %shl } ; CHECK-LABEL: test_sllq_1: ; CHECK-NOT: vpsllq $0, %ymm0, %ymm0 ; CHECK: ret define <4 x i64> @test_sllq_2(<4 x i64> %InVec) { entry: %shl = shl <4 x i64> %InVec, ret <4 x i64> %shl } ; CHECK-LABEL: test_sllq_2: ; CHECK: vpaddq %ymm0, %ymm0, %ymm0 ; CHECK: ret define <4 x i64> @test_sllq_3(<4 x i64> %InVec) { entry: %shl = shl <4 x i64> %InVec, ret <4 x i64> %shl } ; CHECK-LABEL: test_sllq_3: ; CHECK: vpsllq $63, %ymm0, %ymm0 ; CHECK: ret ; AVX2 Arithmetic Shift define <16 x i16> @test_sraw_1(<16 x i16> %InVec) { entry: %shl = ashr <16 x i16> %InVec, ret <16 x i16> %shl } ; CHECK-LABEL: test_sraw_1: ; CHECK-NOT: vpsraw $0, %ymm0, %ymm0 ; CHECK: ret define <16 x i16> @test_sraw_2(<16 x i16> %InVec) { entry: %shl = ashr <16 x i16> %InVec, ret <16 x i16> %shl } ; CHECK-LABEL: test_sraw_2: ; CHECK: vpsraw $1, %ymm0, %ymm0 ; CHECK: ret define <16 x i16> @test_sraw_3(<16 x i16> %InVec) { entry: %shl = ashr <16 x i16> %InVec, ret <16 x i16> %shl } ; CHECK-LABEL: test_sraw_3: ; CHECK: vpsraw $15, %ymm0, %ymm0 ; CHECK: ret define <8 x i32> @test_srad_1(<8 x i32> %InVec) { entry: %shl = ashr <8 x i32> %InVec, ret <8 x i32> %shl } ; CHECK-LABEL: test_srad_1: ; CHECK-NOT: vpsrad $0, %ymm0, %ymm0 ; CHECK: ret define <8 x i32> @test_srad_2(<8 x i32> %InVec) { entry: %shl = ashr <8 x i32> %InVec, ret <8 x i32> %shl } ; CHECK-LABEL: test_srad_2: ; CHECK: vpsrad $1, %ymm0, %ymm0 ; CHECK: ret define <8 x i32> @test_srad_3(<8 x i32> %InVec) { entry: %shl = ashr <8 x i32> %InVec, ret <8 x i32> %shl } ; CHECK-LABEL: test_srad_3: ; CHECK: vpsrad $31, %ymm0, %ymm0 ; CHECK: ret ; SSE Logical Shift Right define <16 x i16> @test_srlw_1(<16 x i16> %InVec) { entry: %shl = lshr <16 x i16> %InVec, ret <16 x i16> %shl } ; CHECK-LABEL: test_srlw_1: ; CHECK-NOT: vpsrlw $0, %ymm0, %ymm0 ; CHECK: ret define <16 x i16> @test_srlw_2(<16 x i16> %InVec) { entry: %shl = lshr <16 x i16> %InVec, ret <16 x i16> %shl } ; CHECK-LABEL: test_srlw_2: ; CHECK: vpsrlw $1, %ymm0, %ymm0 ; CHECK: ret define <16 x i16> @test_srlw_3(<16 x i16> %InVec) { entry: %shl = lshr <16 x i16> %InVec, ret <16 x i16> %shl } ; CHECK-LABEL: test_srlw_3: ; CHECK: vpsrlw $15, %ymm0, %ymm0 ; CHECK: ret define <8 x i32> @test_srld_1(<8 x i32> %InVec) { entry: %shl = lshr <8 x i32> %InVec, ret <8 x i32> %shl } ; CHECK-LABEL: test_srld_1: ; CHECK-NOT: vpsrld $0, %ymm0, %ymm0 ; CHECK: ret define <8 x i32> @test_srld_2(<8 x i32> %InVec) { entry: %shl = lshr <8 x i32> %InVec, ret <8 x i32> %shl } ; CHECK-LABEL: test_srld_2: ; CHECK: vpsrld $1, %ymm0, %ymm0 ; CHECK: ret define <8 x i32> @test_srld_3(<8 x i32> %InVec) { entry: %shl = lshr <8 x i32> %InVec, ret <8 x i32> %shl } ; CHECK-LABEL: test_srld_3: ; CHECK: vpsrld $31, %ymm0, %ymm0 ; CHECK: ret define <4 x i64> @test_srlq_1(<4 x i64> %InVec) { entry: %shl = lshr <4 x i64> %InVec, ret <4 x i64> %shl } ; CHECK-LABEL: test_srlq_1: ; CHECK-NOT: vpsrlq $0, %ymm0, %ymm0 ; CHECK: ret define <4 x i64> @test_srlq_2(<4 x i64> %InVec) { entry: %shl = lshr <4 x i64> %InVec, ret <4 x i64> %shl } ; CHECK-LABEL: test_srlq_2: ; CHECK: vpsrlq $1, %ymm0, %ymm0 ; CHECK: ret define <4 x i64> @test_srlq_3(<4 x i64> %InVec) { entry: %shl = lshr <4 x i64> %InVec, ret <4 x i64> %shl } ; CHECK-LABEL: test_srlq_3: ; CHECK: vpsrlq $63, %ymm0, %ymm0 ; CHECK: ret ; CHECK-LABEL: @srl_trunc_and_v4i64 ; CHECK: vpand ; CHECK-NEXT: vpsrlvd ; CHECK: ret define <4 x i32> @srl_trunc_and_v4i64(<4 x i32> %x, <4 x i64> %y) nounwind { %and = and <4 x i64> %y, %trunc = trunc <4 x i64> %and to <4 x i32> %sra = lshr <4 x i32> %x, %trunc ret <4 x i32> %sra } ; ; Vectorized byte shifts ; define <8 x i16> @shl_8i16(<8 x i16> %r, <8 x i16> %a) nounwind { ; CHECK-LABEL: shl_8i16 ; CHECK: vpmovzxwd {{.*#+}} ymm1 = xmm1[0],zero,xmm1[1],zero,xmm1[2],zero,xmm1[3],zero,xmm1[4],zero,xmm1[5],zero,xmm1[6],zero,xmm1[7],zero ; CHECK-NEXT: vpmovzxwd {{.*#+}} ymm0 = xmm0[0],zero,xmm0[1],zero,xmm0[2],zero,xmm0[3],zero,xmm0[4],zero,xmm0[5],zero,xmm0[6],zero,xmm0[7],zero ; CHECK-NEXT: vpsllvd %ymm1, %ymm0, %ymm0 ; CHECK-NEXT: vpshufb {{.*#+}} ymm0 = ymm0[0,1,4,5,8,9,12,13],zero,zero,zero,zero,zero,zero,zero,zero,ymm0[16,17,20,21,24,25,28,29],zero,zero,zero,zero,zero,zero,zero,zero ; CHECK-NEXT: vpermq {{.*#+}} ymm0 = ymm0[0,2,2,3] ; CHECK: retq %shl = shl <8 x i16> %r, %a ret <8 x i16> %shl } define <16 x i16> @shl_16i16(<16 x i16> %r, <16 x i16> %a) nounwind { ; CHECK-LABEL: shl_16i16 ; CHECK: vpxor %ymm2, %ymm2, %ymm2 ; CHECK-NEXT: vpunpckhwd {{.*#+}} ymm3 = ymm1[4],ymm2[4],ymm1[5],ymm2[5],ymm1[6],ymm2[6],ymm1[7],ymm2[7],ymm1[12],ymm2[12],ymm1[13],ymm2[13],ymm1[14],ymm2[14],ymm1[15],ymm2[15] ; CHECK-NEXT: vpunpckhwd {{.*#+}} ymm4 = ymm0[4,4,5,5,6,6,7,7,12,12,13,13,14,14,15,15] ; CHECK-NEXT: vpsllvd %ymm3, %ymm4, %ymm3 ; CHECK-NEXT: vpsrld $16, %ymm3, %ymm3 ; CHECK-NEXT: vpunpcklwd {{.*#+}} ymm1 = ymm1[0],ymm2[0],ymm1[1],ymm2[1],ymm1[2],ymm2[2],ymm1[3],ymm2[3],ymm1[8],ymm2[8],ymm1[9],ymm2[9],ymm1[10],ymm2[10],ymm1[11],ymm2[11] ; CHECK-NEXT: vpunpcklwd {{.*#+}} ymm0 = ymm0[0,0,1,1,2,2,3,3,8,8,9,9,10,10,11,11] ; CHECK-NEXT: vpsllvd %ymm1, %ymm0, %ymm0 ; CHECK-NEXT: vpsrld $16, %ymm0, %ymm0 ; CHECK-NEXT: vpackusdw %ymm3, %ymm0, %ymm0 ; CHECK-NEXT: retq %shl = shl <16 x i16> %r, %a ret <16 x i16> %shl } define <32 x i8> @shl_32i8(<32 x i8> %r, <32 x i8> %a) nounwind { ; CHECK-LABEL: shl_32i8 ; CHECK: vpsllw $5, %ymm1, %ymm1 ; CHECK-NEXT: vpsllw $4, %ymm0, %ymm2 ; CHECK-NEXT: vpand {{.*}}(%rip), %ymm2, %ymm2 ; CHECK-NEXT: vpblendvb %ymm1, %ymm2, %ymm0, %ymm0 ; CHECK-NEXT: vpsllw $2, %ymm0, %ymm2 ; CHECK-NEXT: vpand {{.*}}(%rip), %ymm2, %ymm2 ; CHECK-NEXT: vpaddb %ymm1, %ymm1, %ymm1 ; CHECK-NEXT: vpblendvb %ymm1, %ymm2, %ymm0, %ymm0 ; CHECK-NEXT: vpaddb %ymm0, %ymm0, %ymm2 ; CHECK-NEXT: vpaddb %ymm1, %ymm1, %ymm1 ; CHECK-NEXT: vpblendvb %ymm1, %ymm2, %ymm0, %ymm0 ; CHECK-NEXT: retq %shl = shl <32 x i8> %r, %a ret <32 x i8> %shl } define <8 x i16> @ashr_8i16(<8 x i16> %r, <8 x i16> %a) nounwind { ; CHECK-LABEL: ashr_8i16 ; CHECK: vpmovzxwd {{.*#+}} ymm1 = xmm1[0],zero,xmm1[1],zero,xmm1[2],zero,xmm1[3],zero,xmm1[4],zero,xmm1[5],zero,xmm1[6],zero,xmm1[7],zero ; CHECK-NEXT: vpmovsxwd %xmm0, %ymm0 ; CHECK-NEXT: vpsravd %ymm1, %ymm0, %ymm0 ; CHECK-NEXT: vpshufb {{.*#+}} ymm0 = ymm0[0,1,4,5,8,9,12,13],zero,zero,zero,zero,zero,zero,zero,zero,ymm0[16,17,20,21,24,25,28,29],zero,zero,zero,zero,zero,zero,zero,zero ; CHECK-NEXT: vpermq {{.*#+}} ymm0 = ymm0[0,2,2,3] ; CHECK: retq %ashr = ashr <8 x i16> %r, %a ret <8 x i16> %ashr } define <16 x i16> @ashr_16i16(<16 x i16> %r, <16 x i16> %a) nounwind { ; CHECK-LABEL: ashr_16i16 ; CHECK: vpxor %ymm2, %ymm2, %ymm2 ; CHECK-NEXT: vpunpckhwd {{.*#+}} ymm3 = ymm1[4],ymm2[4],ymm1[5],ymm2[5],ymm1[6],ymm2[6],ymm1[7],ymm2[7],ymm1[12],ymm2[12],ymm1[13],ymm2[13],ymm1[14],ymm2[14],ymm1[15],ymm2[15] ; CHECK-NEXT: vpunpckhwd {{.*#+}} ymm4 = ymm0[4,4,5,5,6,6,7,7,12,12,13,13,14,14,15,15] ; CHECK-NEXT: vpsravd %ymm3, %ymm4, %ymm3 ; CHECK-NEXT: vpsrld $16, %ymm3, %ymm3 ; CHECK-NEXT: vpunpcklwd {{.*#+}} ymm1 = ymm1[0],ymm2[0],ymm1[1],ymm2[1],ymm1[2],ymm2[2],ymm1[3],ymm2[3],ymm1[8],ymm2[8],ymm1[9],ymm2[9],ymm1[10],ymm2[10],ymm1[11],ymm2[11] ; CHECK-NEXT: vpunpcklwd {{.*#+}} ymm0 = ymm0[0,0,1,1,2,2,3,3,8,8,9,9,10,10,11,11] ; CHECK-NEXT: vpsravd %ymm1, %ymm0, %ymm0 ; CHECK-NEXT: vpsrld $16, %ymm0, %ymm0 ; CHECK-NEXT: vpackusdw %ymm3, %ymm0, %ymm0 ; CHECK-NEXT: retq %ashr = ashr <16 x i16> %r, %a ret <16 x i16> %ashr } define <32 x i8> @ashr_32i8(<32 x i8> %r, <32 x i8> %a) nounwind { ; CHECK-LABEL: ashr_32i8 ; CHECK: vpsllw $5, %ymm1, %ymm1 ; CHECK-NEXT: vpunpckhbw {{.*#+}} ymm2 = ymm0[8],ymm1[8],ymm0[9],ymm1[9],ymm0[10],ymm1[10],ymm0[11],ymm1[11],ymm0[12],ymm1[12],ymm0[13],ymm1[13],ymm0[14],ymm1[14],ymm0[15],ymm1[15],ymm0[24],ymm1[24],ymm0[25],ymm1[25],ymm0[26],ymm1[26],ymm0[27],ymm1[27],ymm0[28],ymm1[28],ymm0[29],ymm1[29],ymm0[30],ymm1[30],ymm0[31],ymm1[31] ; CHECK-NEXT: vpunpckhbw {{.*#+}} ymm3 = ymm0[8,8,9,9,10,10,11,11,12,12,13,13,14,14,15,15,24,24,25,25,26,26,27,27,28,28,29,29,30,30,31,31] ; CHECK-NEXT: vpsraw $4, %ymm3, %ymm4 ; CHECK-NEXT: vpblendvb %ymm2, %ymm4, %ymm3, %ymm3 ; CHECK-NEXT: vpsraw $2, %ymm3, %ymm4 ; CHECK-NEXT: vpaddw %ymm2, %ymm2, %ymm2 ; CHECK-NEXT: vpblendvb %ymm2, %ymm4, %ymm3, %ymm3 ; CHECK-NEXT: vpsraw $1, %ymm3, %ymm4 ; CHECK-NEXT: vpaddw %ymm2, %ymm2, %ymm2 ; CHECK-NEXT: vpblendvb %ymm2, %ymm4, %ymm3, %ymm2 ; CHECK-NEXT: vpsrlw $8, %ymm2, %ymm2 ; CHECK-NEXT: vpunpcklbw {{.*#+}} ymm1 = ymm0[0],ymm1[0],ymm0[1],ymm1[1],ymm0[2],ymm1[2],ymm0[3],ymm1[3],ymm0[4],ymm1[4],ymm0[5],ymm1[5],ymm0[6],ymm1[6],ymm0[7],ymm1[7],ymm0[16],ymm1[16],ymm0[17],ymm1[17],ymm0[18],ymm1[18],ymm0[19],ymm1[19],ymm0[20],ymm1[20],ymm0[21],ymm1[21],ymm0[22],ymm1[22],ymm0[23],ymm1[23] ; CHECK-NEXT: vpunpcklbw {{.*#+}} ymm0 = ymm0[0,0,1,1,2,2,3,3,4,4,5,5,6,6,7,7,16,16,17,17,18,18,19,19,20,20,21,21,22,22,23,23] ; CHECK-NEXT: vpsraw $4, %ymm0, %ymm3 ; CHECK-NEXT: vpblendvb %ymm1, %ymm3, %ymm0, %ymm0 ; CHECK-NEXT: vpsraw $2, %ymm0, %ymm3 ; CHECK-NEXT: vpaddw %ymm1, %ymm1, %ymm1 ; CHECK-NEXT: vpblendvb %ymm1, %ymm3, %ymm0, %ymm0 ; CHECK-NEXT: vpsraw $1, %ymm0, %ymm3 ; CHECK-NEXT: vpaddw %ymm1, %ymm1, %ymm1 ; CHECK-NEXT: vpblendvb %ymm1, %ymm3, %ymm0, %ymm0 ; CHECK-NEXT: vpsrlw $8, %ymm0, %ymm0 ; CHECK-NEXT: vpackuswb %ymm2, %ymm0, %ymm0 ; CHECK-NEXT: retq %ashr = ashr <32 x i8> %r, %a ret <32 x i8> %ashr } define <8 x i16> @lshr_8i16(<8 x i16> %r, <8 x i16> %a) nounwind { ; CHECK-LABEL: lshr_8i16 ; CHECK: vpmovzxwd {{.*#+}} ymm1 = xmm1[0],zero,xmm1[1],zero,xmm1[2],zero,xmm1[3],zero,xmm1[4],zero,xmm1[5],zero,xmm1[6],zero,xmm1[7],zero ; CHECK-NEXT: vpmovzxwd {{.*#+}} ymm0 = xmm0[0],zero,xmm0[1],zero,xmm0[2],zero,xmm0[3],zero,xmm0[4],zero,xmm0[5],zero,xmm0[6],zero,xmm0[7],zero ; CHECK-NEXT: vpsrlvd %ymm1, %ymm0, %ymm0 ; CHECK-NEXT: vpshufb {{.*#+}} ymm0 = ymm0[0,1,4,5,8,9,12,13],zero,zero,zero,zero,zero,zero,zero,zero,ymm0[16,17,20,21,24,25,28,29],zero,zero,zero,zero,zero,zero,zero,zero ; CHECK-NEXT: vpermq {{.*#+}} ymm0 = ymm0[0,2,2,3] ; CHECK: retq %lshr = lshr <8 x i16> %r, %a ret <8 x i16> %lshr } define <16 x i16> @lshr_16i16(<16 x i16> %r, <16 x i16> %a) nounwind { ; CHECK-LABEL: lshr_16i16 ; CHECK: vpxor %ymm2, %ymm2, %ymm2 ; CHECK-NEXT: vpunpckhwd {{.*#+}} ymm3 = ymm1[4],ymm2[4],ymm1[5],ymm2[5],ymm1[6],ymm2[6],ymm1[7],ymm2[7],ymm1[12],ymm2[12],ymm1[13],ymm2[13],ymm1[14],ymm2[14],ymm1[15],ymm2[15] ; CHECK-NEXT: vpunpckhwd {{.*#+}} ymm4 = ymm0[4,4,5,5,6,6,7,7,12,12,13,13,14,14,15,15] ; CHECK-NEXT: vpsrlvd %ymm3, %ymm4, %ymm3 ; CHECK-NEXT: vpsrld $16, %ymm3, %ymm3 ; CHECK-NEXT: vpunpcklwd {{.*#+}} ymm1 = ymm1[0],ymm2[0],ymm1[1],ymm2[1],ymm1[2],ymm2[2],ymm1[3],ymm2[3],ymm1[8],ymm2[8],ymm1[9],ymm2[9],ymm1[10],ymm2[10],ymm1[11],ymm2[11] ; CHECK-NEXT: vpunpcklwd {{.*#+}} ymm0 = ymm0[0,0,1,1,2,2,3,3,8,8,9,9,10,10,11,11] ; CHECK-NEXT: vpsrlvd %ymm1, %ymm0, %ymm0 ; CHECK-NEXT: vpsrld $16, %ymm0, %ymm0 ; CHECK-NEXT: vpackusdw %ymm3, %ymm0, %ymm0 ; CHECK-NEXT: retq %lshr = lshr <16 x i16> %r, %a ret <16 x i16> %lshr } define <32 x i8> @lshr_32i8(<32 x i8> %r, <32 x i8> %a) nounwind { ; CHECK-LABEL: lshr_32i8 ; CHECK: vpsllw $5, %ymm1, %ymm1 ; CHECK-NEXT: vpsrlw $4, %ymm0, %ymm2 ; CHECK-NEXT: vpand {{.*}}(%rip), %ymm2, %ymm2 ; CHECK-NEXT: vpblendvb %ymm1, %ymm2, %ymm0, %ymm0 ; CHECK-NEXT: vpsrlw $2, %ymm0, %ymm2 ; CHECK-NEXT: vpand {{.*}}(%rip), %ymm2, %ymm2 ; CHECK-NEXT: vpaddb %ymm1, %ymm1, %ymm1 ; CHECK-NEXT: vpblendvb %ymm1, %ymm2, %ymm0, %ymm0 ; CHECK-NEXT: vpsrlw $1, %ymm0, %ymm2 ; CHECK-NEXT: vpand {{.*}}(%rip), %ymm2, %ymm2 ; CHECK-NEXT: vpaddb %ymm1, %ymm1, %ymm1 ; CHECK-NEXT: vpblendvb %ymm1, %ymm2, %ymm0, %ymm0 ; CHECK-NEXT: retq %lshr = lshr <32 x i8> %r, %a ret <32 x i8> %lshr }