#include <gridwise_gemm_pipeline_v3.hpp>
|
__host__ static constexpr __device__ bool | IsSupported (index_t) |
|
template<typename AGridDesc , typename ABlockDesc , typename ABlockTransfer , typename AGridBuffer , typename ABlockBuffer , typename ABlockTransferStep , typename BGridDesc , typename BBlockDesc , typename BBlockTransfer , typename BGridBuffer , typename BBlockBuffer , typename BBlockTransferStep , typename BlockwiseGemm , typename CThreadBuffer > |
static __device__ void | Run (const AGridDesc &a_grid_desc, const ABlockDesc &a_block_desc, ABlockTransfer &a_blockwise_copy, const AGridBuffer &a_grid_buf, ABlockBuffer &a_block_buf, const ABlockTransferStep &a_block_copy_step, const BGridDesc &b_grid_desc, const BBlockDesc &b_block_desc, BBlockTransfer &b_blockwise_copy, const BGridBuffer &b_grid_buf, BBlockBuffer &b_block_buf, const BBlockTransferStep &b_block_copy_step, const BlockwiseGemm &blockwise_gemm, CThreadBuffer &c_thread_buf, index_t num_loop) |
|
◆ IsSupported()
__host__ static constexpr __device__ bool ck::GridwiseGemmPipeline_v3::IsSupported |
( |
index_t |
| ) |
|
|
inlinestaticconstexpr |
◆ Run()
template<typename AGridDesc , typename ABlockDesc , typename ABlockTransfer , typename AGridBuffer , typename ABlockBuffer , typename ABlockTransferStep , typename BGridDesc , typename BBlockDesc , typename BBlockTransfer , typename BGridBuffer , typename BBlockBuffer , typename BBlockTransferStep , typename BlockwiseGemm , typename CThreadBuffer >
static __device__ void ck::GridwiseGemmPipeline_v3::Run |
( |
const AGridDesc & |
a_grid_desc, |
|
|
const ABlockDesc & |
a_block_desc, |
|
|
ABlockTransfer & |
a_blockwise_copy, |
|
|
const AGridBuffer & |
a_grid_buf, |
|
|
ABlockBuffer & |
a_block_buf, |
|
|
const ABlockTransferStep & |
a_block_copy_step, |
|
|
const BGridDesc & |
b_grid_desc, |
|
|
const BBlockDesc & |
b_block_desc, |
|
|
BBlockTransfer & |
b_blockwise_copy, |
|
|
const BGridBuffer & |
b_grid_buf, |
|
|
BBlockBuffer & |
b_block_buf, |
|
|
const BBlockTransferStep & |
b_block_copy_step, |
|
|
const BlockwiseGemm & |
blockwise_gemm, |
|
|
CThreadBuffer & |
c_thread_buf, |
|
|
index_t |
num_loop |
|
) |
| |
|
inlinestatic |
The documentation for this struct was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/grid/gridwise_gemm_pipeline_v3.hpp