xref: /aosp_15_r20/external/llvm/test/CodeGen/SystemZ/vec-ctlz-01.ll (revision 9880d6810fe72a1726cb53787c6711e909410d58)

; Test vector count leading zeros
;
; RUN: llc < %s -mtriple=s390x-linux-gnu -mcpu=z13 | FileCheck %s

declare <16 x i8> @llvm.ctlz.v16i8(<16 x i8> %src, i1 %is_zero_undef)
declare <8 x i16> @llvm.ctlz.v8i16(<8 x i16> %src, i1 %is_zero_undef)
declare <4 x i32> @llvm.ctlz.v4i32(<4 x i32> %src, i1 %is_zero_undef)
declare <2 x i64> @llvm.ctlz.v2i64(<2 x i64> %src, i1 %is_zero_undef)

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

  %res = call <16 x i8> @llvm.ctlz.v16i8(<16 x i8> %a, i1 false)
  ret <16 x i8> %res
}

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

  %res = call <16 x i8> @llvm.ctlz.v16i8(<16 x i8> %a, i1 true)
  ret <16 x i8> %res
}

define <8 x i16> @f3(<8 x i16> %a) {
; CHECK-LABEL: f3:
; CHECK: vclzh %v24, %v24
; CHECK: br    %r14

  %res = call <8 x i16> @llvm.ctlz.v8i16(<8 x i16> %a, i1 false)
  ret <8 x i16> %res
}

define <8 x i16> @f4(<8 x i16> %a) {
; CHECK-LABEL: f4:
; CHECK: vclzh %v24, %v24
; CHECK: br    %r14

  %res = call <8 x i16> @llvm.ctlz.v8i16(<8 x i16> %a, i1 true)
  ret <8 x i16> %res
}

define <4 x i32> @f5(<4 x i32> %a) {
; CHECK-LABEL: f5:
; CHECK: vclzf %v24, %v24
; CHECK: br    %r14

  %res = call <4 x i32> @llvm.ctlz.v4i32(<4 x i32> %a, i1 false)
  ret <4 x i32> %res
}

define <4 x i32> @f6(<4 x i32> %a) {
; CHECK-LABEL: f6:
; CHECK: vclzf %v24, %v24
; CHECK: br    %r14

  %res = call <4 x i32> @llvm.ctlz.v4i32(<4 x i32> %a, i1 true)
  ret <4 x i32> %res
}

define <2 x i64> @f7(<2 x i64> %a) {
; CHECK-LABEL: f7:
; CHECK: vclzg %v24, %v24
; CHECK: br    %r14

  %res = call <2 x i64> @llvm.ctlz.v2i64(<2 x i64> %a, i1 false)
  ret <2 x i64> %res
}

define <2 x i64> @f8(<2 x i64> %a) {
; CHECK-LABEL: f8:
; CHECK: vclzg %v24, %v24
; CHECK: br    %r14

  %res = call <2 x i64> @llvm.ctlz.v2i64(<2 x i64> %a, i1 true)
  ret <2 x i64> %res
}