xref: /llvm-project/llvm/test/CodeGen/DirectX/BufferStore-sm61.ll (revision 0e51b54b7ac02b0920e20b8ccae26b32bd6b6982) 
1; RUN: opt -S -dxil-op-lower %s | FileCheck %s
2; Before SM6.2 ByteAddressBuffer and StructuredBuffer lower to bufferStore.
3
4target triple = "dxil-pc-shadermodel6.1-compute"
5
6; CHECK-LABEL: define void @storef32_struct
7define 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, i1 false)
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 void
18}
19
20; CHECK-LABEL: define void @storef32_byte
21define 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, i1 false)
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 void
32}
33
34; CHECK-LABEL: define void @storev4f32_struct
35define 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, i1 false)
39
40  ; CHECK: [[DATA0:%.*]] = extractelement <4 x float> %data, i32 0
41  ; CHECK: [[DATA1:%.*]] = extractelement <4 x float> %data, i32 1
42  ; CHECK: [[DATA2:%.*]] = extractelement <4 x float> %data, i32 2
43  ; CHECK: [[DATA3:%.*]] = extractelement <4 x float> %data, i32 3
44  ; 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 void
50}
51
52; CHECK-LABEL: define void @storev4f32_byte
53define 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, i1 false)
57
58  ; CHECK: [[DATA0:%.*]] = extractelement <4 x float> %data, i32 0
59  ; CHECK: [[DATA1:%.*]] = extractelement <4 x float> %data, i32 1
60  ; CHECK: [[DATA2:%.*]] = extractelement <4 x float> %data, i32 2
61  ; CHECK: [[DATA3:%.*]] = extractelement <4 x float> %data, i32 3
62  ; 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 void
68}
69
70; CHECK-LABEL: define void @storeelements
71define 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, i1 false)
75
76  ; CHECK: [[DATA0_0:%.*]] = extractelement <4 x float> %data0, i32 0
77  ; CHECK: [[DATA0_1:%.*]] = extractelement <4 x float> %data0, i32 1
78  ; CHECK: [[DATA0_2:%.*]] = extractelement <4 x float> %data0, i32 2
79  ; CHECK: [[DATA0_3:%.*]] = extractelement <4 x float> %data0, i32 3
80  ; 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 0
86  ; CHECK: [[DATA1_1:%.*]] = extractelement <4 x i32> %data1, i32 1
87  ; CHECK: [[DATA1_2:%.*]] = extractelement <4 x i32> %data1, i32 2
88  ; CHECK: [[DATA1_3:%.*]] = extractelement <4 x i32> %data1, i32 3
89  ; 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 void
95}
96
97; CHECK-LABEL: define void @storenested
98define void @storenested(i32 %index, i32 %data0, <4 x float> %data1, <3 x half> %data2) {
99  %buffer = call
100      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, i1 false)
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 0
109  ; CHECK: [[DATA1_1:%.*]] = extractelement <4 x float> %data1, i32 1
110  ; CHECK: [[DATA1_2:%.*]] = extractelement <4 x float> %data1, i32 2
111  ; CHECK: [[DATA1_3:%.*]] = extractelement <4 x float> %data1, i32 3
112  ; 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 0
118  ; CHECK: [[DATA2_1:%.*]] = extractelement <3 x half> %data2, i32 1
119  ; CHECK: [[DATA2_2:%.*]] = extractelement <3 x half> %data2, i32 2
120  ; 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 void
126}
127