diff --git a/cub/device/dispatch/dispatch_radix_sort.cuh b/cub/device/dispatch/dispatch_radix_sort.cuh index 63651b1bd..e55d3a4d8 100644 --- a/cub/device/dispatch/dispatch_radix_sort.cuh +++ b/cub/device/dispatch/dispatch_radix_sort.cuh @@ -612,7 +612,7 @@ __global__ void DeviceRadixSortExclusiveSumKernel(OffsetT* d_bins) } } -namespace detail +namespace detail { namespace radix { @@ -625,6 +625,8 @@ struct sm90_small_key_tuning static constexpr int items = 23; }; +// 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; }; @@ -652,10 +654,10 @@ template <> struct sm90_small_key_tuning<2, 8, 4> { static constexpr int thread template <> struct sm90_small_key_tuning<2, 8, 8> { static constexpr int threads = 512; static constexpr int items = 23; }; template <> struct sm90_small_key_tuning<2, 16, 4> { static constexpr int threads = 512; static constexpr int items = 21; }; template <> struct sm90_small_key_tuning<2, 16, 8> { static constexpr int threads = 576; static constexpr int items = 22; }; +// clang-format on -} // radix -} // detail - +} // namespace radix +} // namespace detail /****************************************************************************** * Policy