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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
|
//TEST:SIMPLE(filecheck=SAMPLER): -target spirv -stage compute -entry computeMain -DSAMPLER
//TEST:SIMPLE(filecheck=COMBINED_IMAGE_SAMPLER): -target spirv -stage compute -entry computeMain -DCOMBINED_IMAGE_SAMPLER
//TEST:SIMPLE(filecheck=SAMPLED_IMAGE): -target spirv -stage compute -entry computeMain -DSAMPLED_IMAGE
//TEST:SIMPLE(filecheck=STORAGE_IMAGE): -target spirv -stage compute -entry computeMain -DSTORAGE_IMAGE
//TEST:SIMPLE(filecheck=UNIFORM_TEXEL_BUFFER): -target spirv -stage compute -entry computeMain -DUNIFORM_TEXEL_BUFFER
//TEST:SIMPLE(filecheck=STORAGE_TEXEL_BUFFER): -target spirv -stage compute -entry computeMain -DSTORAGE_TEXEL_BUFFER
//TEST:SIMPLE(filecheck=UNIFORM_BUFFER): -target spirv -stage compute -entry computeMain -DUNIFORM_BUFFER
//TEST:SIMPLE(filecheck=STORAGE_BUFFER): -target spirv -stage compute -entry computeMain -DSTORAGE_BUFFER
//TEST:SIMPLE(filecheck=ACCELERATION_STRUCTURE): -target spirv -stage compute -entry computeMain -DACCELERATION_STRUCTURE
//TEST:SIMPLE(filecheck=SAMPLED_IMAGE): -target spirv -stage compute -entry computeMain -DSAMPLED_IMAGE
//TEST:SIMPLE(filecheck=MIX): -target spirv -stage compute -entry computeMain -DSAMPLER -DSTORAGE_TEXEL_BUFFER -DUNIFORM_BUFFER -DACCELERATION_STRUCTURE
export T getDescriptorFromHandle<T : IOpaqueDescriptor>(DescriptorHandle<T> handleValue)
{
return defaultGetDescriptorFromHandle(handleValue, BindlessDescriptorOptions.None);
}
// To intentionally fill up binding slots
[[vk::binding(0, 1)]]
RWTexture1D<float> t1;
[[vk::binding(0, 2)]]
RWTexture1D<float> t2;
[[vk::binding(0, 4)]]
RWTexture1D<float> t3;
[[vk::binding(1, 4)]]
Texture1D<float> t4;
//MIX-DAG: OpDecorate %__slang_resource_heap{{.*}} Binding 0
//MIX-DAG: OpDecorate %__slang_resource_heap{{.*}} Binding 5
//MIX-DAG: OpDecorate %__slang_resource_heap{{.*}} Binding 6
//MIX-DAG: OpDecorate %__slang_resource_heap{{.*}} Binding 7
//MIX-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 1
//MIX-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 2
//MIX-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 3
//MIX-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 4
//MIX-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 8
#ifdef SAMPLER
//SAMPLER: OpDecorate %__slang_resource_heap{{.*}} Binding 0
//SAMPLER-NEXT: OpDecorate %__slang_resource_heap{{.*}} DescriptorSet 3
//SAMPLER-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 1
//SAMPLER-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 2
//SAMPLER-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 3
//SAMPLER-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 4
//SAMPLER-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 5
//SAMPLER-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 6
//SAMPLER-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 7
//SAMPLER-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 8
uniform SamplerState.Handle sampler;
#endif
#ifdef COMBINED_IMAGE_SAMPLER
//COMBINED_IMAGE_SAMPLER: OpDecorate %__slang_resource_heap{{.*}} Binding 1
//COMBINED_IMAGE_SAMPLER-NEXT: OpDecorate %__slang_resource_heap{{.*}} DescriptorSet 3
uniform Sampler1DShadow.Handle combinedSampler;
#endif
#ifdef SAMPLED_IMAGE
//SAMPLED_IMAGE: OpDecorate %__slang_resource_heap{{.*}} Binding 2
//SAMPLED_IMAGE-NEXT: OpDecorate %__slang_resource_heap{{.*}} DescriptorSet 3
uniform Texture1D<float>.Handle texture;
#endif
#ifdef STORAGE_IMAGE
//STORAGE_IMAGE: OpDecorate %__slang_resource_heap{{.*}} Binding 3
//STORAGE_IMAGE-NEXT: OpDecorate %__slang_resource_heap{{.*}} DescriptorSet 3
uniform RWTexture1D<float>.Handle rwTexture1;
uniform RWTexture2D<float>.Handle rwTexture2;
#endif
#ifdef UNIFORM_TEXEL_BUFFER
//UNIFORM_TEXEL_BUFFER: OpDecorate %__slang_resource_heap{{.*}} Binding 4
//UNIFORM_TEXEL_BUFFER-NEXT: OpDecorate %__slang_resource_heap{{.*}} DescriptorSet 3
uniform Buffer<float>.Handle texelBuffer;
#endif
#ifdef STORAGE_TEXEL_BUFFER
//STORAGE_TEXEL_BUFFER: OpDecorate %__slang_resource_heap{{.*}} Binding 5
//STORAGE_TEXEL_BUFFER-NEXT: OpDecorate %__slang_resource_heap{{.*}} DescriptorSet 3
uniform RWBuffer<float>.Handle rwTexelBuffer;
#endif
#ifdef UNIFORM_BUFFER
//UNIFORM_BUFFER: OpDecorate %__slang_resource_heap{{.*}} Binding 6
//UNIFORM_BUFFER-NEXT: OpDecorate %__slang_resource_heap{{.*}} DescriptorSet 3
//UNIFORM_BUFFER: OpDecorate %__slang_resource_heap{{.*}} Binding 7
//UNIFORM_BUFFER-NEXT: OpDecorate %__slang_resource_heap{{.*}} DescriptorSet 3
struct Data
{
float v;
}
uniform ConstantBuffer<Data>.Handle buffer1;
uniform StructuredBuffer<float>.Handle buffer2;
#endif
#ifdef STORAGE_BUFFER
//STORAGE_BUFFER: OpDecorate %__slang_resource_heap{{.*}} Binding 7
//STORAGE_BUFFER-NEXT: OpDecorate %__slang_resource_heap{{.*}} DescriptorSet 3
uniform RWStructuredBuffer<float>.Handle rwBuffer;
#endif
#ifdef ACCELERATION_STRUCTURE
//ACCELERATION_STRUCTURE-NOT: OpDecorate %__slang_resource_heap{{.*}} Binding 8
//ACCELERATION_STRUCTURE: OpConvertUToAccelerationStructureKHR
uniform RaytracingAccelerationStructure.Handle rayAcceleration;
#endif
[numthreads(2,2,1)]
[shader("compute")]
void computeMain()
{
t1[0] = t2[0] + t2[0] + t4[0];
#ifdef SAMPLER
t1[2] = t4.Sample(sampler, 0);
#endif
#ifdef COMBINED_IMAGE_SAMPLER
t1[8] = combinedSampler.Sample(0);
#endif
#ifdef SAMPLED_IMAGE
t1[0] = texture[0];
#endif
#ifdef STORAGE_IMAGE
t1[11] = rwTexture1[0];
t1[12] = rwTexture2[0];
#endif
#ifdef UNIFORM_TEXEL_BUFFER
t1[10] = texelBuffer[0];
#endif
#ifdef STORAGE_TEXEL_BUFFER
t1[9] = rwTexelBuffer[0];
#endif
#ifdef UNIFORM_BUFFER
t1[4] = (*buffer1).v;
t1[6] = buffer2[0];
#endif
#ifdef STORAGE_BUFFER
t1[0] += rwBuffer[0];
#endif
#ifdef ACCELERATION_STRUCTURE
RayDesc ray;
ray.Origin = float3(0.1f, 0.1f, 0.0f);
ray.Direction = float3(0.0f, 0.0f, 1.0f);
ray.TMin = 0.0f;
ray.TMax = 100.0f;
RayQuery<RAY_FLAG_FORCE_NON_OPAQUE> rq;
rq.TraceRayInline(rayAcceleration, RAY_FLAG_FORCE_NON_OPAQUE, 0xff, ray);
bool proceed = rq.Proceed();
rq.CommitNonOpaqueTriangleHit();
rq.Abort();
t1[13] = (float)rq.RayFlags();
#endif
}
|
