/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/topk/block/block_topk_stream_2d_problem.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/topk/block/block_topk_stream_2d_problem.hpp Source File#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/topk/block/block_topk_stream_2d_problem.hpp Source File

Go to the documentation of this file.

 // SPDX-License-Identifier: MIT
 // Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
  
 #pragma once
  
 #include "ck_tile/core.hpp"
  
 namespace ck_tile {
  
 /*
 simple 2d topk implementation, along row (dim=1)
 requirement:
     1). each row is within a warp
 */
 template <typename DataType_, typename IndexType_, index_t ColLanes_>
 struct BlockTopkStream2DProblem
 {
     using DataType                    = remove_cvref_t<DataType_>;
     using IndexType                   = remove_cvref_t<IndexType_>;
     static constexpr index_t ColLanes = ColLanes_;
 };
 } // namespace ck_tile