helper.hpp File Reference

helper.hpp File Reference#

Composable Kernel: helper.hpp File Reference
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

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
Initial value:
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>
ck::tensor_layout::convolution::GNDHWK
Definition tensor_operation/gpu/device/tensor_layout.hpp:345
ck::tensor_layout::convolution::GNHWK
Definition tensor_operation/gpu/device/tensor_layout.hpp:340
ck::tensor_layout::convolution::GNWK
Definition tensor_operation/gpu/device/tensor_layout.hpp:335
ck::tensor_layout::convolution::NDHWGK
Definition tensor_operation/gpu/device/tensor_layout.hpp:362
ck::tensor_layout::convolution::NHWGK
Definition tensor_operation/gpu/device/tensor_layout.hpp:357

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 )