Fix encode bench

gevtushenko · Jun 20, 2023 · f751105 · f751105
1 parent 76c167a
commit f751105
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/benchmarks/bench/run_length_encode/encode.cu b/benchmarks/bench/run_length_encode/encode.cu
@@ -91,6 +91,7 @@ static void rle(nvbench::state &state, nvbench::type_list<T, OffsetT>)
                                               accum_t,
                                               device_reduce_by_key_policy_hub>;
   #else
+  using policy_t = cub::detail::device_run_length_encode_policy_hub<accum_t, key_t>;
   using dispatch_t = cub::DispatchReduceByKey<keys_input_it_t,
                                               unique_output_it_t,
                                               vals_input_it_t,
@@ -99,7 +100,8 @@ static void rle(nvbench::state &state, nvbench::type_list<T, OffsetT>)
                                               equality_op_t,
                                               reduction_op_t,
                                               offset_t,
-                                              accum_t>;
+                                              accum_t,
+                                              policy_t>;
   #endif
 
   const auto elements = static_cast<std::size_t>(state.get_int64("Elements{io}"));