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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
//TEST:COMPARE_COMPUTE(filecheck-buffer=CHECK):-vk -output-using-type
//TEST:COMPARE_COMPUTE(filecheck-buffer=CHECK):-cpu -output-using-type
//TEST_INPUT: set outputBuffer = out ubuffer(data=[0 0 0 0], stride=4)
RWStructuredBuffer<float> outputBuffer;
struct Matrix
{
float data[16];
[mutating]
void map(IFunc<float, float> f)
{
for (int i = 0; i < 16; ++i)
{
data[i] = f(data[i]);
}
}
}
struct Applier
{
float c;
void apply(inout Matrix m)
{
m.map((float x) =>
{
if (x > 2.5)
return x * c;
else
return x + c;
});
}
}
[numthreads(1,1,1)]
void computeMain()
{
Matrix m = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16 };
Applier applier = { 1.5 };
applier.apply(m);
outputBuffer[0] = m.data[1];
// CHECK: 3.5
outputBuffer[1] = m.data[3];
// CHECK: 6.0
}
|
