llvm/test/CodeGen/AArch64/sve-splat-sext.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc -O2 -mtriple=aarch64-linux-gnu -mattr=+sve < %s | FileCheck %s
   3
   4 define <vscale x 8 x i16> @sext_splat_v8i16_128() {
   5 ; CHECK-LABEL: sext_splat_v8i16_128:
   6 ; CHECK:       // %bb.0:
   7 ; CHECK-NEXT:    mov z0.h, #-128 // =0xffffffffffffff80
   8 ; CHECK-NEXT:    ret
   9   %i = insertelement <vscale x 8 x i16> poison, i16 128, i32 0
  10   %s = shufflevector <vscale x 8 x i16> %i, <vscale x 8 x i16> poison, <vscale x 8 x i32> zeroinitializer
  11   %a = shl <vscale x 8 x i16> %s, shufflevector (<vscale x 8 x i16> insertelement(<vscale x 8 x i16> undef, i16 8, i32 0), <vscale x 8 x i16> undef, <vscale x 8 x i32> zeroinitializer)
  12   %b = ashr <vscale x 8 x i16> %a, shufflevector (<vscale x 8 x i16> insertelement(<vscale x 8 x i16> undef, i16 8, i32 0), <vscale x 8 x i16> undef, <vscale x 8 x i32> zeroinitializer)
  13   ret <vscale x 8 x i16> %b
  14 }
  15
  16 define <vscale x 8 x i1> @sext_icmp_splat_v8i16_128(<vscale x 8 x i8> %d) {
  17 ; CHECK-LABEL: sext_icmp_splat_v8i16_128:
  18 ; CHECK:       // %bb.0:
  19 ; CHECK-NEXT:    ptrue p0.h
  20 ; CHECK-NEXT:    sxtb z0.h, p0/m, z0.h
  21 ; CHECK-NEXT:    cmpgt p0.h, p0/z, z0.h, #-1
  22 ; CHECK-NEXT:    ret
  23   %i = insertelement <vscale x 8 x i8> poison, i8 128, i32 0
  24   %s = shufflevector <vscale x 8 x i8> %i, <vscale x 8 x i8> poison, <vscale x 8 x i32> zeroinitializer
  25   %c = icmp ugt <vscale x 8 x i8> %s, %d
  26   ret <vscale x 8 x i1> %c
  27 }
  28
  29 define <vscale x 4 x i1> @sext_icmp_splat_v4i16_128(<vscale x 4 x i8> %d) {
  30 ; CHECK-LABEL: sext_icmp_splat_v4i16_128:
  31 ; CHECK:       // %bb.0:
  32 ; CHECK-NEXT:    ptrue p0.s
  33 ; CHECK-NEXT:    sxtb z0.s, p0/m, z0.s
  34 ; CHECK-NEXT:    cmpgt p0.s, p0/z, z0.s, #-1
  35 ; CHECK-NEXT:    ret
  36   %i = insertelement <vscale x 4 x i8> poison, i8 128, i32 0
  37   %s = shufflevector <vscale x 4 x i8> %i, <vscale x 4 x i8> poison, <vscale x 4 x i32> zeroinitializer
  38   %c = icmp ugt <vscale x 4 x i8> %s, %d
  39   ret <vscale x 4 x i1> %c
  40 }