145 lines · plain
1; RUN: opt -S -dxil-op-lower %s | FileCheck %s2 3target triple = "dxil-pc-shadermodel6.6-compute"4 5; CHECK-LABEL: define void @storef32_struct6define void @storef32_struct(i32 %index, float %data) {7 %buffer = call target("dx.RawBuffer", float, 1, 0, 0)8 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_f32_0_0_0(9 i32 0, i32 0, i32 1, i32 0, ptr null)10 11 ; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 0, float %data, float undef, float undef, float undef, i8 1, i32 4)12 call void @llvm.dx.resource.store.rawbuffer.f32(13 target("dx.RawBuffer", float, 1, 0, 0) %buffer,14 i32 %index, i32 0, float %data)15 16 ret void17}18 19; CHECK-LABEL: define void @storef32_byte20define void @storef32_byte(i32 %offset, float %data) {21 %buffer = call target("dx.RawBuffer", i8, 1, 0, 0)22 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(23 i32 0, i32 0, i32 1, i32 0, ptr null)24 25 ; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %offset, i32 0, float %data, float undef, float undef, float undef, i8 1, i32 4)26 call void @llvm.dx.resource.store.rawbuffer.f32(27 target("dx.RawBuffer", i8, 1, 0, 0) %buffer,28 i32 %offset, i32 0, float %data)29 30 ret void31}32 33; CHECK-LABEL: define void @storev4f32_struct34define void @storev4f32_struct(i32 %index, <4 x float> %data) {35 %buffer = call target("dx.RawBuffer", <4 x float>, 1, 0, 0)36 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_v4f32_0_0_0(37 i32 0, i32 0, i32 1, i32 0, ptr null)38 39 ; CHECK: [[DATA0:%.*]] = extractelement <4 x float> %data, i32 040 ; CHECK: [[DATA1:%.*]] = extractelement <4 x float> %data, i32 141 ; CHECK: [[DATA2:%.*]] = extractelement <4 x float> %data, i32 242 ; CHECK: [[DATA3:%.*]] = extractelement <4 x float> %data, i32 343 ; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 0, float [[DATA0]], float [[DATA1]], float [[DATA2]], float [[DATA3]], i8 15, i32 4)44 call void @llvm.dx.resource.store.rawbuffer.v4f32(45 target("dx.RawBuffer", <4 x float>, 1, 0, 0) %buffer,46 i32 %index, i32 0, <4 x float> %data)47 48 ret void49}50 51; CHECK-LABEL: define void @storev4f32_byte52define void @storev4f32_byte(i32 %offset, <4 x float> %data) {53 %buffer = call target("dx.RawBuffer", i8, 1, 0, 0)54 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(55 i32 0, i32 0, i32 1, i32 0, ptr null)56 57 ; CHECK: [[DATA0:%.*]] = extractelement <4 x float> %data, i32 058 ; CHECK: [[DATA1:%.*]] = extractelement <4 x float> %data, i32 159 ; CHECK: [[DATA2:%.*]] = extractelement <4 x float> %data, i32 260 ; CHECK: [[DATA3:%.*]] = extractelement <4 x float> %data, i32 361 ; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %offset, i32 0, float [[DATA0]], float [[DATA1]], float [[DATA2]], float [[DATA3]], i8 15, i32 4)62 call void @llvm.dx.resource.store.rawbuffer.v4f32(63 target("dx.RawBuffer", i8, 1, 0, 0) %buffer,64 i32 %offset, i32 0, <4 x float> %data)65 66 ret void67}68 69; CHECK-LABEL: define void @storeelements70define void @storeelements(i32 %index, <4 x float> %data0, <4 x i32> %data1) {71 %buffer = call target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 1, 0, 0)72 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_sl_v4f32v4i32s_0_0_0(73 i32 0, i32 0, i32 1, i32 0, ptr null)74 75 ; CHECK: [[DATA0_0:%.*]] = extractelement <4 x float> %data0, i32 076 ; CHECK: [[DATA0_1:%.*]] = extractelement <4 x float> %data0, i32 177 ; CHECK: [[DATA0_2:%.*]] = extractelement <4 x float> %data0, i32 278 ; CHECK: [[DATA0_3:%.*]] = extractelement <4 x float> %data0, i32 379 ; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 0, float [[DATA0_0]], float [[DATA0_1]], float [[DATA0_2]], float [[DATA0_3]], i8 15, i32 4)80 call void @llvm.dx.resource.store.rawbuffer.v4f32(81 target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 1, 0, 0) %buffer,82 i32 %index, i32 0, <4 x float> %data0)83 84 ; CHECK: [[DATA1_0:%.*]] = extractelement <4 x i32> %data1, i32 085 ; CHECK: [[DATA1_1:%.*]] = extractelement <4 x i32> %data1, i32 186 ; CHECK: [[DATA1_2:%.*]] = extractelement <4 x i32> %data1, i32 287 ; CHECK: [[DATA1_3:%.*]] = extractelement <4 x i32> %data1, i32 388 ; CHECK: call void @dx.op.rawBufferStore.i32(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 16, i32 [[DATA1_0]], i32 [[DATA1_1]], i32 [[DATA1_2]], i32 [[DATA1_3]], i8 15, i32 4)89 call void @llvm.dx.resource.store.rawbuffer.v4i32(90 target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 1, 0, 0) %buffer,91 i32 %index, i32 16, <4 x i32> %data1)92 93 ret void94}95 96; CHECK-LABEL: define void @storenested97define void @storenested(i32 %index, i32 %data0, <4 x float> %data1, <3 x half> %data2) {98 %buffer = call99 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0)100 @llvm.dx.resource.handlefrombinding(i32 0, i32 0, i32 1, i32 0, ptr null)101 102 ; CHECK: call void @dx.op.rawBufferStore.i32(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 0, i32 %data0, i32 undef, i32 undef, i32 undef, i8 1, i32 4)103 call void @llvm.dx.resource.store.rawbuffer.i32(104 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) %buffer,105 i32 %index, i32 0, i32 %data0)106 107 ; CHECK: [[DATA1_0:%.*]] = extractelement <4 x float> %data1, i32 0108 ; CHECK: [[DATA1_1:%.*]] = extractelement <4 x float> %data1, i32 1109 ; CHECK: [[DATA1_2:%.*]] = extractelement <4 x float> %data1, i32 2110 ; CHECK: [[DATA1_3:%.*]] = extractelement <4 x float> %data1, i32 3111 ; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 4, float [[DATA1_0]], float [[DATA1_1]], float [[DATA1_2]], float [[DATA1_3]], i8 15, i32 4)112 call void @llvm.dx.resource.store.rawbuffer.v4f32(113 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) %buffer,114 i32 %index, i32 4, <4 x float> %data1)115 116 ; CHECK: [[DATA2_0:%.*]] = extractelement <3 x half> %data2, i32 0117 ; CHECK: [[DATA2_1:%.*]] = extractelement <3 x half> %data2, i32 1118 ; CHECK: [[DATA2_2:%.*]] = extractelement <3 x half> %data2, i32 2119 ; CHECK: call void @dx.op.rawBufferStore.f16(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 20, half [[DATA2_0]], half [[DATA2_1]], half [[DATA2_2]], half undef, i8 7, i32 2)120 call void @llvm.dx.resource.store.rawbuffer.v3f16(121 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) %buffer,122 i32 %index, i32 20, <3 x half> %data2)123 124 ret void125}126 127; byteaddressbuf.Store<int64_t4>128; CHECK-LABEL: define void @storev4f64_byte129define void @storev4f64_byte(i32 %offset, <4 x double> %data) {130 %buffer = call target("dx.RawBuffer", i8, 1, 0, 0)131 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(132 i32 0, i32 0, i32 1, i32 0, ptr null)133 134 ; CHECK: [[DATA0:%.*]] = extractelement <4 x double> %data, i32 0135 ; CHECK: [[DATA1:%.*]] = extractelement <4 x double> %data, i32 1136 ; CHECK: [[DATA2:%.*]] = extractelement <4 x double> %data, i32 2137 ; CHECK: [[DATA3:%.*]] = extractelement <4 x double> %data, i32 3138 ; CHECK: call void @dx.op.rawBufferStore.f64(i32 140, %dx.types.Handle %buffer_annot, i32 %offset, i32 0, double [[DATA0]], double [[DATA1]], double [[DATA2]], double [[DATA3]], i8 15, i32 8)139 call void @llvm.dx.resource.store.rawbuffer.v4i64(140 target("dx.RawBuffer", i8, 1, 0, 0) %buffer,141 i32 %offset, i32 0, <4 x double> %data)142 143 ret void144}145