llvm/test/CodeGen/AArch64/sve-ptest-removal-cmplt.ll

   1 ; RUN: llc -mtriple=aarch64--linux-gnu -mattr=+sve %s -o - | FileCheck %s
   2
   3 ;
   4 ; Immediate Compares
   5 ;
   6
   7 define i32 @cmplt_imm_nxv16i8(<vscale x 16 x i1> %pg, <vscale x 16 x i8> %a) {
   8 ; CHECK-LABEL: cmplt_imm_nxv16i8:
   9 ; CHECK: cmplt p0.b, p0/z, z0.b, #0
  10 ; CHECK-NEXT: cset w0, ne
  11 ; CHECK-NEXT: ret
  12   %1 = tail call <vscale x 16 x i1> @llvm.aarch64.sve.cmpgt.nxv16i8(<vscale x 16 x i1> %pg, <vscale x 16 x i8> zeroinitializer, <vscale x 16 x i8> %a)
  13   %2 = tail call <vscale x 16 x i1> @llvm.aarch64.sve.ptrue.nxv16i1(i32 31)
  14   %3 = tail call i1 @llvm.aarch64.sve.ptest.any(<vscale x 16 x i1> %2, <vscale x 16 x i1> %1)
  15   %conv = zext i1 %3 to i32
  16   ret i32 %conv
  17 }
  18
  19 ;
  20 ; Wide Compares
  21 ;
  22
  23 define i32 @cmplt_wide_nxv16i8(<vscale x 16 x i1> %pg, <vscale x 16 x i8> %a, <vscale x 2 x i64> %b) {
  24 ; CHECK-LABEL: cmplt_wide_nxv16i8:
  25 ; CHECK: cmplt p0.b, p0/z, z0.b, z1.d
  26 ; CHECK-NEXT: cset w0, ne
  27 ; CHECK-NEXT: ret
  28   %1 = tail call <vscale x 16 x i1> @llvm.aarch64.sve.cmplt.wide.nxv16i8(<vscale x 16 x i1> %pg, <vscale x 16 x i8> %a, <vscale x 2 x i64> %b)
  29   %2 = tail call i1 @llvm.aarch64.sve.ptest.any(<vscale x 16 x i1> %pg, <vscale x 16 x i1> %1)
  30   %conv = zext i1 %2 to i32
  31   ret i32 %conv
  32 }
  33
  34 declare <vscale x 16 x i1> @llvm.aarch64.sve.cmpgt.nxv16i8(<vscale x 16 x i1>, <vscale x 16 x i8>, <vscale x 16 x i8>)
  35 declare <vscale x 16 x i1> @llvm.aarch64.sve.cmplt.wide.nxv16i8(<vscale x 16 x i1>, <vscale x 16 x i8>, <vscale x 2 x i64>)
  36
  37 declare i1 @llvm.aarch64.sve.ptest.any(<vscale x 16 x i1>, <vscale x 16 x i1>)
  38
  39 declare <vscale x 16 x i1> @llvm.aarch64.sve.ptrue.nxv16i1(i32)