NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-03-14 20:27:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change CPU timer start point.

Browse Source
This commit is contained in:
Allison Vacanti
2021-04-10 20:00:48 -04:00
parent 34479b3116
commit 4ab0f5de7c
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
nvbench/detail/measure_cold.cuh
View File

@@ -133,7 +133,10 @@ struct measure_cold_base::kernel_launch_timer
m_measure.block_stream();
}
m_measure.m_cuda_timer.start(m_measure.m_launch.get_stream());
m_measure.m_cpu_timer.start();
if constexpr (!use_blocking_kernel)
{
m_measure.m_cpu_timer.start();
}
}
__forceinline__ void stop()
@@ -141,6 +144,7 @@ struct measure_cold_base::kernel_launch_timer
m_measure.m_cuda_timer.stop(m_measure.m_launch.get_stream());
if constexpr (use_blocking_kernel)
{
m_measure.m_cpu_timer.start();
m_measure.unblock_stream();
}
m_measure.sync_stream();