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

• develop
• include
• ck_tile
• ops
• gemm
• block

block_gemm_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 // Problem Description for BlockGemm

11 template <typename ADataType_,

12  typename BDataType_,

13  typename CDataType_,

14  index_t kBlockSize_,

15  typename BlockGemmShape_,

16  index_t NumWaveGroups_ = 1>

17 struct BlockGemmProblem

18 {

19  using ADataType = remove_cvref_t<ADataType_>;

20  using BDataType = remove_cvref_t<BDataType_>;

21  using CDataType = remove_cvref_t<CDataType_>;

22  using BlockGemmShape = remove_cvref_t<BlockGemmShape_>;

23 

24  static constexpr index_t kBlockSize = kBlockSize_;

25  static constexpr index_t NumWaveGroups = NumWaveGroups_;

26 };

27 

28 } // 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:21

ck_tile::BlockGemmProblem

Definition: block_gemm_problem.hpp:18

ck_tile::BlockGemmProblem::NumWaveGroups

static constexpr index_t NumWaveGroups

Definition: block_gemm_problem.hpp:25

ck_tile::BlockGemmProblem::ADataType

remove_cvref_t< ADataType_ > ADataType

Definition: block_gemm_problem.hpp:19

ck_tile::BlockGemmProblem::BlockGemmShape

remove_cvref_t< BlockGemmShape_ > BlockGemmShape

Definition: block_gemm_problem.hpp:22

ck_tile::BlockGemmProblem::BDataType

remove_cvref_t< BDataType_ > BDataType

Definition: block_gemm_problem.hpp:20

ck_tile::BlockGemmProblem::kBlockSize

static constexpr index_t kBlockSize

Definition: block_gemm_problem.hpp:24

ck_tile::BlockGemmProblem::CDataType

remove_cvref_t< CDataType_ > CDataType

Definition: block_gemm_problem.hpp:21