127 lines · plain
1; RUN: opt -S -dxil-op-lower %s | FileCheck %s2; Before SM6.2 ByteAddressBuffer and StructuredBuffer lower to bufferStore.3 4target triple = "dxil-pc-shadermodel6.1-compute"5 6; CHECK-LABEL: define void @storef32_struct7define void @storef32_struct(i32 %index, float %data) {8 %buffer = call target("dx.RawBuffer", float, 1, 0, 0)9 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_f32_0_0_0(10 i32 0, i32 0, i32 1, i32 0, ptr null)11 12 ; CHECK: call void @dx.op.bufferStore.f32(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 0, float %data, float undef, float undef, float undef, i8 1)13 call void @llvm.dx.resource.store.rawbuffer.f32(14 target("dx.RawBuffer", float, 1, 0, 0) %buffer,15 i32 %index, i32 0, float %data)16 17 ret void18}19 20; CHECK-LABEL: define void @storef32_byte21define void @storef32_byte(i32 %offset, float %data) {22 %buffer = call target("dx.RawBuffer", i8, 1, 0, 0)23 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(24 i32 0, i32 0, i32 1, i32 0, ptr null)25 26 ; CHECK: call void @dx.op.bufferStore.f32(i32 69, %dx.types.Handle %{{.*}}, i32 %offset, i32 0, float %data, float undef, float undef, float undef, i8 1)27 call void @llvm.dx.resource.store.rawbuffer.f32(28 target("dx.RawBuffer", i8, 1, 0, 0) %buffer,29 i32 %offset, i32 0, float %data)30 31 ret void32}33 34; CHECK-LABEL: define void @storev4f32_struct35define void @storev4f32_struct(i32 %index, <4 x float> %data) {36 %buffer = call target("dx.RawBuffer", <4 x float>, 1, 0, 0)37 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_v4f32_0_0_0(38 i32 0, i32 0, i32 1, i32 0, ptr null)39 40 ; CHECK: [[DATA0:%.*]] = extractelement <4 x float> %data, i32 041 ; CHECK: [[DATA1:%.*]] = extractelement <4 x float> %data, i32 142 ; CHECK: [[DATA2:%.*]] = extractelement <4 x float> %data, i32 243 ; CHECK: [[DATA3:%.*]] = extractelement <4 x float> %data, i32 344 ; CHECK: call void @dx.op.bufferStore.f32(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 0, float [[DATA0]], float [[DATA1]], float [[DATA2]], float [[DATA3]], i8 15)45 call void @llvm.dx.resource.store.rawbuffer.v4f32(46 target("dx.RawBuffer", <4 x float>, 1, 0, 0) %buffer,47 i32 %index, i32 0, <4 x float> %data)48 49 ret void50}51 52; CHECK-LABEL: define void @storev4f32_byte53define void @storev4f32_byte(i32 %offset, <4 x float> %data) {54 %buffer = call target("dx.RawBuffer", i8, 1, 0, 0)55 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(56 i32 0, i32 0, i32 1, i32 0, ptr null)57 58 ; CHECK: [[DATA0:%.*]] = extractelement <4 x float> %data, i32 059 ; CHECK: [[DATA1:%.*]] = extractelement <4 x float> %data, i32 160 ; CHECK: [[DATA2:%.*]] = extractelement <4 x float> %data, i32 261 ; CHECK: [[DATA3:%.*]] = extractelement <4 x float> %data, i32 362 ; CHECK: call void @dx.op.bufferStore.f32(i32 69, %dx.types.Handle %{{.*}}, i32 %offset, i32 0, float [[DATA0]], float [[DATA1]], float [[DATA2]], float [[DATA3]], i8 15)63 call void @llvm.dx.resource.store.rawbuffer.v4f32(64 target("dx.RawBuffer", i8, 1, 0, 0) %buffer,65 i32 %offset, i32 0, <4 x float> %data)66 67 ret void68}69 70; CHECK-LABEL: define void @storeelements71define void @storeelements(i32 %index, <4 x float> %data0, <4 x i32> %data1) {72 %buffer = call target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 1, 0, 0)73 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_sl_v4f32v4i32s_0_0_0(74 i32 0, i32 0, i32 1, i32 0, ptr null)75 76 ; CHECK: [[DATA0_0:%.*]] = extractelement <4 x float> %data0, i32 077 ; CHECK: [[DATA0_1:%.*]] = extractelement <4 x float> %data0, i32 178 ; CHECK: [[DATA0_2:%.*]] = extractelement <4 x float> %data0, i32 279 ; CHECK: [[DATA0_3:%.*]] = extractelement <4 x float> %data0, i32 380 ; CHECK: call void @dx.op.bufferStore.f32(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 0, float [[DATA0_0]], float [[DATA0_1]], float [[DATA0_2]], float [[DATA0_3]], i8 15)81 call void @llvm.dx.resource.store.rawbuffer.v4f32(82 target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 1, 0, 0) %buffer,83 i32 %index, i32 0, <4 x float> %data0)84 85 ; CHECK: [[DATA1_0:%.*]] = extractelement <4 x i32> %data1, i32 086 ; CHECK: [[DATA1_1:%.*]] = extractelement <4 x i32> %data1, i32 187 ; CHECK: [[DATA1_2:%.*]] = extractelement <4 x i32> %data1, i32 288 ; CHECK: [[DATA1_3:%.*]] = extractelement <4 x i32> %data1, i32 389 ; CHECK: call void @dx.op.bufferStore.i32(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 16, i32 [[DATA1_0]], i32 [[DATA1_1]], i32 [[DATA1_2]], i32 [[DATA1_3]], i8 15)90 call void @llvm.dx.resource.store.rawbuffer.v4i32(91 target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 1, 0, 0) %buffer,92 i32 %index, i32 16, <4 x i32> %data1)93 94 ret void95}96 97; CHECK-LABEL: define void @storenested98define void @storenested(i32 %index, i32 %data0, <4 x float> %data1, <3 x half> %data2) {99 %buffer = call100 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0)101 @llvm.dx.resource.handlefrombinding(i32 0, i32 0, i32 1, i32 0, ptr null)102 103 ; CHECK: call void @dx.op.bufferStore.i32(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 0, i32 %data0, i32 undef, i32 undef, i32 undef, i8 1)104 call void @llvm.dx.resource.store.rawbuffer.i32(105 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) %buffer,106 i32 %index, i32 0, i32 %data0)107 108 ; CHECK: [[DATA1_0:%.*]] = extractelement <4 x float> %data1, i32 0109 ; CHECK: [[DATA1_1:%.*]] = extractelement <4 x float> %data1, i32 1110 ; CHECK: [[DATA1_2:%.*]] = extractelement <4 x float> %data1, i32 2111 ; CHECK: [[DATA1_3:%.*]] = extractelement <4 x float> %data1, i32 3112 ; CHECK: call void @dx.op.bufferStore.f32(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 4, float [[DATA1_0]], float [[DATA1_1]], float [[DATA1_2]], float [[DATA1_3]], i8 15)113 call void @llvm.dx.resource.store.rawbuffer.v4f32(114 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) %buffer,115 i32 %index, i32 4, <4 x float> %data1)116 117 ; CHECK: [[DATA2_0:%.*]] = extractelement <3 x half> %data2, i32 0118 ; CHECK: [[DATA2_1:%.*]] = extractelement <3 x half> %data2, i32 1119 ; CHECK: [[DATA2_2:%.*]] = extractelement <3 x half> %data2, i32 2120 ; CHECK: call void @dx.op.bufferStore.f16(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 20, half [[DATA2_0]], half [[DATA2_1]], half [[DATA2_2]], half undef, i8 7)121 call void @llvm.dx.resource.store.rawbuffer.v3f16(122 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) %buffer,123 i32 %index, i32 20, <3 x half> %data2)124 125 ret void126}127