; 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( %v0, %v1, %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( %v0, %v1, %pred, i8* %addr) ret void } ; ; ST2H ; define void @st2h_i16( %v0, %v1, %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( %v0, %v1, %pred, i16* %addr) ret void } define void @st2h_f16( %v0, %v1, %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( %v0, %v1, %pred, half* %addr) ret void } define void @st2h_bf16( %v0, %v1, %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( %v0, %v1, %pred, bfloat* %addr) ret void } ; ; ST2W ; define void @st2w_i32( %v0, %v1, %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( %v0, %v1, %pred, i32* %addr) ret void } define void @st2w_f32( %v0, %v1, %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( %v0, %v1, %pred, float* %addr) ret void } ; ; ST2D ; define void @st2d_i64( %v0, %v1, %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( %v0, %v1, %pred, i64* %addr) ret void } define void @st2d_f64( %v0, %v1, %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( %v0, %v1, %pred, double* %addr) ret void } define void @st2d_ptr( %v0, %v1, %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( %v0, %v1, %pred, i8** %addr) ret void } ; ; ST3B ; define void @st3b_i8( %v0, %v1, %v2, %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( %v0, %v1, %v2, %pred, i8* %addr) ret void } ; ; ST3H ; define void @st3h_i16( %v0, %v1, %v2, %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( %v0, %v1, %v2, %pred, i16* %addr) ret void } define void @st3h_f16( %v0, %v1, %v2, %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( %v0, %v1, %v2, %pred, half* %addr) ret void } define void @st3h_bf16( %v0, %v1, %v2, %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( %v0, %v1, %v2, %pred, bfloat* %addr) ret void } ; ; ST3W ; define void @st3w_i32( %v0, %v1, %v2, %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( %v0, %v1, %v2, %pred, i32* %addr) ret void } define void @st3w_f32( %v0, %v1, %v2, %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( %v0, %v1, %v2, %pred, float* %addr) ret void } ; ; ST3D ; define void @st3d_i64( %v0, %v1, %v2, %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( %v0, %v1, %v2, %pred, i64* %addr) ret void } define void @st3d_f64( %v0, %v1, %v2, %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( %v0, %v1, %v2, %pred, double* %addr) ret void } define void @st3d_ptr( %v0, %v1, %v2, %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( %v0, %v1, %v2, %pred, i8** %addr) ret void } ; ; ST4B ; define void @st4b_i8( %v0, %v1, %v2, %v3, %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( %v0, %v1, %v2, %v3, %pred, i8* %addr) ret void } ; ; ST4H ; define void @st4h_i16( %v0, %v1, %v2, %v3, %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( %v0, %v1, %v2, %v3, %pred, i16* %addr) ret void } define void @st4h_f16( %v0, %v1, %v2, %v3, %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( %v0, %v1, %v2, %v3, %pred, half* %addr) ret void } define void @st4h_bf16( %v0, %v1, %v2, %v3, %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( %v0, %v1, %v2, %v3, %pred, bfloat* %addr) ret void } ; ; ST4W ; define void @st4w_i32( %v0, %v1, %v2, %v3, %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( %v0, %v1, %v2, %v3, %pred, i32* %addr) ret void } define void @st4w_f32( %v0, %v1, %v2, %v3, %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( %v0, %v1, %v2, %v3, %pred, float* %addr) ret void } ; ; ST4D ; define void @st4d_i64( %v0, %v1, %v2, %v3, %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( %v0, %v1, %v2, %v3, %pred, i64* %addr) ret void } define void @st4d_f64( %v0, %v1, %v2, %v3, %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( %v0, %v1, %v2, %v3, %pred, double* %addr) ret void } define void @st4d_ptr( %v0, %v1, %v2, %v3, %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( %v0, %v1, %v2, %v3, %pred, i8** %addr) ret void } ; ; STNT1B ; define void @stnt1b_i8( %data, %pred, i8* %addr) { ; CHECK-LABEL: stnt1b_i8: ; CHECK: stnt1b { z0.b }, p0, [x0] ; CHECK-NEXT: ret call void @llvm.aarch64.sve.stnt1.nxv16i8( %data, %pred, i8* %addr) ret void } ; ; STNT1H ; define void @stnt1h_i16( %data, %pred, i16* %addr) { ; CHECK-LABEL: stnt1h_i16: ; CHECK: stnt1h { z0.h }, p0, [x0] ; CHECK-NEXT: ret call void @llvm.aarch64.sve.stnt1.nxv8i16( %data, %pred, i16* %addr) ret void } define void @stnt1h_f16( %data, %pred, half* %addr) { ; CHECK-LABEL: stnt1h_f16: ; CHECK: stnt1h { z0.h }, p0, [x0] ; CHECK-NEXT: ret call void @llvm.aarch64.sve.stnt1.nxv8f16( %data, %pred, half* %addr) ret void } define void @stnt1h_bf16( %data, %pred, bfloat* %addr) #0 { ; CHECK-LABEL: stnt1h_bf16: ; CHECK: stnt1h { z0.h }, p0, [x0] ; CHECK-NEXT: ret call void @llvm.aarch64.sve.stnt1.nxv8bf16( %data, %pred, bfloat* %addr) ret void } ; ; STNT1W ; define void @stnt1w_i32( %data, %pred, i32* %addr) { ; CHECK-LABEL: stnt1w_i32: ; CHECK: stnt1w { z0.s }, p0, [x0] ; CHECK-NEXT: ret call void @llvm.aarch64.sve.stnt1.nxv4i32( %data, %pred, i32* %addr) ret void } define void @stnt1w_f32( %data, %pred, float* %addr) { ; CHECK-LABEL: stnt1w_f32: ; CHECK: stnt1w { z0.s }, p0, [x0] ; CHECK-NEXT: ret call void @llvm.aarch64.sve.stnt1.nxv4f32( %data, %pred, float* %addr) ret void } ; ; STNT1D ; define void @stnt1d_i64( %data, %pred, i64* %addr) { ; CHECK-LABEL: stnt1d_i64: ; CHECK: stnt1d { z0.d }, p0, [x0] ; CHECK-NEXT: ret call void @llvm.aarch64.sve.stnt1.nxv2i64( %data, %pred, i64* %addr) ret void } define void @stnt1d_f64( %data, %pred, double* %addr) { ; CHECK-LABEL: stnt1d_f64: ; CHECK: stnt1d { z0.d }, p0, [x0] ; CHECK-NEXT: ret call void @llvm.aarch64.sve.stnt1.nxv2f64( %data, %pred, double* %addr) ret void } ; Stores (tuples) define void @store_i64_tuple3(* %out, %in1, %in2, %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 @llvm.aarch64.sve.tuple.create3.nxv6i64.nxv2i64( %in1, %in2, %in3) store %tuple, * %out ret void } define void @store_i64_tuple4(* %out, %in1, %in2, %in3, %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 @llvm.aarch64.sve.tuple.create4.nxv8i64.nxv2i64( %in1, %in2, %in3, %in4) store %tuple, * %out ret void } define void @store_i16_tuple2(* %out, %in1, %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 @llvm.aarch64.sve.tuple.create2.nxv16i16.nxv8i16( %in1, %in2) store %tuple, * %out ret void } define void @store_i16_tuple3(* %out, %in1, %in2, %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 @llvm.aarch64.sve.tuple.create3.nxv24i16.nxv8i16( %in1, %in2, %in3) store %tuple, * %out ret void } define void @store_f32_tuple3(* %out, %in1, %in2, %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 @llvm.aarch64.sve.tuple.create3.nxv12f32.nxv4f32( %in1, %in2, %in3) store %tuple, * %out ret void } define void @store_f32_tuple4(* %out, %in1, %in2, %in3, %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 @llvm.aarch64.sve.tuple.create4.nxv16f32.nxv4f32( %in1, %in2, %in3, %in4) store %tuple, * %out ret void } declare void @llvm.aarch64.sve.st2.nxv16i8(, , , i8*) declare void @llvm.aarch64.sve.st2.nxv8i16(, , , i16*) declare void @llvm.aarch64.sve.st2.nxv4i32(, , , i32*) declare void @llvm.aarch64.sve.st2.nxv2i64(, , , i64*) declare void @llvm.aarch64.sve.st2.nxv8f16(, , , half*) declare void @llvm.aarch64.sve.st2.nxv8bf16(, , , bfloat*) declare void @llvm.aarch64.sve.st2.nxv4f32(, , , float*) declare void @llvm.aarch64.sve.st2.nxv2f64(, , , double*) declare void @llvm.aarch64.sve.st2.nxv2p0i8(, , , i8** nocapture) declare void @llvm.aarch64.sve.st3.nxv16i8(, , , , i8*) declare void @llvm.aarch64.sve.st3.nxv8i16(, , , , i16*) declare void @llvm.aarch64.sve.st3.nxv4i32(, , , , i32*) declare void @llvm.aarch64.sve.st3.nxv2i64(, , , , i64*) declare void @llvm.aarch64.sve.st3.nxv8f16(, , , , half*) declare void @llvm.aarch64.sve.st3.nxv8bf16(, , , , bfloat*) declare void @llvm.aarch64.sve.st3.nxv4f32(, , , , float*) declare void @llvm.aarch64.sve.st3.nxv2f64(, , , , double*) declare void @llvm.aarch64.sve.st3.nxv2p0i8(, , , , i8** nocapture) declare void @llvm.aarch64.sve.st4.nxv16i8(, , , , , i8*) declare void @llvm.aarch64.sve.st4.nxv8i16(, , , , , i16*) declare void @llvm.aarch64.sve.st4.nxv4i32(, , , , , i32*) declare void @llvm.aarch64.sve.st4.nxv2i64(, , , , , i64*) declare void @llvm.aarch64.sve.st4.nxv8f16(, , , , , half*) declare void @llvm.aarch64.sve.st4.nxv8bf16(, , , , , bfloat*) declare void @llvm.aarch64.sve.st4.nxv4f32(, , , , , float*) declare void @llvm.aarch64.sve.st4.nxv2f64(, , , , , double*) declare void @llvm.aarch64.sve.st4.nxv2p0i8(, , , , , i8** nocapture) declare void @llvm.aarch64.sve.stnt1.nxv16i8(, , i8*) declare void @llvm.aarch64.sve.stnt1.nxv8i16(, , i16*) declare void @llvm.aarch64.sve.stnt1.nxv4i32(, , i32*) declare void @llvm.aarch64.sve.stnt1.nxv2i64(, , i64*) declare void @llvm.aarch64.sve.stnt1.nxv8f16(, , half*) declare void @llvm.aarch64.sve.stnt1.nxv8bf16(, , bfloat*) declare void @llvm.aarch64.sve.stnt1.nxv4f32(, , float*) declare void @llvm.aarch64.sve.stnt1.nxv2f64(, , double*) declare @llvm.aarch64.sve.tuple.create3.nxv6i64.nxv2i64(, , ) declare @llvm.aarch64.sve.tuple.create4.nxv8i64.nxv2i64(, , , ) declare @llvm.aarch64.sve.tuple.create2.nxv16i16.nxv8i16(, ) declare @llvm.aarch64.sve.tuple.create3.nxv24i16.nxv8i16(, , ) declare @llvm.aarch64.sve.tuple.create3.nxv12f32.nxv4f32(, , ) declare @llvm.aarch64.sve.tuple.create4.nxv16f32.nxv4f32(, , , ) ; +bf16 is required for the bfloat version. attributes #0 = { "target-features"="+sve,+bf16" }