/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/device/helper.hpp File Reference

• develop
• include
• ck
• tensor_operation
• gpu
• device

Typedefs | Functions

helper.hpp File Reference

#include "ck/utility/common_header.hpp"
#include "ck/tensor_description/multi_index_transform_helper.hpp"
#include "ck/tensor_operation/gpu/grid/block_to_ctile_map.hpp"
#include "ck/tensor_operation/operator_transform/transform_conv_fwd_to_gemm.hpp"
#include "ck/tensor_operation/gpu/device/matrix_padder.hpp"
#include <fstream>
#include <variant>

Go to the source code of this file.

Typedefs
using	layouts = std::variant< ck::tensor_layout::convolution::GNWK, ck::tensor_layout::convolution::GNHWK, ck::tensor_layout::convolution::NHWGK, ck::tensor_layout::convolution::GNDHWK, ck::tensor_layout::convolution::NDHWGK >

Functions
auto	layout_type (std::string type)
ck::tensor_operation::device::GemmSpecialization	gemm_type (std::string type)
ck::tensor_operation::device::ConvolutionForwardSpecialization	conv_type (std::string type)
template<typename CDesc_MRaw_NRaw >
auto	pad (ck::index_t mpb, ck::index_t npb, ck::index_t kpb, ck::tensor_operation::device::GemmSpecialization gemm, CDesc_MRaw_NRaw conv)
auto	transform_conv (ck::index_t num_dim, ck::tensor_operation::device::ConvolutionForwardSpecialization spec, ck::Array< ck::index_t, 5 > out_lengths, ck::Array< ck::index_t, 5 > out_strides)
auto	transform_conv_3d (ck::index_t num_dim, ck::tensor_operation::device::ConvolutionForwardSpecialization spec, ck::Array< ck::index_t, 6 > out_lengths, ck::Array< ck::index_t, 6 > out_strides)
auto	transform_conv_1d (ck::index_t num_dim, ck::tensor_operation::device::ConvolutionForwardSpecialization spec, ck::Array< ck::index_t, 4 > out_lengths, ck::Array< ck::index_t, 4 > out_strides)
template<typename CGridDesc_M_N >
auto	block_2_etile (ck::index_t m_per_block, ck::index_t n_per_block, CGridDesc_M_N matrix_padder)
auto	get_launch_params_1d (ck::host::Solution solution, ck::Array< ck::index_t, 4 > out_lengths, ck::Array< ck::index_t, 4 > out_strides)
auto	get_launch_params (ck::host::Solution solution, ck::Array< ck::index_t, 5 > out_lengths, ck::Array< ck::index_t, 5 > out_strides)
auto	get_launch_params_3d (ck::host::Solution solution, ck::Array< ck::index_t, 6 > out_lengths, ck::Array< ck::index_t, 6 > out_strides)

Typedef Documentation

◆ layouts

using layouts = std::variant<ck::tensor_layout::convolution::GNWK, ck::tensor_layout::convolution::GNHWK, ck::tensor_layout::convolution::NHWGK, ck::tensor_layout::convolution::GNDHWK, ck::tensor_layout::convolution::NDHWGK>

Function Documentation

◆ block_2_etile()

template<typename CGridDesc_M_N >

auto block_2_etile	(	ck::index_t	m_per_block,
		ck::index_t	n_per_block,
		CGridDesc_M_N	matrix_padder
	)

◆ conv_type()

ck::tensor_operation::device::ConvolutionForwardSpecialization conv_type

(

std::string

type

)

◆ gemm_type()

ck::tensor_operation::device::GemmSpecialization gemm_type

(

std::string

type

)

◆ get_launch_params()

auto get_launch_params	(	ck::host::Solution	solution,
		ck::Array< ck::index_t, 5 >	out_lengths,
		ck::Array< ck::index_t, 5 >	out_strides
	)

◆ get_launch_params_1d()

auto get_launch_params_1d	(	ck::host::Solution	solution,
		ck::Array< ck::index_t, 4 >	out_lengths,
		ck::Array< ck::index_t, 4 >	out_strides
	)

◆ get_launch_params_3d()

auto get_launch_params_3d	(	ck::host::Solution	solution,
		ck::Array< ck::index_t, 6 >	out_lengths,
		ck::Array< ck::index_t, 6 >	out_strides
	)

◆ layout_type()

auto layout_type

(

std::string

type

)

◆ pad()

template<typename CDesc_MRaw_NRaw >

auto pad	(	ck::index_t	mpb,
		ck::index_t	npb,
		ck::index_t	kpb,
		ck::tensor_operation::device::GemmSpecialization	gemm,
		CDesc_MRaw_NRaw	conv
	)

◆ transform_conv()

auto transform_conv	(	ck::index_t	num_dim,
		ck::tensor_operation::device::ConvolutionForwardSpecialization	spec,
		ck::Array< ck::index_t, 5 >	out_lengths,
		ck::Array< ck::index_t, 5 >	out_strides
	)

◆ transform_conv_1d()

auto transform_conv_1d	(	ck::index_t	num_dim,
		ck::tensor_operation::device::ConvolutionForwardSpecialization	spec,
		ck::Array< ck::index_t, 4 >	out_lengths,
		ck::Array< ck::index_t, 4 >	out_strides
	)

◆ transform_conv_3d()

auto transform_conv_3d	(	ck::index_t	num_dim,
		ck::tensor_operation::device::ConvolutionForwardSpecialization	spec,
		ck::Array< ck::index_t, 6 >	out_lengths,
		ck::Array< ck::index_t, 6 >	out_strides
	)