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
|
//TEST:SIMPLE(filecheck=CHECK): -target metal
//TEST:SIMPLE(filecheck=CHECK-ASM): -target metallib
//TEST:REFLECTION(filecheck=REFLECT):-target metal -entry main_kernel -stage compute
uniform RWStructuredBuffer<float> outputBuffer;
struct MyBlock
{
StructuredBuffer<float> b1;
StructuredBuffer<float> b2;
}
ParameterBlock<MyBlock> block;
ParameterBlock<MyBlock> block2;
// CHECK: {{\[\[}}kernel{{\]\]}} void main_kernel(float device* {{.*}} {{\[\[}}buffer(0){{\]\]}}, MyBlock{{.*}} constant* block{{.*}} {{\[\[}}buffer(1){{\]\]}}, MyBlock{{.*}} constant* block2{{.*}} {{\[\[}}buffer(2){{\]\]}})
// CHECK-ASM: define void @main_kernel
// REFLECT: "elementVarLayout": {
// REFLECT: "name": "b1",
// REFLECT: "binding": {"kind": "uniform", "offset": 0, "size": 8, "elementStride": 0}
// REFLECT: "name": "b2",
// REFLECT: "binding": {"kind": "uniform", "offset": 8, "size": 8, "elementStride": 0}
// REFLECT: "binding": {"kind": "metalArgumentBufferElement", "index": 0, "count": 2}
// REFLECT: "name": "outputBuffer",
// REFLECT-NEXT: "binding": {"kind": "constantBuffer", "index": 0{{.*}}}
// REFLECT: "name": "block",
// REFLECT-NEXT: "binding": {"kind": "constantBuffer", "index": 1{{.*}}}
void func(float v)
{
outputBuffer[0] = v;
outputBuffer[1] = outputBuffer.Load(0);
outputBuffer[2] = block.b1[0] + block2.b2[0];
}
[numthreads(1,1,1)]
void main_kernel()
{
func(3.0f);
}
|
