tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ > Struct Template Reference

tile_window_with_tile_dstr_base&lt; TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ > Struct Template Reference
Classes | Public Types | Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ > Struct Template Reference

#include <tile_window_base.hpp>

Inheritance diagram for ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >:
ck_tile::tile_window_base< TileWindowType_, BottomTensorView_, WindowLengths_ >

Classes

struct  Traits

Public Types

using TileDstr = remove_cvref_t<StaticTileDistribution_>
using TileWindowBase = tile_window_base<TileWindowType_, BottomTensorView_, WindowLengths_>
using WindowAdaptor = typename TileDstr::PsYs2XsAdaptor
using AdaptorTopIndex = array<index_t, NDimWindowAdaptorTop>
using WindowAdaptorCoord
using BottomTensorCoord
Public Types inherited from ck_tile::tile_window_base< TileWindowType_, BottomTensorView_, WindowLengths_ >
using BottomTensorView = remove_reference_t<BottomTensorView_>
using WindowLengths = remove_cvref_t<WindowLengths_>
using BottomTensorDesc = typename BottomTensorView::TensorDesc
using DataType = remove_cvref_t<typename BottomTensorView::DataType>
using BottomTensorIndex = array<index_t, NDimBottomTensor>

Public Member Functions

CK_TILE_DEVICE constexpr auto get_tile_distribution () const
CK_TILE_HOST_DEVICE void init_raw ()
template<typename ATopIndex>
CK_TILE_DEVICE void move_window_adaptor_and_bottom_tensor_thread_coordinate (WindowAdaptorCoord &window_adaptor_thread_coord, BottomTensorCoord &bottom_tensor_thread_coord, const ATopIndex &idx_diff_adaptor_top) const
CK_TILE_DEVICE constexpr auto get_num_of_access () const
Public Member Functions inherited from ck_tile::tile_window_base< TileWindowType_, BottomTensorView_, WindowLengths_ >
CK_TILE_DEVICE constexpr auto get_window_origin () const
CK_TILE_DEVICE constexpr auto get_window_lengths () const
CK_TILE_DEVICE constexpr auto get_bottom_tensor_view () const
CK_TILE_DEVICE void set_window_origin (const BottomTensorIndex &new_window_origin)
CK_TILE_DEVICE void set_window_origin_extended (const BottomTensorIndex &)
CK_TILE_DEVICE constexpr void set_bottom_tensor_view_data_ptr (typename BottomTensorView::DataType *data)
CK_TILE_DEVICE void move (const BottomTensorIndex &step)
CK_TILE_DEVICE void move_extended (const BottomTensorIndex &)

Static Public Member Functions

static CK_TILE_DEVICE constexpr bool has_static_tile_distribution ()
static CK_TILE_DEVICE constexpr auto get_window_adaptor_ys_safe_vector_length_strides ()
Static Public Member Functions inherited from ck_tile::tile_window_base< TileWindowType_, BottomTensorView_, WindowLengths_ >
static CK_TILE_DEVICE constexpr index_t get_num_of_dimension ()

Public Attributes

TileDstr tile_dstr_
Public Attributes inherited from ck_tile::tile_window_base< TileWindowType_, BottomTensorView_, WindowLengths_ >
BottomTensorIndex window_origin_
WindowLengths window_lengths_
BottomTensorView bottom_tensor_view_

Static Public Attributes

static constexpr index_t NDimWindowAdaptorTop = WindowAdaptor::get_num_of_top_dimension()
static constexpr index_t NDimP = TileDstr::get_num_of_dimension_p()
static constexpr index_t NDimY = TileDstr::get_num_of_dimension_y()
Static Public Attributes inherited from ck_tile::tile_window_base< TileWindowType_, BottomTensorView_, WindowLengths_ >
static constexpr index_t NDimBottomTensor = BottomTensorDesc::get_num_of_dimension()

Member Typedef Documentation

◆ AdaptorTopIndex

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
using ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::AdaptorTopIndex = array<index_t, NDimWindowAdaptorTop>

◆ BottomTensorCoord

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
using ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::BottomTensorCoord
Initial value:
decltype(make_tensor_coordinate(
typename TileWindowBase::BottomTensorDesc{}, typename TileWindowBase::BottomTensorIndex{}))
ck_tile::make_tensor_coordinate
CK_TILE_HOST_DEVICE constexpr auto make_tensor_coordinate(const TensorDesc &tensor_desc, const TopIndex &idx_top)
Definition tensor_coordinate.hpp:60
ck_tile::tile_window_base::BottomTensorDesc
typename BottomTensorView::TensorDesc BottomTensorDesc
Definition tile_window_base.hpp:35
ck_tile::tile_window_base::BottomTensorIndex
array< index_t, NDimBottomTensor > BottomTensorIndex
Definition tile_window_base.hpp:43

◆ TileDstr

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
using ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::TileDstr = remove_cvref_t<StaticTileDistribution_>

◆ TileWindowBase

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
using ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::TileWindowBase = tile_window_base<TileWindowType_, BottomTensorView_, WindowLengths_>

◆ WindowAdaptor

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
using ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::WindowAdaptor = typename TileDstr::PsYs2XsAdaptor

◆ WindowAdaptorCoord

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
using ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::WindowAdaptorCoord
Initial value:
decltype(make_tensor_adaptor_coordinate(WindowAdaptor{}, AdaptorTopIndex{}))
ck_tile::make_tensor_adaptor_coordinate
CK_TILE_HOST_DEVICE constexpr auto make_tensor_adaptor_coordinate(const Adaptor &adaptor, const TopIndex &idx_top)
Definition tensor_adaptor_coordinate.hpp:55
ck_tile::tile_scatter_gather::WindowAdaptor
typename TileDstr::PsYs2XsAdaptor WindowAdaptor
Definition tile_scatter_gather.hpp:48
ck_tile::tile_window_with_tile_dstr_base::AdaptorTopIndex
array< index_t, NDimWindowAdaptorTop > AdaptorTopIndex
Definition tile_window_base.hpp:105

Member Function Documentation

◆ get_num_of_access()

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
CK_TILE_DEVICE constexpr auto ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::get_num_of_access ( ) const
inlineconstexpr

◆ get_tile_distribution()

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
CK_TILE_DEVICE constexpr auto ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::get_tile_distribution ( ) const
inlineconstexpr

◆ get_window_adaptor_ys_safe_vector_length_strides()

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
CK_TILE_DEVICE constexpr auto ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::get_window_adaptor_ys_safe_vector_length_strides ( )
inlinestaticconstexpr

◆ has_static_tile_distribution()

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
CK_TILE_DEVICE constexpr bool ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::has_static_tile_distribution ( )
inlinestaticconstexpr

◆ init_raw()

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
CK_TILE_HOST_DEVICE void ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::init_raw ( )
inline

◆ move_window_adaptor_and_bottom_tensor_thread_coordinate()

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
template<typename ATopIndex>
CK_TILE_DEVICE void ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::move_window_adaptor_and_bottom_tensor_thread_coordinate ( WindowAdaptorCoord & window_adaptor_thread_coord,
BottomTensorCoord & bottom_tensor_thread_coord,
const ATopIndex & idx_diff_adaptor_top ) const
inline

Member Data Documentation

◆ NDimP

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
index_t ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::NDimP = TileDstr::get_num_of_dimension_p()
staticconstexpr

◆ NDimWindowAdaptorTop

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
index_t ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::NDimWindowAdaptorTop = WindowAdaptor::get_num_of_top_dimension()
staticconstexpr

◆ NDimY

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
index_t ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::NDimY = TileDstr::get_num_of_dimension_y()
staticconstexpr

◆ tile_dstr_

template<typename TileWindowType_, typename BottomTensorView_, typename WindowLengths_, typename StaticTileDistribution_>
TileDstr ck_tile::tile_window_with_tile_dstr_base< TileWindowType_, BottomTensorView_, WindowLengths_, StaticTileDistribution_ >::tile_dstr_
The documentation for this struct was generated from the following file: