llvm/test/CodeGen/AArch64/sve2-intrinsics-nt-scatter-stores-vector-base-scalar-offset.ll

   1 ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve2 < %s | FileCheck %s
   2
   3 ;
   4 ; STNT1B, STNT1W, STNT1H, STNT1D: vector base + scalar offset
   5 ;   stnt1b { z0.s }, p0/z, [z0.s, x0]
   6 ;
   7
   8 ; STNT1B
   9 define void @stnt1b_s(<vscale x 4 x i32> %data, <vscale x 4 x i1> %pg, <vscale x 4 x i32> %base, i64 %offset) {
  10 ; CHECK-LABEL: stnt1b_s:
  11 ; CHECK: stnt1b { z0.s }, p0, [z1.s, x0]
  12 ; CHECK-NEXT: ret
  13   %data_trunc = trunc <vscale x 4 x i32> %data to <vscale x 4 x i8>
  14   call void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv4i8.nxv4i32(<vscale x 4 x i8> %data_trunc,
  15                                                                          <vscale x 4 x i1> %pg,
  16                                                                          <vscale x 4 x i32> %base,
  17                                                                          i64 %offset)
  18   ret void
  19 }
  20
  21 define void @stnt1b_d(<vscale x 2 x i64> %data, <vscale x 2 x i1> %pg, <vscale x 2 x i64> %base, i64 %offset) {
  22 ; CHECK-LABEL: stnt1b_d:
  23 ; CHECK: stnt1b { z0.d }, p0, [z1.d, x0]
  24 ; CHECK-NEXT: ret
  25   %data_trunc = trunc <vscale x 2 x i64> %data to <vscale x 2 x i8>
  26   call void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv2i8.nxv2i64(<vscale x 2 x i8> %data_trunc,
  27                                                                          <vscale x 2 x i1> %pg,
  28                                                                          <vscale x 2 x i64> %base,
  29                                                                          i64 %offset)
  30   ret void
  31 }
  32
  33 ; STNT1H
  34 define void @stnt1h_s(<vscale x 4 x i32> %data, <vscale x 4 x i1> %pg, <vscale x 4 x i32> %base, i64 %offset) {
  35 ; CHECK-LABEL: stnt1h_s:
  36 ; CHECK: stnt1h { z0.s }, p0, [z1.s, x0]
  37 ; CHECK-NEXT: ret
  38   %data_trunc = trunc <vscale x 4 x i32> %data to <vscale x 4 x i16>
  39   call void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv4i16.nxv4i32(<vscale x 4 x i16> %data_trunc,
  40                                                                           <vscale x 4 x i1> %pg,
  41                                                                           <vscale x 4 x i32> %base,
  42                                                                           i64 %offset)
  43   ret void
  44 }
  45
  46 define void @stnt1h_d(<vscale x 2 x i64> %data, <vscale x 2 x i1> %pg, <vscale x 2 x i64> %base, i64 %offset) {
  47 ; CHECK-LABEL: stnt1h_d:
  48 ; CHECK: stnt1h { z0.d }, p0, [z1.d, x0]
  49 ; CHECK-NEXT: ret
  50   %data_trunc = trunc <vscale x 2 x i64> %data to <vscale x 2 x i16>
  51   call void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv2i16.nxv2i64(<vscale x 2 x i16> %data_trunc,
  52                                                                           <vscale x 2 x i1> %pg,
  53                                                                           <vscale x 2 x i64> %base,
  54                                                                           i64 %offset)
  55   ret void
  56 }
  57
  58 ; STNT1W
  59 define void @stnt1w_s(<vscale x 4 x i32> %data, <vscale x 4 x i1> %pg, <vscale x 4 x i32> %base, i64 %offset) {
  60 ; CHECK-LABEL: stnt1w_s:
  61 ; CHECK: stnt1w { z0.s }, p0, [z1.s, x0]
  62 ; CHECK-NEXT: ret
  63   call void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv4i32.nxv4i32(<vscale x 4 x i32> %data,
  64                                                                           <vscale x 4 x i1> %pg,
  65                                                                           <vscale x 4 x i32> %base,
  66                                                                           i64 %offset)
  67   ret void
  68 }
  69
  70 define void @stnt1w_f32_s(<vscale x 4 x float> %data, <vscale x 4 x i1> %pg, <vscale x 4 x i32> %base, i64 %offset) {
  71 ; CHECK-LABEL: stnt1w_f32_s:
  72 ; CHECK: stnt1w { z0.s }, p0, [z1.s, x0]
  73 ; CHECK-NEXT: ret
  74   call void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv4f32.nxv4i32(<vscale x 4 x float> %data,
  75                                                                           <vscale x 4 x i1> %pg,
  76                                                                           <vscale x 4 x i32> %base,
  77                                                                           i64 %offset)
  78   ret void
  79 }
  80
  81 define void @stnt1w_d(<vscale x 2 x i64> %data, <vscale x 2 x i1> %pg, <vscale x 2 x i64> %base, i64 %offset) {
  82 ; CHECK-LABEL: stnt1w_d:
  83 ; CHECK: stnt1w { z0.d }, p0, [z1.d, x0]
  84 ; CHECK-NEXT: ret
  85   %data_trunc = trunc <vscale x 2 x i64> %data to <vscale x 2 x i32>
  86   call void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv2i32.nxv2i64(<vscale x 2 x i32> %data_trunc,
  87                                                                           <vscale x 2 x i1> %pg,
  88                                                                           <vscale x 2 x i64> %base,
  89                                                                           i64 %offset)
  90   ret void
  91 }
  92
  93 ; STNT1D
  94 define void @stnt1d_d(<vscale x 2 x i64> %data, <vscale x 2 x i1> %pg, <vscale x 2 x i64> %base, i64 %offset) {
  95 ; CHECK-LABEL: stnt1d_d:
  96 ; CHECK: stnt1d { z0.d }, p0, [z1.d, x0]
  97 ; CHECK-NEXT: ret
  98   call void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv2i64.nxv2i64(<vscale x 2 x i64> %data,
  99                                                                           <vscale x 2 x i1> %pg,
 100                                                                           <vscale x 2 x i64> %base,
 101                                                                           i64 %offset)
 102   ret void
 103 }
 104
 105 define void @stnt1d_f64_d(<vscale x 2 x double> %data, <vscale x 2 x i1> %pg, <vscale x 2 x i64> %base, i64 %offset) {
 106 ; CHECK-LABEL: stnt1d_f64_d:
 107 ; CHECK: stnt1d { z0.d }, p0, [z1.d, x0]
 108 ; CHECK-NEXT: ret
 109   call void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv2f64.nxv2i64(<vscale x 2 x double> %data,
 110                                                                           <vscale x 2 x i1> %pg,
 111                                                                           <vscale x 2 x i64> %base,
 112                                                                           i64 %offset)
 113   ret void
 114 }
 115
 116 ; STNT1B
 117 declare void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv2i8.nxv2i64(<vscale x 2 x i8>, <vscale x 2 x i1>, <vscale x 2 x i64>, i64)
 118 declare void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv4i8.nxv4i32(<vscale x 4 x i8>, <vscale x 4 x i1>, <vscale x 4 x i32>, i64)
 119
 120 ; STNT1H
 121 declare void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv2i16.nxv2i64(<vscale x 2 x i16>, <vscale x 2 x i1>, <vscale x 2 x i64>, i64)
 122 declare void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv4i16.nxv4i32(<vscale x 4 x i16>, <vscale x 4 x i1>, <vscale x 4 x i32>, i64)
 123
 124 ; STNT1W
 125 declare void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv2i32.nxv2i64(<vscale x 2 x i32>, <vscale x 2 x i1>, <vscale x 2 x i64>, i64)
 126 declare void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv4i32.nxv4i32(<vscale x 4 x i32>, <vscale x 4 x i1>, <vscale x 4 x i32>, i64)
 127
 128 declare void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv4f32.nxv4i32(<vscale x 4 x float>, <vscale x 4 x i1>, <vscale x 4 x i32>, i64)
 129
 130 ; STNT1D
 131 declare void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv2i64.nxv2i64(<vscale x 2 x i64>, <vscale x 2 x i1>, <vscale x 2 x i64>, i64)
 132
 133 declare void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv2f32.nxv2i64(<vscale x 2 x float>, <vscale x 2 x i1>, <vscale x 2 x i64>, i64)
 134 declare void @llvm.aarch64.sve.stnt1.scatter.scalar.offset.nxv2f64.nxv2i64(<vscale x 2 x double>, <vscale x 2 x i1>, <vscale x 2 x i64>, i64)