[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:
Thomas Raoux 2020-05-19 18:56:48 -07:00
parent a45fb1942f
commit 92577e6a0f
1 changed files with 1 additions and 1 deletions

View File

@ -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);