GenericPermute< Problem_ > Struct Template Reference

GenericPermute&lt; Problem_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::GenericPermute< Problem_ > Struct Template Reference
ck_tile::GenericPermute< Problem_ > Struct Template Reference

#include <generic_permute_kernel.hpp>

Classes

struct  Kargs

Public Types

using Problem = ck_tile::remove_cvref_t<Problem_>
using DataType = remove_cvref_t<typename Problem::DataType>

Public Member Functions

CK_TILE_DEVICE void operator() (Kargs kargs) const

Static Public Member Functions

static CK_TILE_HOST constexpr index_t TotalElements (const GenericPermuteHostArgs &h)
static CK_TILE_HOST constexpr Kargs MakeKargs (const GenericPermuteHostArgs &h)
static CK_TILE_HOST constexpr auto GridSize (GenericPermuteHostArgs h)
static CK_TILE_HOST_DEVICE constexpr auto BlockSize ()

Static Public Attributes

static constexpr index_t kBlockSize = Problem::kBlockSize
static constexpr index_t kMaxRanks = Problem::kMaxRanks
static constexpr bool KeepLastDim = Problem::KeepLastDim

Member Typedef Documentation

◆ DataType

template<typename Problem_>
using ck_tile::GenericPermute< Problem_ >::DataType = remove_cvref_t<typename Problem::DataType>

◆ Problem

template<typename Problem_>
using ck_tile::GenericPermute< Problem_ >::Problem = ck_tile::remove_cvref_t<Problem_>

Member Function Documentation

◆ BlockSize()

template<typename Problem_>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::GenericPermute< Problem_ >::BlockSize ( )
inlinestaticconstexpr

◆ GridSize()

template<typename Problem_>
CK_TILE_HOST constexpr auto ck_tile::GenericPermute< Problem_ >::GridSize ( GenericPermuteHostArgs h)
inlinestaticconstexpr

◆ MakeKargs()

template<typename Problem_>
CK_TILE_HOST constexpr Kargs ck_tile::GenericPermute< Problem_ >::MakeKargs ( const GenericPermuteHostArgs & h)
inlinestaticconstexpr

◆ operator()()

template<typename Problem_>
CK_TILE_DEVICE void ck_tile::GenericPermute< Problem_ >::operator() ( Kargs kargs) const
inline

◆ TotalElements()

template<typename Problem_>
CK_TILE_HOST constexpr index_t ck_tile::GenericPermute< Problem_ >::TotalElements ( const GenericPermuteHostArgs & h)
inlinestaticconstexpr

Member Data Documentation

◆ kBlockSize

template<typename Problem_>
index_t ck_tile::GenericPermute< Problem_ >::kBlockSize = Problem::kBlockSize
staticconstexpr

◆ KeepLastDim

template<typename Problem_>
bool ck_tile::GenericPermute< Problem_ >::KeepLastDim = Problem::KeepLastDim
staticconstexpr

◆ kMaxRanks

template<typename Problem_>
index_t ck_tile::GenericPermute< Problem_ >::kMaxRanks = Problem::kMaxRanks
staticconstexpr

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