Skip to content

Commit

Permalink
more bench
Browse files Browse the repository at this point in the history
  • Loading branch information
luitjens committed Aug 25, 2022
1 parent a66b222 commit c3c006f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions bench/00_transform/conv.cu
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
using namespace matx;

using conv_types =
nvbench::type_list<cuda::std::complex<float>, cuda::std::complex<double>>;
nvbench::type_list<cuda::std::complex<float>, cuda::std::complex<double>, float, double>;

/* FFT benchmarks */
template <typename ValueType>
Expand Down Expand Up @@ -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));
NVBENCH_BENCH_TYPES(conv1d_2d_batch, NVBENCH_TYPE_AXES(conv_types));

0 comments on commit c3c006f

Please sign in to comment.