Skip to content

Commit

Permalink
Formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
gevtushenko committed May 7, 2023
1 parent 26d869c commit f4b02e8
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions cub/device/dispatch/dispatch_radix_sort.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -612,7 +612,7 @@ __global__ void DeviceRadixSortExclusiveSumKernel(OffsetT* d_bins)
}
}

namespace detail
namespace detail
{
namespace radix
{
Expand All @@ -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; };
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit f4b02e8

Please sign in to comment.