GroupedConvHostArgs Struct Reference

GroupedConvHostArgs Struct Reference#

Composable Kernel: ck_tile::GroupedConvHostArgs Struct Reference

The Grouped Conv kernel host arguments. More...

#include <grouped_convolution_utils.hpp>

Inheritance diagram for ck_tile::GroupedConvHostArgs:

Public Member Functions
CK_TILE_HOST	GroupedConvHostArgs ()=delete

CK_TILE_HOST	GroupedConvHostArgs (ConvParam conv_param, const void in_ptr_, const void wei_ptr_, const std::vector< const void * > ds_ptr_, void *out_ptr_, index_t k_batch_)

Public Member Functions inherited from ck_tile::conv::ConvParam
	ConvParam (ck_tile::index_t n_dim, ck_tile::index_t group_count, ck_tile::index_t n_batch, ck_tile::index_t n_out_channels, ck_tile::index_t n_in_channels, const std::vector< ck_tile::index_t > &filters_len, const std::vector< ck_tile::index_t > &input_len, const std::vector< ck_tile::index_t > &strides, const std::vector< ck_tile::index_t > &dilations, const std::vector< ck_tile::index_t > &left_pads, const std::vector< ck_tile::index_t > &right_pads)

	ConvParam (ck_tile::long_index_t n_dim, ck_tile::long_index_t group_count, ck_tile::long_index_t n_batch, ck_tile::long_index_t n_out_channels, ck_tile::long_index_t n_in_channels, const std::vector< ck_tile::long_index_t > &filters_len, const std::vector< ck_tile::long_index_t > &input_len, const std::vector< ck_tile::long_index_t > &strides, const std::vector< ck_tile::long_index_t > &dilations, const std::vector< ck_tile::long_index_t > &left_pads, const std::vector< ck_tile::long_index_t > &right_pads)

std::vector< ck_tile::long_index_t >	GetOutputSpatialLengths () const

std::size_t	GetFlops () const

template<typename InDataType >
std::size_t	GetInputByte () const

template<typename WeiDataType >
std::size_t	GetWeightByte () const

template<typename OutDataType >
std::size_t	GetOutputByte () const

template<typename InDataType , typename WeiDataType , typename OutDataType >
std::size_t	GetByte () const

Public Attributes
const void *	in_ptr

const void *	wei_ptr

const std::vector< const void * >	ds_ptr

void *	out_ptr

index_t	k_batch

Public Attributes inherited from ck_tile::conv::ConvParam
ck_tile::long_index_t	num_dim_spatial_

ck_tile::long_index_t	G_

ck_tile::long_index_t	N_

ck_tile::long_index_t	K_

ck_tile::long_index_t	C_

std::vector< ck_tile::long_index_t >	filter_spatial_lengths_

std::vector< ck_tile::long_index_t >	input_spatial_lengths_

std::vector< ck_tile::long_index_t >	output_spatial_lengths_

std::vector< ck_tile::long_index_t >	conv_filter_strides_

std::vector< ck_tile::long_index_t >	conv_filter_dilations_

std::vector< ck_tile::long_index_t >	input_left_pads_

std::vector< ck_tile::long_index_t >	input_right_pads_

Detailed Description

The Grouped Conv kernel host arguments.

Overview: This structure is passed to Grouped Convolution Kernels when creating kernel arguments object. It contain all necessary information required to build proper kernel argument and launch kernel on GPU.

Constructor & Destructor Documentation

◆ GroupedConvHostArgs() [1/2]

CK_TILE_HOST ck_tile::GroupedConvHostArgs::GroupedConvHostArgs ( )

delete

◆ GroupedConvHostArgs() [2/2]

CK_TILE_HOST ck_tile::GroupedConvHostArgs::GroupedConvHostArgs	(	ConvParam	conv_param,
		const void *	in_ptr_,
		const void *	wei_ptr_,
		const std::vector< const void * >	ds_ptr_,
		void *	out_ptr_,
		index_t	k_batch_
	)

inline

Member Data Documentation

◆ ds_ptr

const std::vector<const void*> ck_tile::GroupedConvHostArgs::ds_ptr

◆ in_ptr

const void* ck_tile::GroupedConvHostArgs::in_ptr

◆ k_batch

index_t ck_tile::GroupedConvHostArgs::k_batch

◆ out_ptr

void* ck_tile::GroupedConvHostArgs::out_ptr

◆ wei_ptr

const void* ck_tile::GroupedConvHostArgs::wei_ptr

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-7.0.1/include/ck_tile/ops/grouped_convolution/utils/grouped_convolution_utils.hpp