tile_window.hpp File Reference

tile_window.hpp File Reference#

Composable Kernel: tile_window.hpp File Reference
tile_window.hpp File Reference

Go to the source code of this file.

Classes

struct  ck_tile::tile_window_with_static_distribution< BottomTensorView_, WindowLengths_, StaticTileDistribution_, NumCoord >
 This class provides tile (windowed) view and access to the device memory. More...
struct  ck_tile::tile_window_with_static_lengths< BottomTensorView_, WindowLengths_ >
 This class provides description of tile windowed view on the device memory. More...
struct  ck_tile::is_tile_window_with_static_distribution< T >
 Type trait to determine if a type is a tile window with static distribution. More...
struct  ck_tile::is_tile_window_with_static_distribution< tile_window_with_static_distribution< BottomTensorView_, WindowLengths_, StaticTileDistribution_, NumCoord > >
 Specialization for tile_window_with_static_distribution to evaluate to true_type. More...
struct  ck_tile::is_tile_window_with_static_lengths< T >
 Type trait to determine if a type is a tile window with static lengths. More...
struct  ck_tile::is_tile_window_with_static_lengths< tile_window_with_static_lengths< BottomTensorView_, WindowLengths_ > >
 Specialization for tile_window_with_static_lengths to evaluate to true_type. More...

Namespaces

namespace  ck_tile

Functions

template<typename TensorView_, typename WindowLengths_, typename StaticTileDistribution_, index_t NumCoord = 1>
CK_TILE_DEVICE constexpr auto ck_tile::make_tile_window (const TensorView_ &tensor_view, const WindowLengths_ &window_lengths, const multi_index< TensorView_::get_num_of_dimension()> &origin, const StaticTileDistribution_ &tile_distribution, number< NumCoord >={})
template<typename TensorView_, typename WindowLengths_, typename StaticTileDistribution_, index_t NumCoord = 1>
CK_TILE_DEVICE auto ck_tile::make_tile_window_raw (const TensorView_ &tensor_view, const WindowLengths_ &window_lengths, const multi_index< TensorView_::get_num_of_dimension()> &origin, const StaticTileDistribution_ &tile_distribution, number< NumCoord >={})
template<typename TensorView_, typename WindowLengths_, typename StaticTileDistribution_, index_t NumCoord>
CK_TILE_DEVICE void ck_tile::move_tile_window (tile_window_with_static_distribution< TensorView_, WindowLengths_, StaticTileDistribution_, NumCoord > &window, const typename tile_window_with_static_distribution< TensorView_, WindowLengths_, StaticTileDistribution_, NumCoord >::BottomTensorIndex &step)
template<typename TensorView_, typename WindowLengths_, typename StaticTileDistribution_, index_t NumCoord>
CK_TILE_DEVICE void ck_tile::move_tile_window (tuple< tile_window_with_static_distribution< TensorView_, WindowLengths_, StaticTileDistribution_, NumCoord > > &window, const typename tile_window_with_static_distribution< TensorView_, WindowLengths_, StaticTileDistribution_, NumCoord >::BottomTensorIndex &step)
template<typename TileWindowWithStaticDistributionType, typename StepType, typename std::enable_if_t< is_detected< is_tuple, TileWindowWithStaticDistributionType >::value > * = nullptr>
CK_TILE_DEVICE void ck_tile::move_tile_window (TileWindowWithStaticDistributionType &window, StepType &step)
template<typename TensorView_, typename WindowLengths_>
CK_TILE_DEVICE constexpr auto ck_tile::make_tile_window (const TensorView_ &tensor_view, const WindowLengths_ &window_lengths, const multi_index< TensorView_::get_num_of_dimension()> &origin)
template<typename TensorView, typename WindowLengths>
CK_TILE_DEVICE constexpr auto ck_tile::make_tile_window (const tile_window_with_static_lengths< TensorView, WindowLengths > &tile_window, const multi_index< TensorView::get_num_of_dimension()> &origin)
template<typename TensorView, typename WindowLengths, typename StaticTileDistribution>
CK_TILE_DEVICE constexpr auto ck_tile::make_tile_window (const tile_window_with_static_lengths< TensorView, WindowLengths > &tile_window, const multi_index< TensorView::get_num_of_dimension()> &origin, const StaticTileDistribution &tile_distribution)
template<typename TensorView, typename WindowLengths, typename StaticTileDistribution>
CK_TILE_DEVICE constexpr auto ck_tile::make_tile_window (const tile_window_with_static_lengths< TensorView, WindowLengths > &tile_window, const StaticTileDistribution &tile_distribution)
template<typename TensorView, typename WindowLengths, typename StaticTileDistribution>
CK_TILE_DEVICE constexpr auto ck_tile::make_tile_window_raw (const tile_window_with_static_lengths< TensorView, WindowLengths > &tile_window, const StaticTileDistribution &tile_distribution)
template<typename TensorView_, typename WindowLengths_>
CK_TILE_DEVICE void ck_tile::move_tile_window (tile_window_with_static_lengths< TensorView_, WindowLengths_ > &window, const typename tile_window_with_static_lengths< TensorView_, WindowLengths_ >::BottomTensorIndex &step)
template<typename NewTensorView_, typename OldTensorView_, typename WindowLengths_, typename StaticTileDistribution_, index_t NumCoord = 1>
CK_TILE_DEVICE auto ck_tile::replace_bottom_tensor_view (const NewTensorView_ &new_tensor_view, const tile_window_with_static_distribution< OldTensorView_, WindowLengths_, StaticTileDistribution_, NumCoord > &tile_window)
template<typename NewTensorView_, typename OldTensorView_, typename WindowLengths_>
CK_TILE_DEVICE auto ck_tile::replace_bottom_tensor_view (const NewTensorView_ &new_tensor_view, const tile_window_with_static_lengths< OldTensorView_, WindowLengths_ > &tile_window)

Variables

template<typename T>
constexpr bool ck_tile::is_tile_window_with_static_distribution_v
 Helper variable template to check if a type is a tile window with static distribution.
template<typename T>
constexpr bool ck_tile::is_tile_window_with_static_lengths_v
 Helper variable template to check if a type is a tile window with static lengths.