BatchedTransposeKernel< Pipeline_ > Struct Template Reference

BatchedTransposeKernel&lt; Pipeline_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::BatchedTransposeKernel< Pipeline_ > Struct Template Reference
ck_tile::BatchedTransposeKernel< Pipeline_ > Struct Template Reference

#include <batched_transpose_kernel.hpp>

Classes

struct  BatchedTransposeKargs

Public Types

using Pipeline = remove_cvref_t<Pipeline_>
using Problem = remove_cvref_t<typename Pipeline::Problem>
using Type = typename Problem::DataType
using Kargs = BatchedTransposeKargs
using Hargs = BatchedTransposeHostArgs

Public Member Functions

CK_TILE_DEVICE void operator() (Kargs kargs) const

Static Public Member Functions

static CK_TILE_HOST constexpr auto GridSize (const Hargs &host_args)
static CK_TILE_HOST constexpr auto MakeKargs (const Hargs &h)
static CK_TILE_HOST constexpr auto BlockSize ()

Static Public Attributes

static CK_TILE_DEVICE index_t counter = 0
static constexpr index_t kBlockSize = Problem::kBlockSize

Member Typedef Documentation

◆ Hargs

template<typename Pipeline_>
using ck_tile::BatchedTransposeKernel< Pipeline_ >::Hargs = BatchedTransposeHostArgs

◆ Kargs

template<typename Pipeline_>
using ck_tile::BatchedTransposeKernel< Pipeline_ >::Kargs = BatchedTransposeKargs

◆ Pipeline

template<typename Pipeline_>
using ck_tile::BatchedTransposeKernel< Pipeline_ >::Pipeline = remove_cvref_t<Pipeline_>

◆ Problem

template<typename Pipeline_>
using ck_tile::BatchedTransposeKernel< Pipeline_ >::Problem = remove_cvref_t<typename Pipeline::Problem>

◆ Type

template<typename Pipeline_>
using ck_tile::BatchedTransposeKernel< Pipeline_ >::Type = typename Problem::DataType

Member Function Documentation

◆ BlockSize()

template<typename Pipeline_>
CK_TILE_HOST constexpr auto ck_tile::BatchedTransposeKernel< Pipeline_ >::BlockSize ( )
inlinestaticconstexpr

◆ GridSize()

template<typename Pipeline_>
CK_TILE_HOST constexpr auto ck_tile::BatchedTransposeKernel< Pipeline_ >::GridSize ( const Hargs & host_args)
inlinestaticconstexpr

◆ MakeKargs()

template<typename Pipeline_>
CK_TILE_HOST constexpr auto ck_tile::BatchedTransposeKernel< Pipeline_ >::MakeKargs ( const Hargs & h)
inlinestaticconstexpr

◆ operator()()

template<typename Pipeline_>
CK_TILE_DEVICE void ck_tile::BatchedTransposeKernel< Pipeline_ >::operator() ( Kargs kargs) const
inline

Member Data Documentation

◆ counter

template<typename Pipeline_>
CK_TILE_DEVICE index_t ck_tile::BatchedTransposeKernel< Pipeline_ >::counter = 0
static

◆ kBlockSize

template<typename Pipeline_>
index_t ck_tile::BatchedTransposeKernel< Pipeline_ >::kBlockSize = Problem::kBlockSize
staticconstexpr

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