intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2026-01-03 06:49:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Return timestamps for each subdevice via queryTimestampExp

Browse Source 
Signed-off-by: Aravind Gopalakrishnan <aravind.gopalakrishnan@intel.com>
This commit is contained in:
Aravind Gopalakrishnan
2021-09-09 07:59:45 +00:00
committed by Compute-Runtime-Automation
parent 7c1ea18f74
commit 6061b61aea
2 changed files with 56 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
level_zero/core/source/event/event_impl.inl
View File

@@ -261,23 +261,33 @@ ze_result_t EventImp<TagSizeT>::queryTimestampsExp(Device *device, uint32_t *pCo
isStaticPartitioning = false;
}
if (deviceImp->isSubdevice || !isStaticPartitioning) {
if (!isStaticPartitioning) {
return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
}
uint32_t numPacketsUsed = 1u;
if (!deviceImp->isSubdevice) {
numPacketsUsed = this->getPacketsInUse();
}
if ((*pCount == 0) ||
(*pCount > kernelTimestampsData[timestampPacket].getPacketsUsed())) {
*pCount = this->getPacketsInUse();
(*pCount > numPacketsUsed)) {
*pCount = numPacketsUsed;
return ZE_RESULT_SUCCESS;
}
for (auto packetId = 0u; packetId < *pCount; packetId++) {
ze_kernel_timestamp_result_t &result = *(pTimestamps + packetId);
for (auto i = 0u; i < *pCount; i++) {
ze_kernel_timestamp_result_t &result = *(pTimestamps + i);
auto queryTsEventAssignFunc = [&](uint64_t &timestampFieldForWriting, uint64_t &timestampFieldToCopy) {
memcpy_s(&timestampFieldForWriting, sizeof(uint64_t), static_cast<void *>(&timestampFieldToCopy), sizeof(uint64_t));
};
auto packetId = i;
if (deviceImp->isSubdevice) {
packetId = static_cast<NEO::SubDevice *>(deviceImp->neoDevice)->getSubDeviceIndex();
}
globalStartTs = kernelTimestampsData[timestampPacket].getGlobalStartValue(packetId);
contextStartTs = kernelTimestampsData[timestampPacket].getContextStartValue(packetId);
contextEndTs = kernelTimestampsData[timestampPacket].getContextEndValue(packetId);