100 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py2; RUN: llc -mtriple=aarch64--linux-gnu -mattr=+sve < %s | FileCheck %s3 4define void @masked_scatter_nxv2i8(<vscale x 2 x i8> %data, <vscale x 2 x ptr> %bases, i64 %offset, <vscale x 2 x i1> %mask) {5; CHECK-LABEL: masked_scatter_nxv2i8:6; CHECK: // %bb.0:7; CHECK-NEXT: st1b { z0.d }, p0, [x0, z1.d]8; CHECK-NEXT: ret9 %ptrs = getelementptr i8, <vscale x 2 x ptr> %bases, i64 %offset10 call void @llvm.masked.scatter.nxv2i8(<vscale x 2 x i8> %data, <vscale x 2 x ptr> %ptrs, i32 1, <vscale x 2 x i1> %mask)11 ret void12}13 14define void @masked_scatter_nxv2i16(<vscale x 2 x i16> %data, <vscale x 2 x ptr> %bases, i64 %offset, <vscale x 2 x i1> %mask) {15; CHECK-LABEL: masked_scatter_nxv2i16:16; CHECK: // %bb.0:17; CHECK-NEXT: st1h { z0.d }, p0, [x0, z1.d]18; CHECK-NEXT: ret19 %byte_ptrs = getelementptr i8, <vscale x 2 x ptr> %bases, i64 %offset20 %ptrs = bitcast <vscale x 2 x ptr> %byte_ptrs to <vscale x 2 x ptr>21 call void @llvm.masked.scatter.nxv2i16(<vscale x 2 x i16> %data, <vscale x 2 x ptr> %ptrs, i32 2, <vscale x 2 x i1> %mask)22 ret void23}24 25define void @masked_scatter_nxv2i32(<vscale x 2 x i32> %data, <vscale x 2 x ptr> %bases, i64 %offset, <vscale x 2 x i1> %mask) {26; CHECK-LABEL: masked_scatter_nxv2i32:27; CHECK: // %bb.0:28; CHECK-NEXT: st1w { z0.d }, p0, [x0, z1.d]29; CHECK-NEXT: ret30 %byte_ptrs = getelementptr i8, <vscale x 2 x ptr> %bases, i64 %offset31 %ptrs = bitcast <vscale x 2 x ptr> %byte_ptrs to <vscale x 2 x ptr>32 call void @llvm.masked.scatter.nxv2i32(<vscale x 2 x i32> %data, <vscale x 2 x ptr> %ptrs, i32 4, <vscale x 2 x i1> %mask)33 ret void34}35 36define void @masked_scatter_nxv2i64(<vscale x 2 x i64> %data, <vscale x 2 x ptr> %bases, i64 %offset, <vscale x 2 x i1> %mask) {37; CHECK-LABEL: masked_scatter_nxv2i64:38; CHECK: // %bb.0:39; CHECK-NEXT: st1d { z0.d }, p0, [x0, z1.d]40; CHECK-NEXT: ret41 %byte_ptrs = getelementptr i8, <vscale x 2 x ptr> %bases, i64 %offset42 %ptrs = bitcast <vscale x 2 x ptr> %byte_ptrs to <vscale x 2 x ptr>43 call void @llvm.masked.scatter.nxv2i64(<vscale x 2 x i64> %data, <vscale x 2 x ptr> %ptrs, i32 8, <vscale x 2 x i1> %mask)44 ret void45}46 47define void @masked_scatter_nxv2f16(<vscale x 2 x half> %data, <vscale x 2 x ptr> %bases, i64 %offset, <vscale x 2 x i1> %mask) {48; CHECK-LABEL: masked_scatter_nxv2f16:49; CHECK: // %bb.0:50; CHECK-NEXT: st1h { z0.d }, p0, [x0, z1.d]51; CHECK-NEXT: ret52 %byte_ptrs = getelementptr i8, <vscale x 2 x ptr> %bases, i64 %offset53 %ptrs = bitcast <vscale x 2 x ptr> %byte_ptrs to <vscale x 2 x ptr>54 call void @llvm.masked.scatter.nxv2f16(<vscale x 2 x half> %data, <vscale x 2 x ptr> %ptrs, i32 2, <vscale x 2 x i1> %mask)55 ret void56}57 58define void @masked_scatter_nxv2bf16(<vscale x 2 x bfloat> %data, <vscale x 2 x ptr> %bases, i64 %offset, <vscale x 2 x i1> %mask) #0 {59; CHECK-LABEL: masked_scatter_nxv2bf16:60; CHECK: // %bb.0:61; CHECK-NEXT: st1h { z0.d }, p0, [x0, z1.d]62; CHECK-NEXT: ret63 %byte_ptrs = getelementptr i8, <vscale x 2 x ptr> %bases, i64 %offset64 %ptrs = bitcast <vscale x 2 x ptr> %byte_ptrs to <vscale x 2 x ptr>65 call void @llvm.masked.scatter.nxv2bf16(<vscale x 2 x bfloat> %data, <vscale x 2 x ptr> %ptrs, i32 2, <vscale x 2 x i1> %mask)66 ret void67}68 69define void @masked_scatter_nxv2f32(<vscale x 2 x float> %data, <vscale x 2 x ptr> %bases, i64 %offset, <vscale x 2 x i1> %mask) {70; CHECK-LABEL: masked_scatter_nxv2f32:71; CHECK: // %bb.0:72; CHECK-NEXT: st1w { z0.d }, p0, [x0, z1.d]73; CHECK-NEXT: ret74 %byte_ptrs = getelementptr i8, <vscale x 2 x ptr> %bases, i64 %offset75 %ptrs = bitcast <vscale x 2 x ptr> %byte_ptrs to <vscale x 2 x ptr>76 call void @llvm.masked.scatter.nxv2f32(<vscale x 2 x float> %data, <vscale x 2 x ptr> %ptrs, i32 4, <vscale x 2 x i1> %mask)77 ret void78}79 80define void @masked_scatter_nxv2f64(<vscale x 2 x double> %data, <vscale x 2 x ptr> %bases, i64 %offset, <vscale x 2 x i1> %mask) {81; CHECK-LABEL: masked_scatter_nxv2f64:82; CHECK: // %bb.0:83; CHECK-NEXT: st1d { z0.d }, p0, [x0, z1.d]84; CHECK-NEXT: ret85 %byte_ptrs = getelementptr i8, <vscale x 2 x ptr> %bases, i64 %offset86 %ptrs = bitcast <vscale x 2 x ptr> %byte_ptrs to <vscale x 2 x ptr>87 call void @llvm.masked.scatter.nxv2f64(<vscale x 2 x double> %data, <vscale x 2 x ptr> %ptrs, i32 8, <vscale x 2 x i1> %mask)88 ret void89}90 91declare void @llvm.masked.scatter.nxv2i8(<vscale x 2 x i8>, <vscale x 2 x ptr>, i32, <vscale x 2 x i1>)92declare void @llvm.masked.scatter.nxv2i16(<vscale x 2 x i16>, <vscale x 2 x ptr>, i32, <vscale x 2 x i1>)93declare void @llvm.masked.scatter.nxv2i32(<vscale x 2 x i32>, <vscale x 2 x ptr>, i32, <vscale x 2 x i1>)94declare void @llvm.masked.scatter.nxv2i64(<vscale x 2 x i64>, <vscale x 2 x ptr>, i32, <vscale x 2 x i1>)95declare void @llvm.masked.scatter.nxv2f16(<vscale x 2 x half>, <vscale x 2 x ptr>, i32, <vscale x 2 x i1>)96declare void @llvm.masked.scatter.nxv2bf16(<vscale x 2 x bfloat>, <vscale x 2 x ptr>, i32, <vscale x 2 x i1>)97declare void @llvm.masked.scatter.nxv2f32(<vscale x 2 x float>, <vscale x 2 x ptr>, i32, <vscale x 2 x i1>)98declare void @llvm.masked.scatter.nxv2f64(<vscale x 2 x double>, <vscale x 2 x ptr>, i32, <vscale x 2 x i1>)99attributes #0 = { "target-features"="+sve,+bf16" }100