forked from OSchip/llvm-project
[mlir][vulkan-runner] Minor fix in timestamp flag for vulkan runner.
The first timestamp query should use VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT. Differential Revision: https://reviews.llvm.org/D80183
This commit is contained in:
parent
a45fb1942f
commit
92577e6a0f
|
@ -728,7 +728,7 @@ LogicalResult VulkanRuntime::createComputeCommandBuffer() {
|
|||
descriptorSets.data(), 0, 0);
|
||||
// Get a timestamp before invoking the compute shader.
|
||||
if (queryPool != VK_NULL_HANDLE)
|
||||
vkCmdWriteTimestamp(commandBuffer, VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,
|
||||
vkCmdWriteTimestamp(commandBuffer, VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT,
|
||||
queryPool, 0);
|
||||
vkCmdDispatch(commandBuffer, numWorkGroups.x, numWorkGroups.y,
|
||||
numWorkGroups.z);
|
||||
|
|
Loading…
Reference in New Issue