From c3c006fd2d2271d5f5a9241ae3d3a451e4bf9f93 Mon Sep 17 00:00:00 2001 From: jluitjens Date: Thu, 25 Aug 2022 14:06:38 -0700 Subject: [PATCH] more bench --- bench/00_transform/conv.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bench/00_transform/conv.cu b/bench/00_transform/conv.cu index 48518cdb2..fcf6a42b6 100644 --- a/bench/00_transform/conv.cu +++ b/bench/00_transform/conv.cu @@ -4,7 +4,7 @@ using namespace matx; using conv_types = - nvbench::type_list, cuda::std::complex>; + nvbench::type_list, cuda::std::complex, float, double>; /* FFT benchmarks */ template @@ -44,4 +44,4 @@ void conv1d_2d_batch(nvbench::state &state, state.exec( [&out, &at, &bt](nvbench::launch &launch) { conv1d(out, at, bt, MATX_C_MODE_FULL, launch.get_stream()); }); } -NVBENCH_BENCH_TYPES(conv1d_2d_batch, NVBENCH_TYPE_AXES(conv_types)); \ No newline at end of file +NVBENCH_BENCH_TYPES(conv1d_2d_batch, NVBENCH_TYPE_AXES(conv_types));