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
48
49
50
51
52
53
54
55
56
57
58
59
|
//TEST:SIMPLE(filecheck=CHECK): -stage compute -entry computeMain -target glsl
//TEST:REFLECTION: -stage compute -entry computeMain -target glsl
// CHECK-DAG: layout(binding = 0, set = 1)
// CHECK-DAG: texture2D g_TextureHeap_1[]
// CHECK-DAG: layout(binding = 0, set = 1)
// CHECK-DAG: utexture2D g_TextureHeap_0[]
[vk::binding(0, 1)]
__DynamicResource<__DynamicResourceKind.General> g_TextureHeap[];
// CHECK-DAG: layout(binding = 2, set = 1)
// CHECK-DAG: sampler g_SamplerHeap_0[]
[vk::binding(2, 1)]
__DynamicResource<__DynamicResourceKind.Sampler> g_SamplerHeap[];
// CHECK-DAG: layout(binding = 0, set = 2)
// CHECK-DAG: texture2D g_SingleDynamicResourceA_0;
[vk::binding(0, 2)]
__DynamicResource g_SingleDynamicResourceA;
// CHECK-DAG: layout(binding = 1, set = 2)
// CHECK-DAG: texture2D g_SingleDynamicResourceB_0;
[vk::binding(1, 2)]
__DynamicResource g_SingleDynamicResourceB;
// CHECK-DAG: layout(std430, binding = 2, set = 2) buffer
[vk::binding(2, 2)]
__DynamicResource g_SingleDynamicResourceC;
// CHECK: IndirectAccess_0
// CHECK: g_TextureHeap_0[_S1]
[noinline]
float4 IndirectAccess(__DynamicResource resource, int2 pos)
{
let texture = resource.as<Texture2D<uint4>>();
return texture[pos] / 255.0;
}
// CHECK: main
// CHECK-DAG: g_TextureHeap_1[3]
// CHECK-DAG: g_SamplerHeap_0[2]
[numthreads(16)]
void computeMain(int2 tid: SV_DispatchThreadID)
{
Texture2D texture = g_TextureHeap[3];
SamplerState sampler = g_SamplerHeap[2].as<SamplerState>();
float4 value = texture.SampleLevel(sampler, tid / 64.0, 0.0);
value += IndirectAccess(g_TextureHeap[1], tid);
value += g_SingleDynamicResourceA.as<Texture2D>()[tid];
value += g_SingleDynamicResourceB.as<Texture2D>()[tid];
RWStructuredBuffer<float4> outputBuffer = g_SingleDynamicResourceC;
outputBuffer[tid.x + tid.y * 64] = value;
}
|
