include/ck_tile/ops/topk/block/block_topk_stream_2d_problem.hpp Source File

• include
• ck_tile
• ops
• topk
• block

block_topk_stream_2d_problem.hpp

Go to the documentation of this file.

1 // SPDX-License-Identifier: MIT

2 // Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.

3 

4 #pragma once

5 

6 #include "ck_tile/core.hpp"

7 

8 namespace ck_tile {

9 

10 /*

11 simple 2d topk implementation, along row (dim=1)

12 requirement:

13  1). each row is within a warp

14 */

15 template <typename DataType_, typename IndexType_, index_t ColLanes_>

16 struct BlockTopkStream2DProblem

17 {

18  using DataType = remove_cvref_t<DataType_>;

19  using IndexType = remove_cvref_t<IndexType_>;

20  static constexpr index_t ColLanes = ColLanes_;

21 };

22 } // namespace ck_tile

core.hpp

ck_tile

Definition: cluster_descriptor.hpp:13

ck_tile::index_t

int32_t index_t

Definition: integer.hpp:9

ck_tile::remove_cvref_t

remove_cv_t< std::remove_reference_t< T > > remove_cvref_t

Definition: type_traits.hpp:20

ck_tile::BlockTopkStream2DProblem

Definition: block_topk_stream_2d_problem.hpp:17

ck_tile::BlockTopkStream2DProblem::IndexType

remove_cvref_t< IndexType_ > IndexType

Definition: block_topk_stream_2d_problem.hpp:19

ck_tile::BlockTopkStream2DProblem::DataType

remove_cvref_t< DataType_ > DataType

Definition: block_topk_stream_2d_problem.hpp:18

ck_tile::BlockTopkStream2DProblem::ColLanes

static constexpr index_t ColLanes

Definition: block_topk_stream_2d_problem.hpp:20