Fix tuning for keys only

gevtushenko · May 7, 2023 · 0a80e78 · 0a80e78
1 parent d18fa6a
commit 0a80e78
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cub/device/dispatch/dispatch_radix_sort.cuh b/cub/device/dispatch/dispatch_radix_sort.cuh
@@ -628,8 +628,8 @@ struct sm90_small_key_tuning
 // clang-format off
 
 // keys
-template <> struct sm90_small_key_tuning<1,  0, 4> { static constexpr int threads = 512; static constexpr int items = 15; };
-template <> struct sm90_small_key_tuning<2,  0, 4> { static constexpr int threads = 512; static constexpr int items = 15; };
+template <> struct sm90_small_key_tuning<1,  0, 4> { static constexpr int threads = 512; static constexpr int items = 19; };
+template <> struct sm90_small_key_tuning<2,  0, 4> { static constexpr int threads = 512; static constexpr int items = 19; };
 
 // pairs  8:xx
 template <> struct sm90_small_key_tuning<1,  1, 4> { static constexpr int threads = 512; static constexpr int items = 15; };