588 lines
27 KiB
LLVM
588 lines
27 KiB
LLVM
; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve,+bf16 < %s 2>%t | FileCheck %s
|
|
; RUN: FileCheck --check-prefix=WARN --allow-empty %s <%t
|
|
|
|
; If this check fails please read test/CodeGen/AArch64/README for instructions on how to resolve it.
|
|
; WARN-NOT: warning
|
|
|
|
;
|
|
; ST2B
|
|
;
|
|
|
|
define void @st2b_i8(<vscale x 16 x i8> %v0, <vscale x 16 x i8> %v1, <vscale x 16 x i1> %pred, i8* %addr) {
|
|
; CHECK-LABEL: st2b_i8:
|
|
; CHECK: st2b { z0.b, z1.b }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st2.nxv16i8(<vscale x 16 x i8> %v0,
|
|
<vscale x 16 x i8> %v1,
|
|
<vscale x 16 x i1> %pred,
|
|
i8* %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; ST2H
|
|
;
|
|
|
|
define void @st2h_i16(<vscale x 8 x i16> %v0, <vscale x 8 x i16> %v1, <vscale x 8 x i1> %pred, i16* %addr) {
|
|
; CHECK-LABEL: st2h_i16:
|
|
; CHECK: st2h { z0.h, z1.h }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st2.nxv8i16(<vscale x 8 x i16> %v0,
|
|
<vscale x 8 x i16> %v1,
|
|
<vscale x 8 x i1> %pred,
|
|
i16* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st2h_f16(<vscale x 8 x half> %v0, <vscale x 8 x half> %v1, <vscale x 8 x i1> %pred, half* %addr) {
|
|
; CHECK-LABEL: st2h_f16:
|
|
; CHECK: st2h { z0.h, z1.h }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st2.nxv8f16(<vscale x 8 x half> %v0,
|
|
<vscale x 8 x half> %v1,
|
|
<vscale x 8 x i1> %pred,
|
|
half* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st2h_bf16(<vscale x 8 x bfloat> %v0, <vscale x 8 x bfloat> %v1, <vscale x 8 x i1> %pred, bfloat* %addr) #0 {
|
|
; CHECK-LABEL: st2h_bf16:
|
|
; CHECK: st2h { z0.h, z1.h }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st2.nxv8bf16(<vscale x 8 x bfloat> %v0,
|
|
<vscale x 8 x bfloat> %v1,
|
|
<vscale x 8 x i1> %pred,
|
|
bfloat* %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; ST2W
|
|
;
|
|
|
|
define void @st2w_i32(<vscale x 4 x i32> %v0, <vscale x 4 x i32> %v1, <vscale x 4 x i1> %pred, i32* %addr) {
|
|
; CHECK-LABEL: st2w_i32:
|
|
; CHECK: st2w { z0.s, z1.s }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st2.nxv4i32(<vscale x 4 x i32> %v0,
|
|
<vscale x 4 x i32> %v1,
|
|
<vscale x 4 x i1> %pred,
|
|
i32* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st2w_f32(<vscale x 4 x float> %v0, <vscale x 4 x float> %v1, <vscale x 4 x i1> %pred, float* %addr) {
|
|
; CHECK-LABEL: st2w_f32:
|
|
; CHECK: st2w { z0.s, z1.s }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st2.nxv4f32(<vscale x 4 x float> %v0,
|
|
<vscale x 4 x float> %v1,
|
|
<vscale x 4 x i1> %pred,
|
|
float* %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; ST2D
|
|
;
|
|
|
|
define void @st2d_i64(<vscale x 2 x i64> %v0, <vscale x 2 x i64> %v1, <vscale x 2 x i1> %pred, i64* %addr) {
|
|
; CHECK-LABEL: st2d_i64:
|
|
; CHECK: st2d { z0.d, z1.d }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st2.nxv2i64(<vscale x 2 x i64> %v0,
|
|
<vscale x 2 x i64> %v1,
|
|
<vscale x 2 x i1> %pred,
|
|
i64* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st2d_f64(<vscale x 2 x double> %v0, <vscale x 2 x double> %v1, <vscale x 2 x i1> %pred, double* %addr) {
|
|
; CHECK-LABEL: st2d_f64:
|
|
; CHECK: st2d { z0.d, z1.d }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st2.nxv2f64(<vscale x 2 x double> %v0,
|
|
<vscale x 2 x double> %v1,
|
|
<vscale x 2 x i1> %pred,
|
|
double* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st2d_ptr(<vscale x 2 x i8*> %v0, <vscale x 2 x i8*> %v1, <vscale x 2 x i1> %pred, i8** %addr) {
|
|
; CHECK-LABEL: st2d_ptr:
|
|
; CHECK: st2d { z0.d, z1.d }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st2.nxv2p0i8(<vscale x 2 x i8*> %v0,
|
|
<vscale x 2 x i8*> %v1,
|
|
<vscale x 2 x i1> %pred,
|
|
i8** %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; ST3B
|
|
;
|
|
|
|
define void @st3b_i8(<vscale x 16 x i8> %v0, <vscale x 16 x i8> %v1, <vscale x 16 x i8> %v2, <vscale x 16 x i1> %pred, i8* %addr) {
|
|
; CHECK-LABEL: st3b_i8:
|
|
; CHECK: st3b { z0.b, z1.b, z2.b }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st3.nxv16i8(<vscale x 16 x i8> %v0,
|
|
<vscale x 16 x i8> %v1,
|
|
<vscale x 16 x i8> %v2,
|
|
<vscale x 16 x i1> %pred,
|
|
i8* %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; ST3H
|
|
;
|
|
|
|
define void @st3h_i16(<vscale x 8 x i16> %v0, <vscale x 8 x i16> %v1, <vscale x 8 x i16> %v2, <vscale x 8 x i1> %pred, i16* %addr) {
|
|
; CHECK-LABEL: st3h_i16:
|
|
; CHECK: st3h { z0.h, z1.h, z2.h }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st3.nxv8i16(<vscale x 8 x i16> %v0,
|
|
<vscale x 8 x i16> %v1,
|
|
<vscale x 8 x i16> %v2,
|
|
<vscale x 8 x i1> %pred,
|
|
i16* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st3h_f16(<vscale x 8 x half> %v0, <vscale x 8 x half> %v1, <vscale x 8 x half> %v2, <vscale x 8 x i1> %pred, half* %addr) {
|
|
; CHECK-LABEL: st3h_f16:
|
|
; CHECK: st3h { z0.h, z1.h, z2.h }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st3.nxv8f16(<vscale x 8 x half> %v0,
|
|
<vscale x 8 x half> %v1,
|
|
<vscale x 8 x half> %v2,
|
|
<vscale x 8 x i1> %pred,
|
|
half* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st3h_bf16(<vscale x 8 x bfloat> %v0, <vscale x 8 x bfloat> %v1, <vscale x 8 x bfloat> %v2, <vscale x 8 x i1> %pred, bfloat* %addr) #0 {
|
|
; CHECK-LABEL: st3h_bf16:
|
|
; CHECK: st3h { z0.h, z1.h, z2.h }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st3.nxv8bf16(<vscale x 8 x bfloat> %v0,
|
|
<vscale x 8 x bfloat> %v1,
|
|
<vscale x 8 x bfloat> %v2,
|
|
<vscale x 8 x i1> %pred,
|
|
bfloat* %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; ST3W
|
|
;
|
|
|
|
define void @st3w_i32(<vscale x 4 x i32> %v0, <vscale x 4 x i32> %v1, <vscale x 4 x i32> %v2, <vscale x 4 x i1> %pred, i32* %addr) {
|
|
; CHECK-LABEL: st3w_i32:
|
|
; CHECK: st3w { z0.s, z1.s, z2.s }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st3.nxv4i32(<vscale x 4 x i32> %v0,
|
|
<vscale x 4 x i32> %v1,
|
|
<vscale x 4 x i32> %v2,
|
|
<vscale x 4 x i1> %pred,
|
|
i32* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st3w_f32(<vscale x 4 x float> %v0, <vscale x 4 x float> %v1, <vscale x 4 x float> %v2, <vscale x 4 x i1> %pred, float* %addr) {
|
|
; CHECK-LABEL: st3w_f32:
|
|
; CHECK: st3w { z0.s, z1.s, z2.s }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st3.nxv4f32(<vscale x 4 x float> %v0,
|
|
<vscale x 4 x float> %v1,
|
|
<vscale x 4 x float> %v2,
|
|
<vscale x 4 x i1> %pred,
|
|
float* %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; ST3D
|
|
;
|
|
|
|
define void @st3d_i64(<vscale x 2 x i64> %v0, <vscale x 2 x i64> %v1, <vscale x 2 x i64> %v2, <vscale x 2 x i1> %pred, i64* %addr) {
|
|
; CHECK-LABEL: st3d_i64:
|
|
; CHECK: st3d { z0.d, z1.d, z2.d }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st3.nxv2i64(<vscale x 2 x i64> %v0,
|
|
<vscale x 2 x i64> %v1,
|
|
<vscale x 2 x i64> %v2,
|
|
<vscale x 2 x i1> %pred,
|
|
i64* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st3d_f64(<vscale x 2 x double> %v0, <vscale x 2 x double> %v1, <vscale x 2 x double> %v2, <vscale x 2 x i1> %pred, double* %addr) {
|
|
; CHECK-LABEL: st3d_f64:
|
|
; CHECK: st3d { z0.d, z1.d, z2.d }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st3.nxv2f64(<vscale x 2 x double> %v0,
|
|
<vscale x 2 x double> %v1,
|
|
<vscale x 2 x double> %v2,
|
|
<vscale x 2 x i1> %pred,
|
|
double* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st3d_ptr(<vscale x 2 x i8*> %v0, <vscale x 2 x i8*> %v1, <vscale x 2 x i8*> %v2, <vscale x 2 x i1> %pred, i8** %addr) {
|
|
; CHECK-LABEL: st3d_ptr:
|
|
; CHECK: st3d { z0.d, z1.d, z2.d }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st3.nxv2p0i8(<vscale x 2 x i8*> %v0,
|
|
<vscale x 2 x i8*> %v1,
|
|
<vscale x 2 x i8*> %v2,
|
|
<vscale x 2 x i1> %pred,
|
|
i8** %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; ST4B
|
|
;
|
|
|
|
define void @st4b_i8(<vscale x 16 x i8> %v0, <vscale x 16 x i8> %v1, <vscale x 16 x i8> %v2, <vscale x 16 x i8> %v3, <vscale x 16 x i1> %pred, i8* %addr) {
|
|
; CHECK-LABEL: st4b_i8:
|
|
; CHECK: st4b { z0.b, z1.b, z2.b, z3.b }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st4.nxv16i8(<vscale x 16 x i8> %v0,
|
|
<vscale x 16 x i8> %v1,
|
|
<vscale x 16 x i8> %v2,
|
|
<vscale x 16 x i8> %v3,
|
|
<vscale x 16 x i1> %pred,
|
|
i8* %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; ST4H
|
|
;
|
|
|
|
define void @st4h_i16(<vscale x 8 x i16> %v0, <vscale x 8 x i16> %v1, <vscale x 8 x i16> %v2, <vscale x 8 x i16> %v3, <vscale x 8 x i1> %pred, i16* %addr) {
|
|
; CHECK-LABEL: st4h_i16:
|
|
; CHECK: st4h { z0.h, z1.h, z2.h, z3.h }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st4.nxv8i16(<vscale x 8 x i16> %v0,
|
|
<vscale x 8 x i16> %v1,
|
|
<vscale x 8 x i16> %v2,
|
|
<vscale x 8 x i16> %v3,
|
|
<vscale x 8 x i1> %pred,
|
|
i16* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st4h_f16(<vscale x 8 x half> %v0, <vscale x 8 x half> %v1, <vscale x 8 x half> %v2, <vscale x 8 x half> %v3, <vscale x 8 x i1> %pred, half* %addr) {
|
|
; CHECK-LABEL: st4h_f16:
|
|
; CHECK: st4h { z0.h, z1.h, z2.h, z3.h }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st4.nxv8f16(<vscale x 8 x half> %v0,
|
|
<vscale x 8 x half> %v1,
|
|
<vscale x 8 x half> %v2,
|
|
<vscale x 8 x half> %v3,
|
|
<vscale x 8 x i1> %pred,
|
|
half* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st4h_bf16(<vscale x 8 x bfloat> %v0, <vscale x 8 x bfloat> %v1, <vscale x 8 x bfloat> %v2, <vscale x 8 x bfloat> %v3, <vscale x 8 x i1> %pred, bfloat* %addr) #0 {
|
|
; CHECK-LABEL: st4h_bf16:
|
|
; CHECK: st4h { z0.h, z1.h, z2.h, z3.h }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st4.nxv8bf16(<vscale x 8 x bfloat> %v0,
|
|
<vscale x 8 x bfloat> %v1,
|
|
<vscale x 8 x bfloat> %v2,
|
|
<vscale x 8 x bfloat> %v3,
|
|
<vscale x 8 x i1> %pred,
|
|
bfloat* %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; ST4W
|
|
;
|
|
|
|
define void @st4w_i32(<vscale x 4 x i32> %v0, <vscale x 4 x i32> %v1, <vscale x 4 x i32> %v2, <vscale x 4 x i32> %v3, <vscale x 4 x i1> %pred, i32* %addr) {
|
|
; CHECK-LABEL: st4w_i32:
|
|
; CHECK: st4w { z0.s, z1.s, z2.s, z3.s }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st4.nxv4i32(<vscale x 4 x i32> %v0,
|
|
<vscale x 4 x i32> %v1,
|
|
<vscale x 4 x i32> %v2,
|
|
<vscale x 4 x i32> %v3,
|
|
<vscale x 4 x i1> %pred,
|
|
i32* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st4w_f32(<vscale x 4 x float> %v0, <vscale x 4 x float> %v1, <vscale x 4 x float> %v2, <vscale x 4 x float> %v3, <vscale x 4 x i1> %pred, float* %addr) {
|
|
; CHECK-LABEL: st4w_f32:
|
|
; CHECK: st4w { z0.s, z1.s, z2.s, z3.s }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st4.nxv4f32(<vscale x 4 x float> %v0,
|
|
<vscale x 4 x float> %v1,
|
|
<vscale x 4 x float> %v2,
|
|
<vscale x 4 x float> %v3,
|
|
<vscale x 4 x i1> %pred,
|
|
float* %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; ST4D
|
|
;
|
|
|
|
define void @st4d_i64(<vscale x 2 x i64> %v0, <vscale x 2 x i64> %v1, <vscale x 2 x i64> %v2, <vscale x 2 x i64> %v3, <vscale x 2 x i1> %pred, i64* %addr) {
|
|
; CHECK-LABEL: st4d_i64:
|
|
; CHECK: st4d { z0.d, z1.d, z2.d, z3.d }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st4.nxv2i64(<vscale x 2 x i64> %v0,
|
|
<vscale x 2 x i64> %v1,
|
|
<vscale x 2 x i64> %v2,
|
|
<vscale x 2 x i64> %v3,
|
|
<vscale x 2 x i1> %pred,
|
|
i64* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st4d_f64(<vscale x 2 x double> %v0, <vscale x 2 x double> %v1, <vscale x 2 x double> %v2, <vscale x 2 x double> %v3, <vscale x 2 x i1> %pred, double* %addr) {
|
|
; CHECK-LABEL: st4d_f64:
|
|
; CHECK: st4d { z0.d, z1.d, z2.d, z3.d }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st4.nxv2f64(<vscale x 2 x double> %v0,
|
|
<vscale x 2 x double> %v1,
|
|
<vscale x 2 x double> %v2,
|
|
<vscale x 2 x double> %v3,
|
|
<vscale x 2 x i1> %pred,
|
|
double* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @st4d_ptr(<vscale x 2 x i8*> %v0, <vscale x 2 x i8*> %v1, <vscale x 2 x i8*> %v2, <vscale x 2 x i8*> %v3, <vscale x 2 x i1> %pred, i8** %addr) {
|
|
; CHECK-LABEL: st4d_ptr:
|
|
; CHECK: st4d { z0.d, z1.d, z2.d, z3.d }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.st4.nxv2p0i8(<vscale x 2 x i8*> %v0,
|
|
<vscale x 2 x i8*> %v1,
|
|
<vscale x 2 x i8*> %v2,
|
|
<vscale x 2 x i8*> %v3,
|
|
<vscale x 2 x i1> %pred,
|
|
i8** %addr)
|
|
ret void
|
|
}
|
|
;
|
|
; STNT1B
|
|
;
|
|
|
|
define void @stnt1b_i8(<vscale x 16 x i8> %data, <vscale x 16 x i1> %pred, i8* %addr) {
|
|
; CHECK-LABEL: stnt1b_i8:
|
|
; CHECK: stnt1b { z0.b }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.stnt1.nxv16i8(<vscale x 16 x i8> %data,
|
|
<vscale x 16 x i1> %pred,
|
|
i8* %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; STNT1H
|
|
;
|
|
|
|
define void @stnt1h_i16(<vscale x 8 x i16> %data, <vscale x 8 x i1> %pred, i16* %addr) {
|
|
; CHECK-LABEL: stnt1h_i16:
|
|
; CHECK: stnt1h { z0.h }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.stnt1.nxv8i16(<vscale x 8 x i16> %data,
|
|
<vscale x 8 x i1> %pred,
|
|
i16* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @stnt1h_f16(<vscale x 8 x half> %data, <vscale x 8 x i1> %pred, half* %addr) {
|
|
; CHECK-LABEL: stnt1h_f16:
|
|
; CHECK: stnt1h { z0.h }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.stnt1.nxv8f16(<vscale x 8 x half> %data,
|
|
<vscale x 8 x i1> %pred,
|
|
half* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @stnt1h_bf16(<vscale x 8 x bfloat> %data, <vscale x 8 x i1> %pred, bfloat* %addr) #0 {
|
|
; CHECK-LABEL: stnt1h_bf16:
|
|
; CHECK: stnt1h { z0.h }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.stnt1.nxv8bf16(<vscale x 8 x bfloat> %data,
|
|
<vscale x 8 x i1> %pred,
|
|
bfloat* %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; STNT1W
|
|
;
|
|
|
|
define void @stnt1w_i32(<vscale x 4 x i32> %data, <vscale x 4 x i1> %pred, i32* %addr) {
|
|
; CHECK-LABEL: stnt1w_i32:
|
|
; CHECK: stnt1w { z0.s }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.stnt1.nxv4i32(<vscale x 4 x i32> %data,
|
|
<vscale x 4 x i1> %pred,
|
|
i32* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @stnt1w_f32(<vscale x 4 x float> %data, <vscale x 4 x i1> %pred, float* %addr) {
|
|
; CHECK-LABEL: stnt1w_f32:
|
|
; CHECK: stnt1w { z0.s }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.stnt1.nxv4f32(<vscale x 4 x float> %data,
|
|
<vscale x 4 x i1> %pred,
|
|
float* %addr)
|
|
ret void
|
|
}
|
|
|
|
;
|
|
; STNT1D
|
|
;
|
|
|
|
define void @stnt1d_i64(<vscale x 2 x i64> %data, <vscale x 2 x i1> %pred, i64* %addr) {
|
|
; CHECK-LABEL: stnt1d_i64:
|
|
; CHECK: stnt1d { z0.d }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.stnt1.nxv2i64(<vscale x 2 x i64> %data,
|
|
<vscale x 2 x i1> %pred,
|
|
i64* %addr)
|
|
ret void
|
|
}
|
|
|
|
define void @stnt1d_f64(<vscale x 2 x double> %data, <vscale x 2 x i1> %pred, double* %addr) {
|
|
; CHECK-LABEL: stnt1d_f64:
|
|
; CHECK: stnt1d { z0.d }, p0, [x0]
|
|
; CHECK-NEXT: ret
|
|
call void @llvm.aarch64.sve.stnt1.nxv2f64(<vscale x 2 x double> %data,
|
|
<vscale x 2 x i1> %pred,
|
|
double* %addr)
|
|
ret void
|
|
}
|
|
|
|
|
|
; Stores (tuples)
|
|
|
|
define void @store_i64_tuple3(<vscale x 6 x i64>* %out, <vscale x 2 x i64> %in1, <vscale x 2 x i64> %in2, <vscale x 2 x i64> %in3) {
|
|
; CHECK-LABEL: store_i64_tuple3
|
|
; CHECK: st1d { z2.d }, p0, [x0, #2, mul vl]
|
|
; CHECK-NEXT: st1d { z1.d }, p0, [x0, #1, mul vl]
|
|
; CHECK-NEXT: st1d { z0.d }, p0, [x0]
|
|
%tuple = tail call <vscale x 6 x i64> @llvm.aarch64.sve.tuple.create3.nxv6i64.nxv2i64(<vscale x 2 x i64> %in1, <vscale x 2 x i64> %in2, <vscale x 2 x i64> %in3)
|
|
store <vscale x 6 x i64> %tuple, <vscale x 6 x i64>* %out
|
|
ret void
|
|
}
|
|
|
|
define void @store_i64_tuple4(<vscale x 8 x i64>* %out, <vscale x 2 x i64> %in1, <vscale x 2 x i64> %in2, <vscale x 2 x i64> %in3, <vscale x 2 x i64> %in4) {
|
|
; CHECK-LABEL: store_i64_tuple4
|
|
; CHECK: st1d { z3.d }, p0, [x0, #3, mul vl]
|
|
; CHECK-NEXT: st1d { z2.d }, p0, [x0, #2, mul vl]
|
|
; CHECK-NEXT: st1d { z1.d }, p0, [x0, #1, mul vl]
|
|
; CHECK-NEXT: st1d { z0.d }, p0, [x0]
|
|
%tuple = tail call <vscale x 8 x i64> @llvm.aarch64.sve.tuple.create4.nxv8i64.nxv2i64(<vscale x 2 x i64> %in1, <vscale x 2 x i64> %in2, <vscale x 2 x i64> %in3, <vscale x 2 x i64> %in4)
|
|
store <vscale x 8 x i64> %tuple, <vscale x 8 x i64>* %out
|
|
ret void
|
|
}
|
|
|
|
define void @store_i16_tuple2(<vscale x 16 x i16>* %out, <vscale x 8 x i16> %in1, <vscale x 8 x i16> %in2) {
|
|
; CHECK-LABEL: store_i16_tuple2
|
|
; CHECK: st1h { z1.h }, p0, [x0, #1, mul vl]
|
|
; CHECK-NEXT: st1h { z0.h }, p0, [x0]
|
|
%tuple = tail call <vscale x 16 x i16> @llvm.aarch64.sve.tuple.create2.nxv16i16.nxv8i16(<vscale x 8 x i16> %in1, <vscale x 8 x i16> %in2)
|
|
store <vscale x 16 x i16> %tuple, <vscale x 16 x i16>* %out
|
|
ret void
|
|
}
|
|
|
|
define void @store_i16_tuple3(<vscale x 24 x i16>* %out, <vscale x 8 x i16> %in1, <vscale x 8 x i16> %in2, <vscale x 8 x i16> %in3) {
|
|
; CHECK-LABEL: store_i16_tuple3
|
|
; CHECK: st1h { z2.h }, p0, [x0, #2, mul vl]
|
|
; CHECK-NEXT: st1h { z1.h }, p0, [x0, #1, mul vl]
|
|
; CHECK-NEXT: st1h { z0.h }, p0, [x0]
|
|
%tuple = tail call <vscale x 24 x i16> @llvm.aarch64.sve.tuple.create3.nxv24i16.nxv8i16(<vscale x 8 x i16> %in1, <vscale x 8 x i16> %in2, <vscale x 8 x i16> %in3)
|
|
store <vscale x 24 x i16> %tuple, <vscale x 24 x i16>* %out
|
|
ret void
|
|
}
|
|
|
|
define void @store_f32_tuple3(<vscale x 12 x float>* %out, <vscale x 4 x float> %in1, <vscale x 4 x float> %in2, <vscale x 4 x float> %in3) {
|
|
; CHECK-LABEL: store_f32_tuple3
|
|
; CHECK: st1w { z2.s }, p0, [x0, #2, mul vl]
|
|
; CHECK-NEXT: st1w { z1.s }, p0, [x0, #1, mul vl]
|
|
; CHECK-NEXT: st1w { z0.s }, p0, [x0]
|
|
%tuple = tail call <vscale x 12 x float> @llvm.aarch64.sve.tuple.create3.nxv12f32.nxv4f32(<vscale x 4 x float> %in1, <vscale x 4 x float> %in2, <vscale x 4 x float> %in3)
|
|
store <vscale x 12 x float> %tuple, <vscale x 12 x float>* %out
|
|
ret void
|
|
}
|
|
|
|
define void @store_f32_tuple4(<vscale x 16 x float>* %out, <vscale x 4 x float> %in1, <vscale x 4 x float> %in2, <vscale x 4 x float> %in3, <vscale x 4 x float> %in4) {
|
|
; CHECK-LABEL: store_f32_tuple4
|
|
; CHECK: st1w { z3.s }, p0, [x0, #3, mul vl]
|
|
; CHECK-NEXT: st1w { z2.s }, p0, [x0, #2, mul vl]
|
|
; CHECK-NEXT: st1w { z1.s }, p0, [x0, #1, mul vl]
|
|
; CHECK-NEXT: st1w { z0.s }, p0, [x0]
|
|
%tuple = tail call <vscale x 16 x float> @llvm.aarch64.sve.tuple.create4.nxv16f32.nxv4f32(<vscale x 4 x float> %in1, <vscale x 4 x float> %in2, <vscale x 4 x float> %in3, <vscale x 4 x float> %in4)
|
|
store <vscale x 16 x float> %tuple, <vscale x 16 x float>* %out
|
|
ret void
|
|
}
|
|
|
|
declare void @llvm.aarch64.sve.st2.nxv16i8(<vscale x 16 x i8>, <vscale x 16 x i8>, <vscale x 16 x i1>, i8*)
|
|
declare void @llvm.aarch64.sve.st2.nxv8i16(<vscale x 8 x i16>, <vscale x 8 x i16>, <vscale x 8 x i1>, i16*)
|
|
declare void @llvm.aarch64.sve.st2.nxv4i32(<vscale x 4 x i32>, <vscale x 4 x i32>, <vscale x 4 x i1>, i32*)
|
|
declare void @llvm.aarch64.sve.st2.nxv2i64(<vscale x 2 x i64>, <vscale x 2 x i64>, <vscale x 2 x i1>, i64*)
|
|
declare void @llvm.aarch64.sve.st2.nxv8f16(<vscale x 8 x half>, <vscale x 8 x half>, <vscale x 8 x i1>, half*)
|
|
declare void @llvm.aarch64.sve.st2.nxv8bf16(<vscale x 8 x bfloat>, <vscale x 8 x bfloat>, <vscale x 8 x i1>, bfloat*)
|
|
declare void @llvm.aarch64.sve.st2.nxv4f32(<vscale x 4 x float>, <vscale x 4 x float>, <vscale x 4 x i1>, float*)
|
|
declare void @llvm.aarch64.sve.st2.nxv2f64(<vscale x 2 x double>, <vscale x 2 x double>, <vscale x 2 x i1>, double*)
|
|
declare void @llvm.aarch64.sve.st2.nxv2p0i8(<vscale x 2 x i8*>, <vscale x 2 x i8*>, <vscale x 2 x i1>, i8** nocapture)
|
|
|
|
declare void @llvm.aarch64.sve.st3.nxv16i8(<vscale x 16 x i8>, <vscale x 16 x i8>, <vscale x 16 x i8>, <vscale x 16 x i1>, i8*)
|
|
declare void @llvm.aarch64.sve.st3.nxv8i16(<vscale x 8 x i16>, <vscale x 8 x i16>, <vscale x 8 x i16>, <vscale x 8 x i1>, i16*)
|
|
declare void @llvm.aarch64.sve.st3.nxv4i32(<vscale x 4 x i32>, <vscale x 4 x i32>, <vscale x 4 x i32>, <vscale x 4 x i1>, i32*)
|
|
declare void @llvm.aarch64.sve.st3.nxv2i64(<vscale x 2 x i64>, <vscale x 2 x i64>, <vscale x 2 x i64>, <vscale x 2 x i1>, i64*)
|
|
declare void @llvm.aarch64.sve.st3.nxv8f16(<vscale x 8 x half>, <vscale x 8 x half>, <vscale x 8 x half>, <vscale x 8 x i1>, half*)
|
|
declare void @llvm.aarch64.sve.st3.nxv8bf16(<vscale x 8 x bfloat>, <vscale x 8 x bfloat>, <vscale x 8 x bfloat>, <vscale x 8 x i1>, bfloat*)
|
|
declare void @llvm.aarch64.sve.st3.nxv4f32(<vscale x 4 x float>, <vscale x 4 x float>, <vscale x 4 x float>, <vscale x 4 x i1>, float*)
|
|
declare void @llvm.aarch64.sve.st3.nxv2f64(<vscale x 2 x double>, <vscale x 2 x double>, <vscale x 2 x double>, <vscale x 2 x i1>, double*)
|
|
declare void @llvm.aarch64.sve.st3.nxv2p0i8(<vscale x 2 x i8*>, <vscale x 2 x i8*>, <vscale x 2 x i8*>, <vscale x 2 x i1>, i8** nocapture)
|
|
|
|
declare void @llvm.aarch64.sve.st4.nxv16i8(<vscale x 16 x i8>, <vscale x 16 x i8>, <vscale x 16 x i8>, <vscale x 16 x i8>, <vscale x 16 x i1>, i8*)
|
|
declare void @llvm.aarch64.sve.st4.nxv8i16(<vscale x 8 x i16>, <vscale x 8 x i16>, <vscale x 8 x i16>, <vscale x 8 x i16>, <vscale x 8 x i1>, i16*)
|
|
declare void @llvm.aarch64.sve.st4.nxv4i32(<vscale x 4 x i32>, <vscale x 4 x i32>, <vscale x 4 x i32>, <vscale x 4 x i32>, <vscale x 4 x i1>, i32*)
|
|
declare void @llvm.aarch64.sve.st4.nxv2i64(<vscale x 2 x i64>, <vscale x 2 x i64>, <vscale x 2 x i64>, <vscale x 2 x i64>, <vscale x 2 x i1>, i64*)
|
|
declare void @llvm.aarch64.sve.st4.nxv8f16(<vscale x 8 x half>, <vscale x 8 x half>, <vscale x 8 x half>, <vscale x 8 x half>, <vscale x 8 x i1>, half*)
|
|
declare void @llvm.aarch64.sve.st4.nxv8bf16(<vscale x 8 x bfloat>, <vscale x 8 x bfloat>, <vscale x 8 x bfloat>, <vscale x 8 x bfloat>, <vscale x 8 x i1>, bfloat*)
|
|
declare void @llvm.aarch64.sve.st4.nxv4f32(<vscale x 4 x float>, <vscale x 4 x float>, <vscale x 4 x float>, <vscale x 4 x float>, <vscale x 4 x i1>, float*)
|
|
declare void @llvm.aarch64.sve.st4.nxv2f64(<vscale x 2 x double>, <vscale x 2 x double>, <vscale x 2 x double>, <vscale x 2 x double>, <vscale x 2 x i1>, double*)
|
|
declare void @llvm.aarch64.sve.st4.nxv2p0i8(<vscale x 2 x i8*>, <vscale x 2 x i8*>, <vscale x 2 x i8*>, <vscale x 2 x i8*>, <vscale x 2 x i1>, i8** nocapture)
|
|
|
|
declare void @llvm.aarch64.sve.stnt1.nxv16i8(<vscale x 16 x i8>, <vscale x 16 x i1>, i8*)
|
|
declare void @llvm.aarch64.sve.stnt1.nxv8i16(<vscale x 8 x i16>, <vscale x 8 x i1>, i16*)
|
|
declare void @llvm.aarch64.sve.stnt1.nxv4i32(<vscale x 4 x i32>, <vscale x 4 x i1>, i32*)
|
|
declare void @llvm.aarch64.sve.stnt1.nxv2i64(<vscale x 2 x i64>, <vscale x 2 x i1>, i64*)
|
|
declare void @llvm.aarch64.sve.stnt1.nxv8f16(<vscale x 8 x half>, <vscale x 8 x i1>, half*)
|
|
declare void @llvm.aarch64.sve.stnt1.nxv8bf16(<vscale x 8 x bfloat>, <vscale x 8 x i1>, bfloat*)
|
|
declare void @llvm.aarch64.sve.stnt1.nxv4f32(<vscale x 4 x float>, <vscale x 4 x i1>, float*)
|
|
declare void @llvm.aarch64.sve.stnt1.nxv2f64(<vscale x 2 x double>, <vscale x 2 x i1>, double*)
|
|
|
|
declare <vscale x 6 x i64> @llvm.aarch64.sve.tuple.create3.nxv6i64.nxv2i64(<vscale x 2 x i64>, <vscale x 2 x i64>, <vscale x 2 x i64>)
|
|
declare <vscale x 8 x i64> @llvm.aarch64.sve.tuple.create4.nxv8i64.nxv2i64(<vscale x 2 x i64>, <vscale x 2 x i64>, <vscale x 2 x i64>, <vscale x 2 x i64>)
|
|
|
|
declare <vscale x 16 x i16> @llvm.aarch64.sve.tuple.create2.nxv16i16.nxv8i16(<vscale x 8 x i16>, <vscale x 8 x i16>)
|
|
declare <vscale x 24 x i16> @llvm.aarch64.sve.tuple.create3.nxv24i16.nxv8i16(<vscale x 8 x i16>, <vscale x 8 x i16>, <vscale x 8 x i16>)
|
|
|
|
declare <vscale x 12 x float> @llvm.aarch64.sve.tuple.create3.nxv12f32.nxv4f32(<vscale x 4 x float>, <vscale x 4 x float>, <vscale x 4 x float>)
|
|
declare <vscale x 16 x float> @llvm.aarch64.sve.tuple.create4.nxv16f32.nxv4f32(<vscale x 4 x float>, <vscale x 4 x float>, <vscale x 4 x float>, <vscale x 4 x float>)
|
|
|
|
; +bf16 is required for the bfloat version.
|
|
attributes #0 = { "target-features"="+sve,+bf16" }
|