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

Make kernel wrapper into an lvalue.

Browse Source
This commit is contained in:
Allison Vacanti
2021-02-18 23:41:23 -05:00
parent b5443e98c8
commit 543488ef75
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
nvbench/detail/state_exec.cuh
View File

@@ -20,7 +20,7 @@ namespace nvbench
template <typename ExecTags, typename KernelLauncher>
void state::exec(ExecTags tags, KernelLauncher &&kernel_launcher)
{
using KL = std::remove_reference_t<KernelLauncher>;
using KL = typename std::remove_reference<KernelLauncher>::type;
using namespace nvbench::exec_tag::impl;
static_assert(is_exec_tag_v<ExecTags>,
"`ExecTags` argument must be a member (or combination of "
@@ -68,7 +68,8 @@ void state::exec(ExecTags tags, KernelLauncher &&kernel_launcher)
using wrapper_t = nvbench::detail::kernel_launch_timer_wrapper<KL>;
using measure_t =
nvbench::detail::measure_cold<wrapper_t, use_blocking_kernel>;
measure_t measure(*this, wrapper_t{kernel_launcher});
wrapper_t wrapper{kernel_launcher};
measure_t measure(*this, wrapper);
measure();
}
}