wehub-resource-sync
42 lines
1.4 KiB
C++
42 lines
1.4 KiB
C++
//
|
|
// OpenCLRunningUtils.cpp
|
|
// MNN
|
|
//
|
|
// Created by MNN on 2023/10/27.
|
|
// Copyright © 2018, Alibaba Group Holding Limited
|
|
//
|
|
|
|
#include "VulkanQueryPool.hpp"
|
|
|
|
namespace MNN {
|
|
VulkanQueryPool::VulkanQueryPool(const VulkanDevice& dev) : mDevice(dev){
|
|
VkQueryPoolCreateInfo queryPoolCreateInfo = {};
|
|
queryPoolCreateInfo.sType = VK_STRUCTURE_TYPE_QUERY_POOL_CREATE_INFO;
|
|
queryPoolCreateInfo.queryType = VK_QUERY_TYPE_TIMESTAMP;
|
|
queryPoolCreateInfo.queryCount = 2;
|
|
|
|
CALL_VK(vkCreateQueryPool(mDevice.get(), &queryPoolCreateInfo, nullptr, &queryPool));
|
|
}
|
|
|
|
VulkanQueryPool::~VulkanQueryPool(){
|
|
vkDestroyQueryPool(mDevice.get(), queryPool, nullptr);
|
|
}
|
|
|
|
void VulkanQueryPool::VulkanCmdResetQueryPool(VkCommandBuffer commandBuffer){
|
|
vkCmdResetQueryPool(commandBuffer, queryPool, 0, 2);
|
|
}
|
|
|
|
void VulkanQueryPool::VulkanCmdWriteTimestamp(VkCommandBuffer commandBuffer, int index){
|
|
vkCmdWriteTimestamp(commandBuffer, VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT, queryPool, index);
|
|
}
|
|
|
|
float VulkanQueryPool::VulkanGetQueryPoolResults(){
|
|
uint64_t timestamps[2];
|
|
CALL_VK(vkGetQueryPoolResults(mDevice.get(), queryPool, 0, 2, sizeof(uint64_t) * 2, timestamps, sizeof(uint64_t), VK_QUERY_RESULT_64_BIT | VK_QUERY_RESULT_WAIT_BIT));
|
|
|
|
double timestampPeriod = mDevice.getTimestampPeriod();
|
|
double executionTime = (timestamps[1] - timestamps[0]) * timestampPeriod / double(1e6); // us
|
|
return executionTime;
|
|
}
|
|
} // namespace MNN
|