llvm/test/CodeGen/AArch64/sve2-intrinsics-nt-scatter-stores-32bit-unscaled-offset.ll

   1 ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve2 < %s | FileCheck %s
   2
   3 ;
   4 ; STNT1B, STNT1W, STNT1H, STNT1D: base + 32-bit unscaled offset, zero (uxtw)
   5 ; extended to 64 bits.
   6 ;   e.g. stnt1h { z0.d }, p0, [z1.d, x0]
   7 ;
   8
   9 ; STNT1B
  10 define void @sstnt1b_s_uxtw(<vscale x 4 x i32> %data, <vscale x 4 x i1> %pg, i8* %base, <vscale x 4 x i32> %offsets) {
  11 ; CHECK-LABEL: sstnt1b_s_uxtw:
  12 ; CHECK: stnt1b { z0.s }, p0, [z1.s, x0]
  13 ; CHECK-NEXT: ret
  14   %data_trunc = trunc <vscale x 4 x i32> %data to <vscale x 4 x i8>
  15   call void  @llvm.aarch64.sve.stnt1.scatter.uxtw.nxv4i8(<vscale x 4 x i8> %data_trunc,
  16                                                          <vscale x 4 x i1> %pg,
  17                                                          i8* %base,
  18                                                          <vscale x 4 x i32> %offsets)
  19   ret void
  20 }
  21
  22 ; STNT1H
  23 define void @sstnt1h_s_uxtw(<vscale x 4 x i32> %data, <vscale x 4 x i1> %pg, i16* %base, <vscale x 4 x i32> %offsets) {
  24 ; CHECK-LABEL: sstnt1h_s_uxtw:
  25 ; CHECK: stnt1h { z0.s }, p0, [z1.s, x0]
  26 ; CHECK-NEXT:   ret
  27   %data_trunc = trunc <vscale x 4 x i32> %data to <vscale x 4 x i16>
  28   call void @llvm.aarch64.sve.stnt1.scatter.uxtw.nxv4i16(<vscale x 4 x i16> %data_trunc,
  29                                                          <vscale x 4 x i1> %pg,
  30                                                          i16* %base,
  31                                                          <vscale x 4 x i32> %offsets)
  32   ret void
  33 }
  34
  35 ; STNT1W
  36 define void @sstnt1w_s_uxtw(<vscale x 4 x i32> %data, <vscale x 4 x i1> %pg, i32* %base, <vscale x 4 x i32> %offsets) {
  37 ; CHECK-LABEL: sstnt1w_s_uxtw:
  38 ; CHECK: stnt1w { z0.s }, p0, [z1.s, x0]
  39 ; CHECK-NEXT:   ret
  40   call void @llvm.aarch64.sve.stnt1.scatter.uxtw.nxv4i32(<vscale x 4 x i32> %data,
  41                                                          <vscale x 4 x i1> %pg,
  42                                                          i32* %base,
  43                                                          <vscale x 4 x i32> %offsets)
  44   ret void
  45 }
  46
  47 define void @sstnt1w_s_uxtw_float(<vscale x 4 x float> %data, <vscale x 4 x i1> %pg, float* %base, <vscale x 4 x i32> %offsets) {
  48 ; CHECK-LABEL: sstnt1w_s_uxtw_float:
  49 ; CHECK: stnt1w { z0.s }, p0, [z1.s, x0]
  50 ; CHECK-NEXT:   ret
  51   call void @llvm.aarch64.sve.stnt1.scatter.uxtw.nxv4f32(<vscale x 4 x float> %data,
  52                                                          <vscale x 4 x i1> %pg,
  53                                                          float* %base,
  54                                                          <vscale x 4 x i32> %offsets)
  55   ret void
  56 }
  57
  58 ; STNT1B
  59 declare void @llvm.aarch64.sve.stnt1.scatter.uxtw.nxv4i8(<vscale x 4 x i8>, <vscale x 4 x i1>, i8*, <vscale x 4 x i32>)
  60 declare void @llvm.aarch64.sve.stnt1.scatter.uxtw.nxv2i8(<vscale x 2 x i8>, <vscale x 2 x i1>, i8*, <vscale x 2 x i32>)
  61 declare void @llvm.aarch64.sve.stnt1.scatter.sxtw.nxv4i8(<vscale x 4 x i8>, <vscale x 4 x i1>, i8*, <vscale x 4 x i32>)
  62 declare void @llvm.aarch64.sve.stnt1.scatter.sxtw.nxv2i8(<vscale x 2 x i8>, <vscale x 2 x i1>, i8*, <vscale x 2 x i32>)
  63
  64 ; STNT1H
  65 declare void @llvm.aarch64.sve.stnt1.scatter.sxtw.nxv4i16(<vscale x 4 x i16>, <vscale x 4 x i1>, i16*, <vscale x 4 x i32>)
  66 declare void @llvm.aarch64.sve.stnt1.scatter.sxtw.nxv2i16(<vscale x 2 x i16>, <vscale x 2 x i1>, i16*, <vscale x 2 x i32>)
  67 declare void @llvm.aarch64.sve.stnt1.scatter.uxtw.nxv4i16(<vscale x 4 x i16>, <vscale x 4 x i1>, i16*, <vscale x 4 x i32>)
  68 declare void @llvm.aarch64.sve.stnt1.scatter.uxtw.nxv2i16(<vscale x 2 x i16>, <vscale x 2 x i1>, i16*, <vscale x 2 x i32>)
  69
  70 ; STNT1W
  71 declare void @llvm.aarch64.sve.stnt1.scatter.sxtw.nxv4i32(<vscale x 4 x i32>, <vscale x 4 x i1>, i32*, <vscale x 4 x i32>)
  72 declare void @llvm.aarch64.sve.stnt1.scatter.sxtw.nxv2i32(<vscale x 2 x i32>, <vscale x 2 x i1>, i32*, <vscale x 2 x i32>)
  73 declare void @llvm.aarch64.sve.stnt1.scatter.uxtw.nxv4i32(<vscale x 4 x i32>, <vscale x 4 x i1>, i32*, <vscale x 4 x i32>)
  74 declare void @llvm.aarch64.sve.stnt1.scatter.uxtw.nxv2i32(<vscale x 2 x i32>, <vscale x 2 x i1>, i32*, <vscale x 2 x i32>)
  75
  76 declare void @llvm.aarch64.sve.stnt1.scatter.sxtw.nxv4f32(<vscale x 4 x float>, <vscale x 4 x i1>, float*, <vscale x 4 x i32>)
  77 declare void @llvm.aarch64.sve.stnt1.scatter.uxtw.nxv4f32(<vscale x 4 x float>, <vscale x 4 x i1>, float*, <vscale x 4 x i32>)