/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

/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#

Composable Kernel: /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
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>
18 {
23 
24  static constexpr index_t kBlockSize = kBlockSize_;
25  static constexpr index_t NumWaveGroups = NumWaveGroups_;
26 };
27 
28 } // namespace ck_tile
Definition: cluster_descriptor.hpp:13
int32_t index_t
Definition: integer.hpp:9
remove_cv_t< std::remove_reference_t< T > > remove_cvref_t
Definition: type_traits.hpp:21
Definition: block_gemm_problem.hpp:18
static constexpr index_t NumWaveGroups
Definition: block_gemm_problem.hpp:25
remove_cvref_t< ADataType_ > ADataType
Definition: block_gemm_problem.hpp:19
remove_cvref_t< BlockGemmShape_ > BlockGemmShape
Definition: block_gemm_problem.hpp:22
remove_cvref_t< BDataType_ > BDataType
Definition: block_gemm_problem.hpp:20
static constexpr index_t kBlockSize
Definition: block_gemm_problem.hpp:24
remove_cvref_t< CDataType_ > CDataType
Definition: block_gemm_problem.hpp:21