diff options
Diffstat (limited to 'tools/gfx-unit-test/root-mutable-shader-object.cpp')
| -rw-r--r-- | tools/gfx-unit-test/root-mutable-shader-object.cpp | 90 |
1 files changed, 37 insertions, 53 deletions
diff --git a/tools/gfx-unit-test/root-mutable-shader-object.cpp b/tools/gfx-unit-test/root-mutable-shader-object.cpp index c3bd95382..4c538d730 100644 --- a/tools/gfx-unit-test/root-mutable-shader-object.cpp +++ b/tools/gfx-unit-test/root-mutable-shader-object.cpp @@ -1,21 +1,21 @@ +#if 0 +// Duplicated: This test is identical to slang-rhi\tests\test-mutable-shader-object.cpp +// TODO: This test failed +// The result buffer is still {0.0f, 1.0f, 2.0f, 3.0f}. Not incremented by the shader + #include "core/slang-basic.h" #include "gfx-test-util.h" -#include "gfx-util/shader-cursor.h" -#include "slang-gfx.h" #include "unit-test/slang-unit-test.h" -using namespace gfx; +#include <slang-rhi.h> +#include <slang-rhi/shader-cursor.h> + +using namespace rhi; namespace gfx_test { void mutableRootShaderObjectTestImpl(IDevice* device, UnitTestContext* context) { - Slang::ComPtr<ITransientResourceHeap> transientHeap; - ITransientResourceHeap::Desc transientHeapDesc = {}; - transientHeapDesc.constantBufferSize = 4096; - GFX_CHECK_CALL_ABORT( - device->createTransientResourceHeap(transientHeapDesc, transientHeap.writeRef())); - ComPtr<IShaderProgram> shaderProgram; slang::ProgramLayout* slangReflection; GFX_CHECK_CALL_ABORT(loadComputeProgram( @@ -25,45 +25,34 @@ void mutableRootShaderObjectTestImpl(IDevice* device, UnitTestContext* context) "computeMain", slangReflection)); - ComputePipelineStateDesc pipelineDesc = {}; + ComputePipelineDesc pipelineDesc = {}; pipelineDesc.program = shaderProgram.get(); - ComPtr<gfx::IPipelineState> pipelineState; + ComPtr<IComputePipeline> computePipeline; GFX_CHECK_CALL_ABORT( - device->createComputePipelineState(pipelineDesc, pipelineState.writeRef())); + device->createComputePipeline(pipelineDesc, computePipeline.writeRef())); float initialData[] = {0.0f, 1.0f, 2.0f, 3.0f}; const int numberCount = SLANG_COUNT_OF(initialData); - IBufferResource::Desc bufferDesc = {}; - bufferDesc.sizeInBytes = sizeof(initialData); - bufferDesc.format = gfx::Format::Unknown; + BufferDesc bufferDesc = {}; + bufferDesc.size = sizeof(initialData); + bufferDesc.format = Format::Undefined; bufferDesc.elementSize = sizeof(float); - bufferDesc.allowedStates = ResourceStateSet( - ResourceState::ShaderResource, - ResourceState::UnorderedAccess, - ResourceState::CopyDestination, - ResourceState::CopySource); + bufferDesc.usage = BufferUsage::UnorderedAccess | BufferUsage::ShaderResource | BufferUsage::CopySource | BufferUsage::CopyDestination; bufferDesc.defaultState = ResourceState::UnorderedAccess; bufferDesc.memoryType = MemoryType::DeviceLocal; - ComPtr<IBufferResource> numbersBuffer; + ComPtr<IBuffer> numbersBuffer; GFX_CHECK_CALL_ABORT( - device->createBufferResource(bufferDesc, (void*)initialData, numbersBuffer.writeRef())); - - ComPtr<IResourceView> bufferView; - IResourceView::Desc viewDesc = {}; - viewDesc.type = IResourceView::Type::UnorderedAccess; - viewDesc.format = Format::Unknown; - GFX_CHECK_CALL_ABORT( - device->createBufferView(numbersBuffer, nullptr, viewDesc, bufferView.writeRef())); + device->createBuffer(bufferDesc, (void*)initialData, numbersBuffer.writeRef())); ComPtr<IShaderObject> rootObject; - device->createMutableRootShaderObject(shaderProgram, rootObject.writeRef()); + device->createRootShaderObject(shaderProgram, rootObject.writeRef()); auto entryPointCursor = ShaderCursor(rootObject->getEntryPoint(0)); - entryPointCursor.getPath("buffer").setResource(bufferView); + entryPointCursor.getPath("buffer").setBinding(Binding(numbersBuffer)); slang::TypeReflection* addTransformerType = slangReflection->findTypeByName("AddTransformer"); ComPtr<IShaderObject> transformer; - GFX_CHECK_CALL_ABORT(device->createMutableShaderObject( + GFX_CHECK_CALL_ABORT(device->createShaderObject( addTransformerType, ShaderObjectContainerType::None, transformer.writeRef())); @@ -74,50 +63,45 @@ void mutableRootShaderObjectTestImpl(IDevice* device, UnitTestContext* context) ShaderCursor(transformer).getPath("c").setData(&c, sizeof(float)); { - ICommandQueue::Desc queueDesc = {ICommandQueue::QueueType::Graphics}; - auto queue = device->createCommandQueue(queueDesc); + auto queue = device->getQueue(QueueType::Graphics); - auto commandBuffer = transientHeap->createCommandBuffer(); + auto commandEncoder = queue->createCommandEncoder(); { - auto encoder = commandBuffer->encodeComputeCommands(); - encoder->bindPipelineWithRootObject(pipelineState, rootObject); + auto encoder = commandEncoder->beginComputePass(); + encoder->bindPipeline(computePipeline, rootObject); encoder->dispatchCompute(1, 1, 1); - encoder->endEncoding(); + encoder->end(); } - auto barrierEncoder = commandBuffer->encodeResourceCommands(); - barrierEncoder->bufferBarrier( - 1, - numbersBuffer.readRef(), - ResourceState::UnorderedAccess, - ResourceState::UnorderedAccess); - barrierEncoder->endEncoding(); + // Set buffer state to ensure writes are visible + commandEncoder->setBufferState(numbersBuffer, ResourceState::UnorderedAccess); // Mutate `transformer` object and run again. c = 2.0f; ShaderCursor(transformer).getPath("c").setData(&c, sizeof(float)); { - auto encoder = commandBuffer->encodeComputeCommands(); - encoder->bindPipelineWithRootObject(pipelineState, rootObject); + auto encoder = commandEncoder->beginComputePass(); + encoder->bindPipeline(computePipeline, rootObject); encoder->dispatchCompute(1, 1, 1); - encoder->endEncoding(); + encoder->end(); } - commandBuffer->close(); - queue->executeCommandBuffer(commandBuffer); + queue->submit(commandEncoder->finish()); queue->waitOnHost(); } - compareComputeResult(device, numbersBuffer, Slang::makeArray<float>(3.0f, 4.0f, 5.0f, 6.0f)); + compareComputeResult(device, numbersBuffer, std::array{3.0f, 4.0f, 5.0f, 6.0f}); } SLANG_UNIT_TEST(mutableRootShaderObjectD3D12) { - runTestImpl(mutableRootShaderObjectTestImpl, unitTestContext, Slang::RenderApiFlag::D3D12); + runTestImpl(mutableRootShaderObjectTestImpl, unitTestContext, DeviceType::D3D12, {}); } /*SLANG_UNIT_TEST(mutableRootShaderObjectVulkan) { - runTestImpl(mutableRootShaderObjectTestImpl, unitTestContext, Slang::RenderApiFlag::Vulkan); + runTestImpl(mutableRootShaderObjectTestImpl, unitTestContext, DeviceType::Vulkan, {}); }*/ } // namespace gfx_test + +#endif
\ No newline at end of file |
