yum-mirror/slang

Making it easier to work with shaders

git clone https://git.yummers.dev/yum-mirror/slang

Gangzheng TongConvert gfx unit tests and examples to use slang-rhi (#7577)43d0c2100

master
branchesmasterall refs →
2.5 KiB72 linesraw
1// Duplicated: This this test is identical to slang-rhi\tests\test-uint16-buffer.cpp
2
3#include "core/slang-basic.h"
4#include "gfx-test-util.h"
5#include "slang-rhi.h"
6#include "slang-rhi/shader-cursor.h"
7#include "unit-test/slang-unit-test.h"
8
9using namespace rhi;
10
11namespace gfx_test
12{
13void uint16BufferTestImpl(IDevice* device, UnitTestContext* context)
14{
15    ComPtr<IShaderProgram> shaderProgram;
16    slang::ProgramLayout* slangReflection;
17    GFX_CHECK_CALL_ABORT(
18        loadComputeProgram(device, shaderProgram, "uint16-buffer", "computeMain", slangReflection));
19
20    ComputePipelineDesc pipelineDesc = {};
21    pipelineDesc.program = shaderProgram.get();
22    ComPtr<IComputePipeline> pipelineState;
23    GFX_CHECK_CALL_ABORT(device->createComputePipeline(pipelineDesc, pipelineState.writeRef()));
24
25    const int numberCount = 4;
26    uint16_t initialData[] = {0, 1, 2, 3};
27    BufferDesc bufferDesc = {};
28    bufferDesc.size = numberCount * sizeof(uint16_t);
29    bufferDesc.format = rhi::Format::Undefined;
30
31    bufferDesc.elementSize = 0; // Let RHI derive from reflection
32    bufferDesc.usage = BufferUsage::ShaderResource | BufferUsage::UnorderedAccess |
33                       BufferUsage::CopyDestination | BufferUsage::CopySource;
34    bufferDesc.defaultState = ResourceState::UnorderedAccess;
35    bufferDesc.memoryType = MemoryType::DeviceLocal;
36
37    ComPtr<IBuffer> numbersBuffer;
38    GFX_CHECK_CALL_ABORT(
39        device->createBuffer(bufferDesc, (void*)initialData, numbersBuffer.writeRef()));
40
41    // We have done all the set up work, now it is time to start recording a command buffer for
42    // GPU execution.
43    {
44        auto queue = device->getQueue(QueueType::Graphics);
45        auto commandEncoder = queue->createCommandEncoder();
46        auto computePassEncoder = commandEncoder->beginComputePass();
47
48        auto rootObject = computePassEncoder->bindPipeline(pipelineState);
49
50        // Bind buffer to the entry point.
51        ShaderCursor(rootObject).getPath("buffer").setBinding(Binding(numbersBuffer));
52
53        computePassEncoder->dispatchCompute(1, 1, 1);
54        computePassEncoder->end();
55        queue->submit(commandEncoder->finish());
56        queue->waitOnHost();
57    }
58
59    compareComputeResult(device, numbersBuffer, std::array<uint16_t, 4>{1, 2, 3, 4});
60}
61
62SLANG_UNIT_TEST(uint16BufferTestD3D12)
63{
64    runTestImpl(uint16BufferTestImpl, unitTestContext, DeviceType::D3D12);
65}
66
67SLANG_UNIT_TEST(uint16BufferTestVulkan)
68{
69    runTestImpl(uint16BufferTestImpl, unitTestContext, DeviceType::Vulkan);
70}
71
72} // namespace gfx_test