block_softmax_2d_problem.hpp Source File

• include
• ck_tile
• ops
• softmax
• block

block_softmax_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

8namespace ck_tile {

9

10template <typename DataType_>

11struct BlockSoftmax2DProblem

12{

13 using DataType = remove_cvref_t<DataType_>;

14};

15

16} // namespace ck_tile

core.hpp

ck_tile

Definition tile/core/algorithm/cluster_descriptor.hpp:13

ck_tile::remove_cvref_t

remove_cv_t< std::remove_reference_t< T > > remove_cvref_t

Definition type_traits.hpp:21

ck_tile::BlockSoftmax2DProblem

Definition block_softmax_2d_problem.hpp:12

ck_tile::BlockSoftmax2DProblem::DataType

remove_cvref_t< DataType_ > DataType

Definition block_softmax_2d_problem.hpp:13