BlockGemmASmemBRegCRegV1< Problem_, Policy_ > Struct Template Reference

BlockGemmASmemBRegCRegV1< Problem_, Policy_ > Struct Template Reference#

Composable Kernel: ck_tile::BlockGemmASmemBRegCRegV1< Problem_, Policy_ > Struct Template Reference

#include <block_gemm_asmem_breg_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>

Public Member Functions
template<typename CBlockTensor, typename ABlockWindowTmp, typename BBlockTensorTmp>
CK_TILE_DEVICE void	operator() (CBlockTensor &c_block_tensor, const ABlockWindowTmp &a_block_window_tmp, const BBlockTensorTmp &b_block_tensor_tmp) const
template<typename ABlockWindowTmp, typename BBlockTensorTmp>
CK_TILE_DEVICE auto	operator() (const ABlockWindowTmp &a_block_window_tmp, const BBlockTensorTmp &b_block_tensor_tmp) const

Static Public Member Functions
static CK_TILE_DEVICE constexpr auto	MakeCBlockTile ()

Static Public Attributes
static constexpr index_t	kBlockSize = Problem::kBlockSize

Member Typedef Documentation

◆ ADataType

template<typename Problem_, typename Policy_ = BlockGemmASmemBRegCRegV1DefaultPolicy>

using ck_tile::BlockGemmASmemBRegCRegV1< Problem_, Policy_ >::ADataType = remove_cvref_t<typename Problem::ADataType>

◆ BDataType

template<typename Problem_, typename Policy_ = BlockGemmASmemBRegCRegV1DefaultPolicy>

using ck_tile::BlockGemmASmemBRegCRegV1< Problem_, Policy_ >::BDataType = remove_cvref_t<typename Problem::BDataType>

◆ BlockGemmShape

template<typename Problem_, typename Policy_ = BlockGemmASmemBRegCRegV1DefaultPolicy>

using ck_tile::BlockGemmASmemBRegCRegV1< Problem_, Policy_ >::BlockGemmShape = remove_cvref_t<typename Problem::BlockGemmShape>

◆ CDataType

template<typename Problem_, typename Policy_ = BlockGemmASmemBRegCRegV1DefaultPolicy>

using ck_tile::BlockGemmASmemBRegCRegV1< Problem_, Policy_ >::CDataType = remove_cvref_t<typename Problem::CDataType>

◆ Policy

template<typename Problem_, typename Policy_ = BlockGemmASmemBRegCRegV1DefaultPolicy>

using ck_tile::BlockGemmASmemBRegCRegV1< Problem_, Policy_ >::Policy = remove_cvref_t<Policy_>

◆ Problem

template<typename Problem_, typename Policy_ = BlockGemmASmemBRegCRegV1DefaultPolicy>

using ck_tile::BlockGemmASmemBRegCRegV1< Problem_, Policy_ >::Problem = remove_cvref_t<Problem_>

Member Function Documentation

◆ MakeCBlockTile()

template<typename Problem_, typename Policy_ = BlockGemmASmemBRegCRegV1DefaultPolicy>

CK_TILE_DEVICE constexpr auto ck_tile::BlockGemmASmemBRegCRegV1< Problem_, Policy_ >::MakeCBlockTile ( )

inlinestaticconstexpr

◆ operator()() [1/2]

template<typename Problem_, typename Policy_ = BlockGemmASmemBRegCRegV1DefaultPolicy>

template<typename CBlockTensor, typename ABlockWindowTmp, typename BBlockTensorTmp>

CK_TILE_DEVICE void ck_tile::BlockGemmASmemBRegCRegV1< Problem_, Policy_ >::operator()	(	CBlockTensor &	c_block_tensor,
		const ABlockWindowTmp &	a_block_window_tmp,
		const BBlockTensorTmp &	b_block_tensor_tmp ) const

inline

◆ operator()() [2/2]

template<typename Problem_, typename Policy_ = BlockGemmASmemBRegCRegV1DefaultPolicy>

template<typename ABlockWindowTmp, typename BBlockTensorTmp>

CK_TILE_DEVICE auto ck_tile::BlockGemmASmemBRegCRegV1< Problem_, Policy_ >::operator()	(	const ABlockWindowTmp &	a_block_window_tmp,
		const BBlockTensorTmp &	b_block_tensor_tmp ) const

inline

Member Data Documentation

◆ kBlockSize

template<typename Problem_, typename Policy_ = BlockGemmASmemBRegCRegV1DefaultPolicy>

index_t ck_tile::BlockGemmASmemBRegCRegV1< Problem_, Policy_ >::kBlockSize = Problem::kBlockSize

staticconstexpr

The documentation for this struct was generated from the following file:

block_gemm_asmem_breg_creg_v1.hpp