Home | History | Annotate | Download | only in SystemZ
      1 ; Test vector population-count instruction
      2 ;
      3 ; RUN: llc < %s -mtriple=s390x-linux-gnu -mcpu=z13 | FileCheck %s
      4 
      5 declare <16 x i8> @llvm.ctpop.v16i8(<16 x i8> %a)
      6 declare <8 x i16> @llvm.ctpop.v8i16(<8 x i16> %a)
      7 declare <4 x i32> @llvm.ctpop.v4i32(<4 x i32> %a)
      8 declare <2 x i64> @llvm.ctpop.v2i64(<2 x i64> %a)
      9 
     10 define <16 x i8> @f1(<16 x i8> %a) {
     11 ; CHECK-LABEL: f1:
     12 ; CHECK: vpopct  %v24, %v24, 0
     13 ; CHECK: br      %r14
     14 
     15   %popcnt = call <16 x i8> @llvm.ctpop.v16i8(<16 x i8> %a)
     16   ret <16 x i8> %popcnt
     17 }
     18 
     19 define <8 x i16> @f2(<8 x i16> %a) {
     20 ; CHECK-LABEL: f2:
     21 ; CHECK: vpopct  [[T1:%v[0-9]+]], %v24, 0
     22 ; CHECK: veslh   [[T2:%v[0-9]+]], [[T1]], 8
     23 ; CHECK: vah     [[T3:%v[0-9]+]], [[T1]], [[T2]]
     24 ; CHECK: vesrlh  %v24, [[T3]], 8
     25 ; CHECK: br      %r14
     26 
     27   %popcnt = call <8 x i16> @llvm.ctpop.v8i16(<8 x i16> %a)
     28   ret <8 x i16> %popcnt
     29 }
     30 
     31 define <4 x i32> @f3(<4 x i32> %a) {
     32 ; CHECK-LABEL: f3:
     33 ; CHECK: vpopct  [[T1:%v[0-9]+]], %v24, 0
     34 ; CHECK: vgbm    [[T2:%v[0-9]+]], 0
     35 ; CHECK: vsumb   %v24, [[T1]], [[T2]]
     36 ; CHECK: br      %r14
     37 
     38   %popcnt = call <4 x i32> @llvm.ctpop.v4i32(<4 x i32> %a)
     39   ret <4 x i32> %popcnt
     40 }
     41 
     42 define <2 x i64> @f4(<2 x i64> %a) {
     43 ; CHECK-LABEL: f4:
     44 ; CHECK: vpopct  [[T1:%v[0-9]+]], %v24, 0
     45 ; CHECK: vgbm    [[T2:%v[0-9]+]], 0
     46 ; CHECK: vsumb   [[T3:%v[0-9]+]], [[T1]], [[T2]]
     47 ; CHECK: vsumgf  %v24, [[T3]], [[T2]]
     48 ; CHECK: br      %r14
     49 
     50   %popcnt = call <2 x i64> @llvm.ctpop.v2i64(<2 x i64> %a)
     51   ret <2 x i64> %popcnt
     52 }
     53 
     54