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
|
#include "slang-ir-lower-dynamic-resource-heap.h"
#include "compiler-core/slang-artifact-associated-impl.h"
#include "slang-ir-util.h"
namespace Slang
{
UInt findUnusedSpaceIndex(TargetProgram* targetProgram, IRModule* module)
{
HashSet<int> usedSpaces;
auto processVarLayout = [&](IRVarLayout* varLayout)
{
UInt spaceOffset = 0;
if (auto spaceAttr = varLayout->findOffsetAttr(LayoutResourceKind::SubElementRegisterSpace))
{
spaceOffset = spaceAttr->getOffset();
}
for (auto sizeAttr : varLayout->getTypeLayout()->getSizeAttrs())
{
auto kind = sizeAttr->getResourceKind();
if (!ShaderBindingRange::isUsageTracked(kind))
continue;
if (auto offsetAttr = varLayout->findOffsetAttr(kind))
{
// Get the binding information from this attribute and insert it into the list
auto spaceIndex = spaceOffset + offsetAttr->getSpace();
usedSpaces.add((int)spaceIndex);
}
}
};
for (auto inst : module->getGlobalInsts())
{
if (as<IRGlobalParam>(inst))
{
auto varLayout = findVarLayout(inst);
if (!varLayout)
continue;
processVarLayout(varLayout);
auto paramGroupTypeLayout = as<IRParameterGroupTypeLayout>(varLayout->getTypeLayout());
if (!paramGroupTypeLayout)
continue;
auto containerVarLayout = paramGroupTypeLayout->getContainerVarLayout();
if (!containerVarLayout)
continue;
processVarLayout(containerVarLayout);
}
}
// Find next unused space index.
int index = targetProgram->getOptionSet().getIntOption(CompilerOptionName::BindlessSpaceIndex);
while (usedSpaces.contains(index))
{
index++;
}
return index;
}
IRVarLayout* createResourceHeapVarLayoutWithSpaceAndBinding(
IRBuilder& builder,
IRInst* param,
UInt spaceIndex,
UInt bindingIndex)
{
SLANG_UNUSED(param);
IRTypeLayout::Builder typeLayoutBuilder(&builder);
typeLayoutBuilder.addResourceUsage(
LayoutResourceKind::DescriptorTableSlot,
LayoutSize::infinite());
auto typeLayout = typeLayoutBuilder.build();
IRVarLayout::Builder varLayoutBuilder(&builder, typeLayout);
varLayoutBuilder.findOrAddResourceInfo(LayoutResourceKind::RegisterSpace)->offset = spaceIndex;
varLayoutBuilder.findOrAddResourceInfo(LayoutResourceKind::DescriptorTableSlot)->offset =
bindingIndex;
return varLayoutBuilder.build();
}
void lowerDynamicResourceHeap(TargetProgram* targetProgram, IRModule* module, DiagnosticSink* sink)
{
SLANG_UNUSED(sink);
auto unusedSpaceIndex = findUnusedSpaceIndex(targetProgram, module);
List<IRInst*> workList;
for (auto globalInst : module->getGlobalInsts())
{
if (globalInst->getOp() == kIROp_GetDynamicResourceHeap)
{
workList.add(globalInst);
}
}
for (auto inst : workList)
{
auto arrayType = as<IRArrayTypeBase>(inst->getDataType());
IRBuilder builder(inst);
builder.setInsertBefore(inst);
auto bindingIndex = (UInt)as<IRIntLit>(inst->getOperand(0))->getValue();
auto param = builder.createGlobalParam(arrayType);
auto varLayout = createResourceHeapVarLayoutWithSpaceAndBinding(
builder,
param,
unusedSpaceIndex,
bindingIndex);
builder.addLayoutDecoration(param, varLayout);
builder.addNameHintDecoration(param, toSlice("__slang_resource_heap"));
inst->replaceUsesWith(param);
}
}
} // namespace Slang
|
