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 constexpr CK_TILE_HOST index_t TotalElements (const GenericPermuteHostArgs &h)
 
static constexpr CK_TILE_HOST Kargs MakeKargs (const GenericPermuteHostArgs &h)
 
static constexpr CK_TILE_HOST auto GridSize (GenericPermuteHostArgs h)
 
static constexpr CK_TILE_HOST_DEVICE 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_ >
static constexpr CK_TILE_HOST_DEVICE auto ck_tile::GenericPermute< Problem_ >::BlockSize ( )
inlinestaticconstexpr

◆ GridSize()

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

◆ MakeKargs()

template<typename Problem_ >
static constexpr CK_TILE_HOST 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_ >
static constexpr CK_TILE_HOST index_t ck_tile::GenericPermute< Problem_ >::TotalElements ( const GenericPermuteHostArgs h)
inlinestaticconstexpr

Member Data Documentation

◆ kBlockSize

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

◆ KeepLastDim

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

◆ kMaxRanks

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

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-6.4.3/include/ck_tile/ops/permute/kernel/generic_permute_kernel.hpp