blob: b5bdb455011d79f1a1b6a3464438d1152d36a394 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
//TEST:SIMPLE(filecheck=CHECK):-target spirv
struct Data {
StructuredBuffer<float> input[2];
RWStructuredBuffer<float> output;
uint input_tensor_count;
StructuredBuffer<uint> index_buffer;
uint index_count;
[mutating]
float fetch(int buffer, int index)
{
return input[buffer][index];
}
};
[shader("compute")]
[numthreads(8, 8, 1)]
void compute_main(uint3 tid: SV_DispatchThreadID, ParameterBlock<Data> data)
{
float result = 0.0;
for (int i = 0; i < data.index_count; ++i) {
uint buffer = data.index_buffer[i];
//CHECK: ([[# @LINE+1]]): error
result += data.fetch(buffer, tid.x * 1024 + tid.y);
}
data.output[tid.x * 1024 + tid.y] = result;
}
|
