summaryrefslogtreecommitdiff
path: root/test/CodeGen/SystemZ/vec-ctpop-01.ll
blob: 0056af73a2e159c09f3086b966335967401a0ceb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
; Test vector population-count instruction
;
; RUN: llc < %s -mtriple=s390x-linux-gnu -mcpu=z13 | FileCheck %s

declare <16 x i8> @llvm.ctpop.v16i8(<16 x i8> %a)
declare <8 x i16> @llvm.ctpop.v8i16(<8 x i16> %a)
declare <4 x i32> @llvm.ctpop.v4i32(<4 x i32> %a)
declare <2 x i64> @llvm.ctpop.v2i64(<2 x i64> %a)

define <16 x i8> @f1(<16 x i8> %a) {
; CHECK-LABEL: f1:
; CHECK: vpopct  %v24, %v24, 0
; CHECK: br      %r14

  %popcnt = call <16 x i8> @llvm.ctpop.v16i8(<16 x i8> %a)
  ret <16 x i8> %popcnt
}

define <8 x i16> @f2(<8 x i16> %a) {
; CHECK-LABEL: f2:
; CHECK: vpopct  [[T1:%v[0-9]+]], %v24, 0
; CHECK: veslh   [[T2:%v[0-9]+]], [[T1]], 8
; CHECK: vah     [[T3:%v[0-9]+]], [[T1]], [[T2]]
; CHECK: vesrlh  %v24, [[T3]], 8
; CHECK: br      %r14

  %popcnt = call <8 x i16> @llvm.ctpop.v8i16(<8 x i16> %a)
  ret <8 x i16> %popcnt
}

define <4 x i32> @f3(<4 x i32> %a) {
; CHECK-LABEL: f3:
; CHECK: vpopct  [[T1:%v[0-9]+]], %v24, 0
; CHECK: vgbm    [[T2:%v[0-9]+]], 0
; CHECK: vsumb   %v24, [[T1]], [[T2]]
; CHECK: br      %r14

  %popcnt = call <4 x i32> @llvm.ctpop.v4i32(<4 x i32> %a)
  ret <4 x i32> %popcnt
}

define <2 x i64> @f4(<2 x i64> %a) {
; CHECK-LABEL: f4:
; CHECK: vpopct  [[T1:%v[0-9]+]], %v24, 0
; CHECK: vgbm    [[T2:%v[0-9]+]], 0
; CHECK: vsumb   [[T3:%v[0-9]+]], [[T1]], [[T2]]
; CHECK: vsumgf  %v24, [[T3]], [[T2]]
; CHECK: br      %r14

  %popcnt = call <2 x i64> @llvm.ctpop.v2i64(<2 x i64> %a)
  ret <2 x i64> %popcnt
}