llvm/test/CodeGen/SystemZ/vec-ctpop-01.ll

   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