device_radix_sort_config.hpp

// Copyright (c) 2022-2023 Advanced Micro Devices, Inc. All rights reserved.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

#ifndef ROCPRIM_DEVICE_DEVICE_RADIX_SORT_CONFIG_HPP_
#define ROCPRIM_DEVICE_DEVICE_RADIX_SORT_CONFIG_HPP_

#include "config_types.hpp"
#include "detail/config/device_radix_sort_block_sort.hpp"
#include "detail/device_config_helper.hpp"


BEGIN_ROCPRIM_NAMESPACE

template<class SingleSortConfig = default_config,
         class MergeSortConfig  = default_config,
         class OnesweepConfig   = default_config,
         size_t MergeSortLimit  = 1024 * 1024>
struct radix_sort_config_v2
{
#ifndef DOXYGEN_SHOULD_SKIP_THIS
    using single_sort_config = SingleSortConfig;
    using merge_sort_config = MergeSortConfig;
    using onesweep_config = OnesweepConfig;
    static constexpr size_t merge_sort_limit = MergeSortLimit;
#endif
};

template<unsigned int LongRadixBits,
         unsigned int ShortRadixBits,
         class ScanConfig,
         class SortConfig,
         class SortSingleConfig               = kernel_config<256, 10>,
         class SortMergeConfig                = kernel_config<1024, 1>,
         unsigned int MergeSizeLimitBlocks    = 1024U,
         bool         ForceSingleKernelConfig = false,
         class OnesweepHistogramConfig        = kernel_config<256, 8>,
         class OnesweepSortConfig             = kernel_config<256, 15>,
         unsigned int OnesweepRadixBits       = 4>
struct [[deprecated("use radix_sort_config_v2")]] radix_sort_config
{
#ifndef DOXYGEN_SHOULD_SKIP_THIS
    using single_sort_config = SortSingleConfig;
    using merge_sort_config = default_config;
    using onesweep_config = radix_sort_onesweep_config<OnesweepHistogramConfig,
                                                       OnesweepSortConfig,
                                                       OnesweepRadixBits>;
    static constexpr size_t merge_sort_limit = 1024 * MergeSizeLimitBlocks;
#endif
};

namespace detail
{

// sub-algorithm onesweep:
template<typename RadixSortOnesweepConfig, typename, typename>
struct wrapped_radix_sort_onesweep_config
{
    template<target_arch Arch>
    struct architecture_config
    {
        static constexpr radix_sort_onesweep_config_params params = RadixSortOnesweepConfig();
    };
};

template<typename Key, typename Value>
struct wrapped_radix_sort_onesweep_config<default_config, Key, Value>
{
    template<target_arch Arch>
    struct architecture_config
    {
        static constexpr radix_sort_onesweep_config_params params
            = default_radix_sort_onesweep_config<static_cast<unsigned int>(Arch), Key, Value>();
    };
};

#ifndef DOXYGEN_SHOULD_SKIP_THIS
template<typename RadixSortOnesweepConfig, typename Key, typename Value>
template<target_arch Arch>
constexpr radix_sort_onesweep_config_params
    wrapped_radix_sort_onesweep_config<RadixSortOnesweepConfig, Key, Value>::architecture_config<
        Arch>::params;

template<typename Key, typename Value>
template<target_arch Arch>
constexpr radix_sort_onesweep_config_params
    wrapped_radix_sort_onesweep_config<default_config, Key, Value>::architecture_config<
        Arch>::params;
#endif // DOXYGEN_SHOULD_SKIP_THIS

// Sub-algorithm block_sort:
template<typename RadixSortBlockSortConfig, typename, typename>
struct wrapped_radix_sort_block_sort_config
{
    template<target_arch Arch>
    struct architecture_config
    {
        static constexpr kernel_config_params params = RadixSortBlockSortConfig();
    };
};

template<typename Key, typename Value>
struct wrapped_radix_sort_block_sort_config<default_config, Key, Value>
{
    template<target_arch Arch>
    struct architecture_config
    {
        static constexpr kernel_config_params params
            = default_radix_sort_block_sort_config<static_cast<unsigned int>(Arch), Key, Value>();
    };
};

#ifndef DOXYGEN_SHOULD_SKIP_THIS
template<typename RadixSortBlockSortConfig, typename Key, typename Value>
template<target_arch Arch>
constexpr kernel_config_params
    wrapped_radix_sort_block_sort_config<RadixSortBlockSortConfig, Key, Value>::architecture_config<
        Arch>::params;

template<typename Key, typename Value>
template<target_arch Arch>
constexpr kernel_config_params wrapped_radix_sort_block_sort_config<default_config, Key, Value>::
    architecture_config<Arch>::params;
#endif // DOXYGEN_SHOULD_SKIP_THIS

} // namespace detail

END_ROCPRIM_NAMESPACE

// end of group primitivesmodule_deviceconfigs

#endif // ROCPRIM_DEVICE_DEVICE_RADIX_SORT_CONFIG_HPP_