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

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

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

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

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()

◆ 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 
)