BlockGemmARegBGmemCRegV1< Problem_, Policy_ > Struct Template Reference

BlockGemmARegBGmemCRegV1&lt; Problem_, Policy_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ > Struct Template Reference
Public Types | Public Member Functions | Static Public Member Functions | Static Public Attributes | List of all members
ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ > Struct Template Reference

#include <block_gemm_areg_bgmem_creg_v1.hpp>

Public Types

using Problem = remove_cvref_t<Problem_>
using Policy = remove_cvref_t<Policy_>
using ADataType = remove_cvref_t<typename Problem::ADataType>
using BDataType = remove_cvref_t<typename Problem::BDataType>
using CDataType = remove_cvref_t<typename Problem::CDataType>
using BlockGemmShape = remove_cvref_t<typename Problem::BlockGemmShape>
using BlockGemmARegBGmemCRegImpl

Public Member Functions

template<typename CBlockTensor, typename ABlockTensor, typename BBlockGmemWindowTmp>
CK_TILE_DEVICE void operator() (CBlockTensor &c_block_tensor, const ABlockTensor &a_block_tensor, const BBlockGmemWindowTmp &b_block_gmem_window_tmp, void *smem_ptr) const
template<typename ABlockTensor, typename BBlockGmemWindowTmp>
CK_TILE_DEVICE auto operator() (const ABlockTensor &a_block_tensor, const BBlockGmemWindowTmp &b_block_gmem_window_tmp, void *smem_ptr) const

Static Public Member Functions

static CK_TILE_HOST_DEVICE constexpr index_t GetStaticLdsSize ()

Static Public Attributes

static constexpr index_t kBlockSize = Problem::kBlockSize

Member Typedef Documentation

◆ ADataType

template<typename Problem_, typename Policy_ = BlockGemmARegBGmemCRegV1DefaultPolicy>
using ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ >::ADataType = remove_cvref_t<typename Problem::ADataType>

◆ BDataType

template<typename Problem_, typename Policy_ = BlockGemmARegBGmemCRegV1DefaultPolicy>
using ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ >::BDataType = remove_cvref_t<typename Problem::BDataType>

◆ BlockGemmARegBGmemCRegImpl

template<typename Problem_, typename Policy_ = BlockGemmARegBGmemCRegV1DefaultPolicy>
using ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ >::BlockGemmARegBGmemCRegImpl
Initial value:
BlockGemmARegBGmemCRegV1<
BlockGemmProblem<ADataType, BDataType, CDataType, kBlockSize, BlockGemmShape>,
BlockGemmARegBGmemCRegV1DefaultPolicy>
ck_tile::BlockGemmARegBGmemCRegV1DefaultPolicy
Definition block_gemm_areg_bgmem_creg_v1_default_policy.hpp:13
ck_tile::BlockGemmARegBGmemCRegV1
Definition block_gemm_areg_bgmem_creg_v1.hpp:20
ck_tile::BlockGemmProblem
Definition block_gemm_problem.hpp:18

◆ BlockGemmShape

template<typename Problem_, typename Policy_ = BlockGemmARegBGmemCRegV1DefaultPolicy>
using ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ >::BlockGemmShape = remove_cvref_t<typename Problem::BlockGemmShape>

◆ CDataType

template<typename Problem_, typename Policy_ = BlockGemmARegBGmemCRegV1DefaultPolicy>
using ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ >::CDataType = remove_cvref_t<typename Problem::CDataType>

◆ Policy

template<typename Problem_, typename Policy_ = BlockGemmARegBGmemCRegV1DefaultPolicy>
using ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ >::Policy = remove_cvref_t<Policy_>

◆ Problem

template<typename Problem_, typename Policy_ = BlockGemmARegBGmemCRegV1DefaultPolicy>
using ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ >::Problem = remove_cvref_t<Problem_>

Member Function Documentation

◆ GetStaticLdsSize()

template<typename Problem_, typename Policy_ = BlockGemmARegBGmemCRegV1DefaultPolicy>
CK_TILE_HOST_DEVICE constexpr index_t ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ >::GetStaticLdsSize ( )
inlinestaticconstexpr

◆ operator()() [1/2]

template<typename Problem_, typename Policy_ = BlockGemmARegBGmemCRegV1DefaultPolicy>
template<typename CBlockTensor, typename ABlockTensor, typename BBlockGmemWindowTmp>
CK_TILE_DEVICE void ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ >::operator() ( CBlockTensor & c_block_tensor,
const ABlockTensor & a_block_tensor,
const BBlockGmemWindowTmp & b_block_gmem_window_tmp,
void * smem_ptr ) const
inline

◆ operator()() [2/2]

template<typename Problem_, typename Policy_ = BlockGemmARegBGmemCRegV1DefaultPolicy>
template<typename ABlockTensor, typename BBlockGmemWindowTmp>
CK_TILE_DEVICE auto ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ >::operator() ( const ABlockTensor & a_block_tensor,
const BBlockGmemWindowTmp & b_block_gmem_window_tmp,
void * smem_ptr ) const
inline

Member Data Documentation

◆ kBlockSize

template<typename Problem_, typename Policy_ = BlockGemmARegBGmemCRegV1DefaultPolicy>
index_t ck_tile::BlockGemmARegBGmemCRegV1< Problem_, Policy_ >::kBlockSize = Problem::kBlockSize
staticconstexpr
The documentation for this struct was generated from the following file: