xref: /llvm-project/llvm/test/CodeGen/DirectX/ResourceAccess/load_rawbuffer.ll (revision 2f39d138dc38a1fdf4754e4e26dd0aeb7409b13d)
1; RUN: opt -S -dxil-resource-access %s | FileCheck %s
2
3target triple = "dxil-pc-shadermodel6.6-compute"
4
5declare void @f32_user(float)
6declare void @v4f32_user(<4 x float>)
7declare void @i32_user(i32)
8declare void @v4i32_user(<4 x i32>)
9declare void @v3f16_user(<3 x half>)
10declare void @v4f64_user(<4 x double>)
11
12; CHECK-LABEL: define void @loadf32_struct
13define void @loadf32_struct(i32 %index) {
14  %buffer = call target("dx.RawBuffer", float, 0, 0, 0)
15      @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_f32_0_0_0(
16          i32 0, i32 0, i32 1, i32 0, i1 false)
17
18  ; CHECK-NOT: @llvm.dx.resource.getpointer
19  %ptr = call ptr @llvm.dx.resource.getpointer(
20      target("dx.RawBuffer", float, 0, 0, 0) %buffer, i32 %index)
21
22  ; CHECK: %[[LOAD:.*]] = call { float, i1 } @llvm.dx.resource.load.rawbuffer.f32.tdx.RawBuffer_f32_0_0_0t(target("dx.RawBuffer", float, 0, 0, 0) %buffer, i32 %index, i32 0)
23  ; CHECK: %[[VAL:.*]] = extractvalue { float, i1 } %[[LOAD]], 0
24  ; CHECK: call void @f32_user(float %[[VAL]])
25  %data = load float, ptr %ptr
26  call void @f32_user(float %data)
27
28  ret void
29}
30
31; CHECK-LABEL: define void @loadf32_byte
32define void @loadf32_byte(i32 %offset) {
33  %buffer = call target("dx.RawBuffer", i8, 0, 0, 0)
34      @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(
35          i32 0, i32 0, i32 1, i32 0, i1 false)
36
37  ; CHECK-NOT: @llvm.dx.resource.getpointer
38  %ptr = call ptr @llvm.dx.resource.getpointer(
39      target("dx.RawBuffer", i8, 0, 0, 0) %buffer, i32 %offset)
40
41  ; CHECK: %[[LOAD:.*]] = call { float, i1 } @llvm.dx.resource.load.rawbuffer.f32.tdx.RawBuffer_i8_0_0_0t(target("dx.RawBuffer", i8, 0, 0, 0) %buffer, i32 %offset, i32 0)
42  ; CHECK: %[[VAL:.*]] = extractvalue { float, i1 } %[[LOAD]], 0
43  ; CHECK: call void @f32_user(float %[[VAL]])
44  %data = load float, ptr %ptr
45  call void @f32_user(float %data)
46
47  ret void
48}
49
50; CHECK-LABEL: define void @loadv4f32_struct
51define void @loadv4f32_struct(i32 %index) {
52  %buffer = call target("dx.RawBuffer", <4 x float>, 0, 0, 0)
53      @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_v4f32_0_0_0(
54          i32 0, i32 0, i32 1, i32 0, i1 false)
55
56  ; CHECK-NOT: @llvm.dx.resource.getpointer
57  %ptr = call ptr @llvm.dx.resource.getpointer(
58      target("dx.RawBuffer", <4 x float>, 0, 0, 0) %buffer, i32 %index)
59
60  ; CHECK: %[[LOAD:.*]] = call { <4 x float>, i1 } @llvm.dx.resource.load.rawbuffer.v4f32.tdx.RawBuffer_v4f32_0_0_0t(target("dx.RawBuffer", <4 x float>, 0, 0, 0) %buffer, i32 %index, i32 0)
61  ; CHECK: %[[VAL:.*]] = extractvalue { <4 x float>, i1 } %[[LOAD]], 0
62  ; CHECK: call void @v4f32_user(<4 x float> %[[VAL]])
63  %data = load <4 x float>, ptr %ptr
64  call void @v4f32_user(<4 x float> %data)
65
66  ret void
67}
68
69; CHECK-LABEL: define void @loadv4f32_byte
70define void @loadv4f32_byte(i32 %offset) {
71  %buffer = call target("dx.RawBuffer", i8, 0, 0, 0)
72      @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(
73          i32 0, i32 0, i32 1, i32 0, i1 false)
74
75  ; CHECK-NOT: @llvm.dx.resource.getpointer
76  %ptr = call ptr @llvm.dx.resource.getpointer(
77      target("dx.RawBuffer", i8, 0, 0, 0) %buffer, i32 %offset)
78
79  ; CHECK: %[[LOAD:.*]] = call { <4 x float>, i1 } @llvm.dx.resource.load.rawbuffer.v4f32.tdx.RawBuffer_i8_0_0_0t(target("dx.RawBuffer", i8, 0, 0, 0) %buffer, i32 %offset, i32 0)
80  ; CHECK: %[[VAL:.*]] = extractvalue { <4 x float>, i1 } %[[LOAD]], 0
81  ; CHECK: call void @v4f32_user(<4 x float> %[[VAL]]
82  %data = load <4 x float>, ptr %ptr
83  call void @v4f32_user(<4 x float> %data)
84
85  ret void
86}
87
88; CHECK-LABEL: define void @loadelements
89define void @loadelements(i32 %index) {
90  %buffer = call target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 0, 0, 0)
91      @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_sl_v4f32v4i32s_0_0_0(
92          i32 0, i32 0, i32 1, i32 0, i1 false)
93
94  ; CHECK-NOT: @llvm.dx.resource.getpointer
95  %ptr = call ptr @llvm.dx.resource.getpointer(
96      target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 0, 0, 0) %buffer,
97      i32 %index)
98
99  ; CHECK: %[[LOADF32:.*]] = call { <4 x float>, i1 } @llvm.dx.resource.load.rawbuffer.v4f32.tdx.RawBuffer_sl_v4f32v4i32s_0_0_0t(target("dx.RawBuffer", { <4 x float>, <4 x i32> }, 0, 0, 0) %buffer, i32 %index, i32 0)
100  ; CHECK: %[[VALF32:.*]] = extractvalue { <4 x float>, i1 } %[[LOADF32]], 0
101  ; CHECK: call void @v4f32_user(<4 x float> %[[VALF32]]
102  %dataf32 = load <4 x float>, ptr %ptr
103  call void @v4f32_user(<4 x float> %dataf32)
104
105  ; CHECK: %[[LOADI32:.*]] = call { <4 x i32>, i1 } @llvm.dx.resource.load.rawbuffer.v4i32.tdx.RawBuffer_sl_v4f32v4i32s_0_0_0t(target("dx.RawBuffer", { <4 x float>, <4 x i32> }, 0, 0, 0) %buffer, i32 %index, i32 16)
106  ; CHECK: %[[VALI32:.*]] = extractvalue { <4 x i32>, i1 } %[[LOADI32]], 0
107  ; CHECK: call void @v4i32_user(<4 x i32> %[[VALI32]]
108  %addri32 = getelementptr inbounds nuw i8, ptr %ptr, i32 16
109  %datai32 = load <4 x i32>, ptr %addri32
110  call void @v4i32_user(<4 x i32> %datai32)
111
112  ret void
113}
114
115; CHECK-LABEL: define void @loadnested
116define void @loadnested(i32 %index) {
117  %buffer = call
118      target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 0, 0, 0)
119      @llvm.dx.resource.handlefrombinding(i32 0, i32 0, i32 1, i32 0, i1 false)
120
121  ; CHECK-NOT: @llvm.dx.resource.getpointer
122  %ptr = call ptr @llvm.dx.resource.getpointer(
123      target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 0, 0, 0) %buffer,
124      i32 %index)
125
126  ; CHECK: %[[LOADI32:.*]] = call { i32, i1 } @llvm.dx.resource.load.rawbuffer.i32.tdx.RawBuffer_sl_i32sl_v4f32v3f16ss_0_0_0t(target("dx.RawBuffer", { i32, { <4 x float>, <3 x half> } }, 0, 0, 0) %buffer, i32 %index, i32 0)
127  ; CHECK: %[[VALI32:.*]] = extractvalue { i32, i1 } %[[LOADI32]], 0
128  ; CHECK: call void @i32_user(i32 %[[VALI32]])
129  %datai32 = load i32, ptr %ptr
130  call void @i32_user(i32 %datai32)
131
132  ; CHECK: %[[LOADF32:.*]] = call { <4 x float>, i1 } @llvm.dx.resource.load.rawbuffer.v4f32.tdx.RawBuffer_sl_i32sl_v4f32v3f16ss_0_0_0t(target("dx.RawBuffer", { i32, { <4 x float>, <3 x half> } }, 0, 0, 0) %buffer, i32 %index, i32 4)
133  ; CHECK: %[[VALF32:.*]] = extractvalue { <4 x float>, i1 } %[[LOADF32]], 0
134  ; CHECK: call void @v4f32_user(<4 x float> %[[VALF32]])
135  %addrf32 = getelementptr inbounds nuw i8, ptr %ptr, i32 4
136  %dataf32 = load <4 x float>, ptr %addrf32
137  call void @v4f32_user(<4 x float> %dataf32)
138
139  ; CHECK: %[[LOADF16:.*]] = call { <3 x half>, i1 } @llvm.dx.resource.load.rawbuffer.v3f16.tdx.RawBuffer_sl_i32sl_v4f32v3f16ss_0_0_0t(target("dx.RawBuffer", { i32, { <4 x float>, <3 x half> } }, 0, 0, 0) %buffer, i32 %index, i32 20)
140  ; CHECK: %[[VALF16:.*]] = extractvalue { <3 x half>, i1 } %[[LOADF16]], 0
141  ; CHECK: call void @v3f16_user(<3 x half> %[[VALF16]])
142  %addrf16 = getelementptr inbounds nuw i8, ptr %ptr, i32 20
143  %dataf16 = load <3 x half>, ptr %addrf16
144  call void @v3f16_user(<3 x half> %dataf16)
145
146  ret void
147}
148
149; byteaddressbuf.Load<int64_t4>
150; CHECK-LABEL: define void @loadv4f64_byte
151define void @loadv4f64_byte(i32 %offset) {
152  %buffer = call target("dx.RawBuffer", i8, 0, 0, 0)
153      @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(
154          i32 0, i32 0, i32 1, i32 0, i1 false)
155
156  ; CHECK-NOT: @llvm.dx.resource.getpointer
157  %ptr = call ptr @llvm.dx.resource.getpointer(
158      target("dx.RawBuffer", i8, 0, 0, 0) %buffer, i32 %offset)
159
160  ; CHECK: %[[LOAD:.*]] = call { <4 x double>, i1 } @llvm.dx.resource.load.rawbuffer.v4f64.tdx.RawBuffer_i8_0_0_0t(target("dx.RawBuffer", i8, 0, 0, 0) %buffer, i32 %offset, i32 0)
161  ; CHECK: %[[VAL:.*]] = extractvalue { <4 x double>, i1 } %[[LOAD]], 0
162  ; CHECK: call void @v4f64_user(<4 x double> %[[VAL]])
163  %data = load <4 x double>, ptr %ptr
164  call void @v4f64_user(<4 x double> %data)
165
166  ret void
167}
168