llvm/test/CodeGen/AArch64/sve-trunc.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve < %s | FileCheck %s
   3
   4 ; For all the functions below should the operation is a nop
   5 define <vscale x 8 x i8> @trunc_i16toi8(<vscale x 8 x i16> %in) {
   6 ; CHECK-LABEL: trunc_i16toi8:
   7 ; CHECK:       // %bb.0: // %entry
   8 ; CHECK-NEXT:    ret
   9 entry:
  10   %out = trunc <vscale x 8 x i16> %in to <vscale x 8 x i8>
  11   ret <vscale x 8 x i8> %out
  12 }
  13
  14 define <vscale x 4 x i8> @trunc_i32toi8(<vscale x 4 x i32> %in) {
  15 ; CHECK-LABEL: trunc_i32toi8:
  16 ; CHECK:       // %bb.0: // %entry
  17 ; CHECK-NEXT:    ret
  18 entry:
  19   %out = trunc <vscale x 4 x i32> %in to <vscale x 4 x i8>
  20   ret <vscale x 4 x i8> %out
  21 }
  22
  23 define <vscale x 2 x i8> @trunc_i64toi8(<vscale x 2 x i64> %in) {
  24 ; CHECK-LABEL: trunc_i64toi8:
  25 ; CHECK:       // %bb.0: // %entry
  26 ; CHECK-NEXT:    ret
  27 entry:
  28   %out = trunc <vscale x 2 x i64> %in to <vscale x 2 x i8>
  29   ret <vscale x 2 x i8> %out
  30 }
  31
  32 define <vscale x 4 x i16> @trunc_i32toi16(<vscale x 4 x i32> %in) {
  33 ; CHECK-LABEL: trunc_i32toi16:
  34 ; CHECK:       // %bb.0: // %entry
  35 ; CHECK-NEXT:    ret
  36 entry:
  37   %out = trunc <vscale x 4 x i32> %in to <vscale x 4 x i16>
  38   ret <vscale x 4 x i16> %out
  39 }
  40
  41 define <vscale x 2 x i16> @trunc_i64toi16(<vscale x 2 x i64> %in) {
  42 ; CHECK-LABEL: trunc_i64toi16:
  43 ; CHECK:       // %bb.0: // %entry
  44 ; CHECK-NEXT:    ret
  45 entry:
  46   %out = trunc <vscale x 2 x i64> %in to <vscale x 2 x i16>
  47   ret <vscale x 2 x i16> %out
  48 }
  49
  50 define <vscale x 2 x i32> @trunc_i64toi32(<vscale x 2 x i64> %in) {
  51 ; CHECK-LABEL: trunc_i64toi32:
  52 ; CHECK:       // %bb.0: // %entry
  53 ; CHECK-NEXT:    ret
  54 entry:
  55   %out = trunc <vscale x 2 x i64> %in to <vscale x 2 x i32>
  56   ret <vscale x 2 x i32> %out
  57 }
  58
  59 ; Truncating to i1 requires convert it to a cmp
  60
  61 define <vscale x 2 x i1> @trunc_i64toi1(<vscale x 2 x i64> %in) {
  62 ; CHECK-LABEL: trunc_i64toi1:
  63 ; CHECK:       // %bb.0: // %entry
  64 ; CHECK-NEXT:    ptrue p0.d
  65 ; CHECK-NEXT:    and z0.d, z0.d, #0x1
  66 ; CHECK-NEXT:    cmpne p0.d, p0/z, z0.d, #0
  67 ; CHECK-NEXT:    ret
  68 entry:
  69   %out = trunc <vscale x 2 x i64> %in to <vscale x 2 x i1>
  70   ret <vscale x 2 x i1> %out
  71 }
  72
  73 define <vscale x 4 x i1> @trunc_i64toi1_split(<vscale x 4 x i64> %in) {
  74 ; CHECK-LABEL: trunc_i64toi1_split:
  75 ; CHECK:       // %bb.0: // %entry
  76 ; CHECK-NEXT:    ptrue p0.d
  77 ; CHECK-NEXT:    and z1.d, z1.d, #0x1
  78 ; CHECK-NEXT:    and z0.d, z0.d, #0x1
  79 ; CHECK-NEXT:    cmpne p1.d, p0/z, z1.d, #0
  80 ; CHECK-NEXT:    cmpne p0.d, p0/z, z0.d, #0
  81 ; CHECK-NEXT:    uzp1 p0.s, p0.s, p1.s
  82 ; CHECK-NEXT:    ret
  83 entry:
  84   %out = trunc <vscale x 4 x i64> %in to <vscale x 4 x i1>
  85   ret <vscale x 4 x i1> %out
  86 }
  87
  88 define <vscale x 8 x i1> @trunc_i64toi1_split2(<vscale x 8 x i64> %in) {
  89 ; CHECK-LABEL: trunc_i64toi1_split2:
  90 ; CHECK:       // %bb.0: // %entry
  91 ; CHECK-NEXT:    ptrue p0.d
  92 ; CHECK-NEXT:    and z3.d, z3.d, #0x1
  93 ; CHECK-NEXT:    and z2.d, z2.d, #0x1
  94 ; CHECK-NEXT:    and z1.d, z1.d, #0x1
  95 ; CHECK-NEXT:    and z0.d, z0.d, #0x1
  96 ; CHECK-NEXT:    cmpne p1.d, p0/z, z3.d, #0
  97 ; CHECK-NEXT:    cmpne p2.d, p0/z, z2.d, #0
  98 ; CHECK-NEXT:    cmpne p3.d, p0/z, z1.d, #0
  99 ; CHECK-NEXT:    cmpne p0.d, p0/z, z0.d, #0
 100 ; CHECK-NEXT:    uzp1 p1.s, p2.s, p1.s
 101 ; CHECK-NEXT:    uzp1 p0.s, p0.s, p3.s
 102 ; CHECK-NEXT:    uzp1 p0.h, p0.h, p1.h
 103 ; CHECK-NEXT:    ret
 104 entry:
 105   %out = trunc <vscale x 8 x i64> %in to <vscale x 8 x i1>
 106   ret <vscale x 8 x i1> %out
 107 }
 108
 109 define <vscale x 16 x i1> @trunc_i64toi1_split3(<vscale x 16 x i64> %in) {
 110 ; CHECK-LABEL: trunc_i64toi1_split3:
 111 ; CHECK:       // %bb.0: // %entry
 112 ; CHECK-NEXT:    str x29, [sp, #-16]! // 8-byte Folded Spill
 113 ; CHECK-NEXT:    addvl sp, sp, #-1
 114 ; CHECK-NEXT:    str p6, [sp, #5, mul vl] // 2-byte Folded Spill
 115 ; CHECK-NEXT:    str p5, [sp, #6, mul vl] // 2-byte Folded Spill
 116 ; CHECK-NEXT:    str p4, [sp, #7, mul vl] // 2-byte Folded Spill
 117 ; CHECK-NEXT:    .cfi_escape 0x0f, 0x0c, 0x8f, 0x00, 0x11, 0x10, 0x22, 0x11, 0x08, 0x92, 0x2e, 0x00, 0x1e, 0x22 // sp + 16 + 8 * VG
 118 ; CHECK-NEXT:    .cfi_offset w29, -16
 119 ; CHECK-NEXT:    ptrue p0.d
 120 ; CHECK-NEXT:    and z7.d, z7.d, #0x1
 121 ; CHECK-NEXT:    and z6.d, z6.d, #0x1
 122 ; CHECK-NEXT:    and z5.d, z5.d, #0x1
 123 ; CHECK-NEXT:    and z4.d, z4.d, #0x1
 124 ; CHECK-NEXT:    and z3.d, z3.d, #0x1
 125 ; CHECK-NEXT:    and z2.d, z2.d, #0x1
 126 ; CHECK-NEXT:    and z1.d, z1.d, #0x1
 127 ; CHECK-NEXT:    and z0.d, z0.d, #0x1
 128 ; CHECK-NEXT:    cmpne p1.d, p0/z, z7.d, #0
 129 ; CHECK-NEXT:    cmpne p2.d, p0/z, z6.d, #0
 130 ; CHECK-NEXT:    cmpne p3.d, p0/z, z5.d, #0
 131 ; CHECK-NEXT:    cmpne p4.d, p0/z, z4.d, #0
 132 ; CHECK-NEXT:    cmpne p5.d, p0/z, z3.d, #0
 133 ; CHECK-NEXT:    cmpne p6.d, p0/z, z2.d, #0
 134 ; CHECK-NEXT:    uzp1 p1.s, p2.s, p1.s
 135 ; CHECK-NEXT:    cmpne p2.d, p0/z, z1.d, #0
 136 ; CHECK-NEXT:    cmpne p0.d, p0/z, z0.d, #0
 137 ; CHECK-NEXT:    uzp1 p3.s, p4.s, p3.s
 138 ; CHECK-NEXT:    uzp1 p4.s, p6.s, p5.s
 139 ; CHECK-NEXT:    ldr p6, [sp, #5, mul vl] // 2-byte Folded Reload
 140 ; CHECK-NEXT:    uzp1 p0.s, p0.s, p2.s
 141 ; CHECK-NEXT:    ldr p5, [sp, #6, mul vl] // 2-byte Folded Reload
 142 ; CHECK-NEXT:    uzp1 p1.h, p3.h, p1.h
 143 ; CHECK-NEXT:    uzp1 p0.h, p0.h, p4.h
 144 ; CHECK-NEXT:    ldr p4, [sp, #7, mul vl] // 2-byte Folded Reload
 145 ; CHECK-NEXT:    uzp1 p0.b, p0.b, p1.b
 146 ; CHECK-NEXT:    addvl sp, sp, #1
 147 ; CHECK-NEXT:    ldr x29, [sp], #16 // 8-byte Folded Reload
 148 ; CHECK-NEXT:    ret
 149 entry:
 150   %out = trunc <vscale x 16 x i64> %in to <vscale x 16 x i1>
 151   ret <vscale x 16 x i1> %out
 152 }
 153
 154
 155 define <vscale x 4 x i1> @trunc_i32toi1(<vscale x 4 x i32> %in) {
 156 ; CHECK-LABEL: trunc_i32toi1:
 157 ; CHECK:       // %bb.0: // %entry
 158 ; CHECK-NEXT:    ptrue p0.s
 159 ; CHECK-NEXT:    and z0.s, z0.s, #0x1
 160 ; CHECK-NEXT:    cmpne p0.s, p0/z, z0.s, #0
 161 ; CHECK-NEXT:    ret
 162 entry:
 163   %out = trunc <vscale x 4 x i32> %in to <vscale x 4 x i1>
 164   ret <vscale x 4 x i1> %out
 165 }
 166
 167 define <vscale x 8 x i1> @trunc_i16toi1(<vscale x 8 x i16> %in) {
 168 ; CHECK-LABEL: trunc_i16toi1:
 169 ; CHECK:       // %bb.0: // %entry
 170 ; CHECK-NEXT:    ptrue p0.h
 171 ; CHECK-NEXT:    and z0.h, z0.h, #0x1
 172 ; CHECK-NEXT:    cmpne p0.h, p0/z, z0.h, #0
 173 ; CHECK-NEXT:    ret
 174 entry:
 175   %out = trunc <vscale x 8 x i16> %in to <vscale x 8 x i1>
 176   ret <vscale x 8 x i1> %out
 177 }
 178
 179 define <vscale x 16 x i1> @trunc_i8toi1(<vscale x 16 x i8> %in) {
 180 ; CHECK-LABEL: trunc_i8toi1:
 181 ; CHECK:       // %bb.0: // %entry
 182 ; CHECK-NEXT:    ptrue p0.b
 183 ; CHECK-NEXT:    and z0.b, z0.b, #0x1
 184 ; CHECK-NEXT:    cmpne p0.b, p0/z, z0.b, #0
 185 ; CHECK-NEXT:    ret
 186 entry:
 187   %out = trunc <vscale x 16 x i8> %in to <vscale x 16 x i1>
 188   ret <vscale x 16 x i1> %out
 189 }
 190
 191 define <vscale x 1 x i1> @trunc_nxv1i32_to_nxv1i1(<vscale x 1 x i32> %in) {
 192 ; CHECK-LABEL: trunc_nxv1i32_to_nxv1i1:
 193 ; CHECK:       // %bb.0:
 194 ; CHECK-NEXT:    ptrue p0.s
 195 ; CHECK-NEXT:    and z0.s, z0.s, #0x1
 196 ; CHECK-NEXT:    cmpne p0.s, p0/z, z0.s, #0
 197 ; CHECK-NEXT:    punpklo p0.h, p0.b
 198 ; CHECK-NEXT:    punpklo p0.h, p0.b
 199 ; CHECK-NEXT:    ret
 200   %out = trunc <vscale x 1 x i32> %in to <vscale x 1 x i1>
 201   ret <vscale x 1 x i1> %out
 202 }
 203
 204 define void @trunc_promoteIntRes(<vscale x 4 x i64> %0, ptr %ptr) {
 205 ; CHECK-LABEL: trunc_promoteIntRes:
 206 ; CHECK:       // %bb.0: // %entry
 207 ; CHECK-NEXT:    ptrue p0.s
 208 ; CHECK-NEXT:    uzp1 z0.s, z0.s, z1.s
 209 ; CHECK-NEXT:    st1h { z0.s }, p0, [x0]
 210 ; CHECK-NEXT:    ret
 211 entry:
 212   %1 = trunc <vscale x 4 x i64> %0 to <vscale x 4 x i16>
 213   store <vscale x 4 x i16> %1, <vscale x 4 x i16>* %ptr, align 2
 214   ret void
 215 }