[Mlir-commits] [mlir] 92577e6 - [mlir][vulkan-runner] Minor fix in timestamp flag for vulkan runner.

[Thomas Raoux]

Author: Thomas Raoux
Date: 2020-05-19T18:56:48-07:00
New Revision: 92577e6a0f7380f91387c04106b8aed0fe40cc8b

URL: https://github.com/llvm/llvm-project/commit/92577e6a0f7380f91387c04106b8aed0fe40cc8b
DIFF: https://github.com/llvm/llvm-project/commit/92577e6a0f7380f91387c04106b8aed0fe40cc8b.diff

LOG: [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

Added: 
    

Modified: 
    mlir/tools/mlir-vulkan-runner/VulkanRuntime.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/tools/mlir-vulkan-runner/VulkanRuntime.cpp b/mlir/tools/mlir-vulkan-runner/VulkanRuntime.cpp
index 7c3dcff545ce..2c42c19badef 100644
--- a/mlir/tools/mlir-vulkan-runner/VulkanRuntime.cpp
+++ b/mlir/tools/mlir-vulkan-runner/VulkanRuntime.cpp
@@ -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);