gfx: Add `resolveQuery` command. (#2125)

Co-authored-by: Yong He <yhe@nvidia.com>

1 files changed, 21 insertions, 0 deletions

diff --git a/tools/gfx/vulkan/render-vk.cpp b/tools/gfx/vulkan/render-vk.cpp
index 80084ae7b..d3f0329d8 100644
--- a/tools/gfx/vulkan/render-vk.cpp
+++ b/tools/gfx/vulkan/render-vk.cpp
@@ -4876,6 +4876,27 @@ public:
                 }
             }
 
+            virtual SLANG_NO_THROW void SLANG_MCALL resolveQuery(
+                IQueryPool* queryPool,
+                uint32_t index,
+                uint32_t count,
+                IBufferResource* buffer,
+                uint64_t offset) override
+            {
+                auto& vkApi = m_commandBuffer->m_renderer->m_api;
+                auto poolImpl = static_cast<QueryPoolImpl*>(queryPool);
+                auto bufferImpl = static_cast<BufferResourceImpl*>(buffer);
+                vkApi.vkCmdCopyQueryPoolResults(
+                    m_commandBuffer->m_commandBuffer,
+                    poolImpl->m_pool,
+                    index,
+                    count,
+                    bufferImpl->m_buffer.m_buffer,
+                    offset,
+                    sizeof(uint64_t),
+                    VK_QUERY_RESULT_64_BIT | VK_QUERY_RESULT_WAIT_BIT);
+            }
+
             virtual SLANG_NO_THROW void SLANG_MCALL copyTextureToBuffer(
                 IBufferResource* dst,
                 size_t dstOffset,