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

• docs-6.4.3
• 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 struct BlockGemmProblem

17 {

18  using ADataType = remove_cvref_t<ADataType_>;

19  using BDataType = remove_cvref_t<BDataType_>;

20  using CDataType = remove_cvref_t<CDataType_>;

21  using BlockGemmShape = remove_cvref_t<BlockGemmShape_>;

22 

23  static constexpr index_t kBlockSize = kBlockSize_;

24 };

25 

26 } // 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::BlockGemmProblem

Definition: block_gemm_problem.hpp:17

ck_tile::BlockGemmProblem::BDataType

remove_cvref_t< BDataType_ > BDataType

Definition: block_gemm_problem.hpp:19

ck_tile::BlockGemmProblem::CDataType

remove_cvref_t< CDataType_ > CDataType

Definition: block_gemm_problem.hpp:20

ck_tile::BlockGemmProblem::BlockGemmShape

remove_cvref_t< BlockGemmShape_ > BlockGemmShape

Definition: block_gemm_problem.hpp:21

ck_tile::BlockGemmProblem::kBlockSize

static constexpr index_t kBlockSize

Definition: block_gemm_problem.hpp:23

ck_tile::BlockGemmProblem::ADataType

remove_cvref_t< ADataType_ > ADataType

Definition: block_gemm_problem.hpp:18