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
|
// Duplicated: This test is identical slang-rhi\tests\test-root-shader-parameter.cpp
#include "core/slang-basic.h"
#include "gfx-test-util.h"
#include "slang-rhi.h"
#include "slang-rhi/shader-cursor.h"
#include "unit-test/slang-unit-test.h"
using namespace rhi;
namespace gfx_test
{
static ComPtr<IBuffer> createBuffer(IDevice* device, uint32_t content)
{
ComPtr<IBuffer> buffer;
BufferDesc bufferDesc = {};
bufferDesc.size = sizeof(uint32_t);
bufferDesc.format = rhi::Format::Undefined;
bufferDesc.elementSize = sizeof(float);
bufferDesc.usage = BufferUsage::ShaderResource | BufferUsage::UnorderedAccess |
BufferUsage::CopyDestination | BufferUsage::CopySource;
bufferDesc.defaultState = ResourceState::UnorderedAccess;
bufferDesc.memoryType = MemoryType::DeviceLocal;
ComPtr<IBuffer> numbersBuffer;
GFX_CHECK_CALL_ABORT(device->createBuffer(bufferDesc, (void*)&content, buffer.writeRef()));
return buffer;
}
void rootShaderParameterTestImpl(IDevice* device, UnitTestContext* context)
{
if (!device->hasFeature(Feature::ParameterBlock))
{
SLANG_CHECK("no support for parameter blocks");
}
ComPtr<IShaderProgram> shaderProgram;
slang::ProgramLayout* slangReflection;
GFX_CHECK_CALL_ABORT(loadComputeProgram(
device,
shaderProgram,
"root-shader-parameter",
"computeMain",
slangReflection));
ComputePipelineDesc pipelineDesc = {};
pipelineDesc.program = shaderProgram.get();
ComPtr<rhi::IComputePipeline> pipeline = device->createComputePipeline(pipelineDesc);
Slang::List<ComPtr<IBuffer>> buffers;
for (uint32_t i = 0; i < 9; i++)
{
buffers.add(createBuffer(device, i == 0 ? 10 : i));
}
ComPtr<IShaderObject> rootObject;
device->createRootShaderObject(shaderProgram, rootObject.writeRef());
ComPtr<IShaderObject> g, s1, s2;
device->createShaderObject(
slangReflection->findTypeByName("S0"),
ShaderObjectContainerType::None,
g.writeRef());
device->createShaderObject(
slangReflection->findTypeByName("S1"),
ShaderObjectContainerType::None,
s1.writeRef());
device->createShaderObject(
slangReflection->findTypeByName("S1"),
ShaderObjectContainerType::None,
s2.writeRef());
{
auto cursor = ShaderCursor(s1);
cursor["c0"].setBinding(buffers[2]);
cursor["c1"].setBinding(buffers[3]);
cursor["c2"].setBinding(buffers[4]);
}
{
auto cursor = ShaderCursor(s2);
cursor["c0"].setBinding(buffers[5]);
cursor["c1"].setBinding(buffers[6]);
cursor["c2"].setBinding(buffers[7]);
}
{
auto cursor = ShaderCursor(g);
cursor["b0"].setBinding(buffers[0]);
cursor["b1"].setBinding(buffers[1]);
cursor["s1"].setObject(s1);
cursor["s2"].setObject(s2);
}
{
auto cursor = ShaderCursor(rootObject);
cursor["g"].setObject(g);
cursor["buffer"].setBinding(buffers[8]);
}
{
auto queue = device->getQueue(QueueType::Graphics);
auto commandBuffer = queue->createCommandEncoder();
{
auto encoder = commandBuffer->beginComputePass();
encoder->bindPipeline(pipeline, rootObject);
encoder->dispatchCompute(1, 1, 1);
encoder->end();
}
queue->submit(commandBuffer->finish());
queue->waitOnHost();
}
compareComputeResult(device, buffers[8], std::array{10 - 1 + 2 - 3 + 4 + 5 - 6 + 7});
}
SLANG_UNIT_TEST(rootShaderParameterD3D12)
{
runTestImpl(rootShaderParameterTestImpl, unitTestContext, DeviceType::D3D12);
}
SLANG_UNIT_TEST(rootShaderParameterVulkan)
{
runTestImpl(rootShaderParameterTestImpl, unitTestContext, DeviceType::Vulkan);
}
} // namespace gfx_test
|
