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

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