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

   1 ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve < %s | FileCheck %s
   2
   3 ;
   4 ; ST1B, ST1W, ST1H, ST1D: vector base + scalar offset
   5 ;   e.g. st1h { z0.s }, p0, [x0, z1.d]
   6 ;
   7
   8 ; ST1B
   9 define void @sst1b_s_scalar_offset(<vscale x 4 x i32> %data, <vscale x 4 x i1> %pg, <vscale x 4 x i32> %base, i64 %offset) {
  10 ; CHECK-LABEL: sst1b_s_scalar_offset:
  11 ; CHECK: st1b { z0.s }, p0, [x0, z1.s, uxtw]
  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.st1.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 @sst1b_d_scalar_offset(<vscale x 2 x i64> %data, <vscale x 2 x i1> %pg, <vscale x 2 x i64> %base, i64 %offset) {
  22 ; CHECK-LABEL: sst1b_d_scalar_offset:
  23 ; CHECK: st1b { z0.d }, p0, [x0, z1.d]
  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.st1.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 ; ST1H
  34 define void @sst1h_s_scalar_offset(<vscale x 4 x i32> %data, <vscale x 4 x i1> %pg, <vscale x 4 x i32> %base, i64 %offset) {
  35 ; CHECK-LABEL: sst1h_s_scalar_offset:
  36 ; CHECK: st1h { z0.s }, p0, [x0, z1.s, uxtw]
  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.st1.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 @sst1h_d_scalar_offset(<vscale x 2 x i64> %data, <vscale x 2 x i1> %pg, <vscale x 2 x i64> %base, i64 %offset) {
  47 ; CHECK-LABEL: sst1h_d_scalar_offset:
  48 ; CHECK: st1h { z0.d }, p0, [x0, z1.d]
  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.st1.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 ; ST1W
  59 define void @sst1w_s_scalar_offset(<vscale x 4 x i32> %data, <vscale x 4 x i1> %pg, <vscale x 4 x i32> %base, i64 %offset) {
  60 ; CHECK-LABEL: sst1w_s_scalar_offset:
  61 ; CHECK: st1w { z0.s }, p0, [x0, z1.s, uxtw]
  62 ; CHECK-NEXT: ret
  63   call void @llvm.aarch64.sve.st1.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 @sst1w_d_scalar_offset(<vscale x 2 x i64> %data, <vscale x 2 x i1> %pg, <vscale x 2 x i64> %base, i64 %offset) {
  71 ; CHECK-LABEL: sst1w_d_scalar_offset:
  72 ; CHECK: st1w { z0.d }, p0, [x0, z1.d]
  73 ; CHECK-NEXT: ret
  74   %data_trunc = trunc <vscale x 2 x i64> %data to <vscale x 2 x i32>
  75   call void @llvm.aarch64.sve.st1.scatter.scalar.offset.nxv2i32.nxv2i64(<vscale x 2 x i32> %data_trunc,
  76                                                                         <vscale x 2 x i1> %pg,
  77                                                                         <vscale x 2 x i64> %base,
  78                                                                         i64 %offset)
  79   ret void
  80 }
  81
  82 define void @sst1w_s_scalar_offset_float(<vscale x 4 x float> %data, <vscale x 4 x i1> %pg, <vscale x 4 x i32> %base, i64 %offset) {
  83 ; CHECK-LABEL: sst1w_s_scalar_offset_float:
  84 ; CHECK: st1w { z0.s }, p0, [x0, z1.s, uxtw]
  85 ; CHECK-NEXT: ret
  86   call void @llvm.aarch64.sve.st1.scatter.scalar.offset.nxv4f32.nxv4i32(<vscale x 4 x float> %data,
  87                                                                         <vscale x 4 x i1> %pg,
  88                                                                         <vscale x 4 x i32> %base,
  89                                                                         i64 %offset)
  90   ret void
  91 }
  92
  93 ; ST1D
  94 define void @sst1d_d_scalar_offset(<vscale x 2 x i64> %data, <vscale x 2 x i1> %pg, <vscale x 2 x i64> %base, i64 %offset) {
  95 ; CHECK-LABEL: sst1d_d_scalar_offset:
  96 ; CHECK: st1d { z0.d }, p0, [x0, z1.d]
  97 ; CHECK-NEXT: ret
  98   call void @llvm.aarch64.sve.st1.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 @sst1d_d_scalar_offset_double(<vscale x 2 x double> %data, <vscale x 2 x i1> %pg, <vscale x 2 x i64> %base, i64 %offset) {
 106 ; CHECK-LABEL: sst1d_d_scalar_offset_double:
 107 ; CHECK: st1d { z0.d }, p0, [x0, z1.d]
 108 ; CHECK-NEXT: ret
 109   call void @llvm.aarch64.sve.st1.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 ; ST1B
 117 declare void @llvm.aarch64.sve.st1.scatter.scalar.offset.nxv4i8.nxv4i32(<vscale x 4 x i8>, <vscale x 4 x i1>, <vscale x 4 x i32>, i64)
 118 declare void @llvm.aarch64.sve.st1.scatter.scalar.offset.nxv2i8.nxv2i64(<vscale x 2 x i8>, <vscale x 2 x i1>, <vscale x 2 x i64>, i64)
 119
 120 ; ST1H
 121 declare void @llvm.aarch64.sve.st1.scatter.scalar.offset.nxv4i16.nxv4i32(<vscale x 4 x i16>, <vscale x 4 x i1>, <vscale x 4 x i32>, i64)
 122 declare void @llvm.aarch64.sve.st1.scatter.scalar.offset.nxv2i16.nxv2i64(<vscale x 2 x i16>, <vscale x 2 x i1>, <vscale x 2 x i64>, i64)
 123
 124 ; ST1W
 125 declare void @llvm.aarch64.sve.st1.scatter.scalar.offset.nxv4i32.nxv4i32(<vscale x 4 x i32>, <vscale x 4 x i1>, <vscale x 4 x i32>, i64)
 126 declare void @llvm.aarch64.sve.st1.scatter.scalar.offset.nxv2i32.nxv2i64(<vscale x 2 x i32>, <vscale x 2 x i1>, <vscale x 2 x i64>, i64)
 127
 128 declare void @llvm.aarch64.sve.st1.scatter.scalar.offset.nxv4f32.nxv4i32(<vscale x 4 x float>, <vscale x 4 x i1>, <vscale x 4 x i32>, i64)
 129
 130 ; ST1D
 131 declare void @llvm.aarch64.sve.st1.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.st1.scatter.scalar.offset.nxv2f64.nxv2i64(<vscale x 2 x double>, <vscale x 2 x i1>, <vscale x 2 x i64>, i64)