From f7511053100cdcc789354a543aeb815929e5681d Mon Sep 17 00:00:00 2001
From: Georgy Evtushenko <evtushenko.georgy@gmail.com>
Date: Tue, 20 Jun 2023 22:28:01 +0400
Subject: [PATCH] Fix encode bench

---
 benchmarks/bench/run_length_encode/encode.cu | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/benchmarks/bench/run_length_encode/encode.cu b/benchmarks/bench/run_length_encode/encode.cu
index 286c1da8c..8d22311d1 100644
--- 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}"));