tile_sweeper< DistributedTensor_, F_, UnpacksPerXDim_ > Struct Template Reference

tile_sweeper&lt; DistributedTensor_, F_, UnpacksPerXDim_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::tile_sweeper< DistributedTensor_, F_, UnpacksPerXDim_ > Struct Template Reference
ck_tile::tile_sweeper< DistributedTensor_, F_, UnpacksPerXDim_ > Struct Template Reference

#include <sweep_tile.hpp>

Public Types

using DistributedTensor = remove_cvref_t< DistributedTensor_ >
 
using F = remove_cvref_t< F_ >
 
using UnpacksPerXDim = remove_cvref_t< UnpacksPerXDim_ >
 

Public Member Functions

CK_TILE_HOST_DEVICE tile_sweeper (const F &f_, UnpacksPerXDim={})
 
CK_TILE_HOST_DEVICE tile_sweeper (const DistributedTensor &, const F &f_, UnpacksPerXDim={})
 
CK_TILE_HOST_DEVICE void operator() () const
 
template<index_t i_access>
CK_TILE_HOST_DEVICE void operator() (number< i_access >) const
 

Static Public Member Functions

static constexpr CK_TILE_HOST_DEVICE index_t get_num_of_access ()
 

Public Attributes

F f
 

Member Typedef Documentation

◆ DistributedTensor

template<typename DistributedTensor_ , typename F_ , typename UnpacksPerXDim_ = typename uniform_sequence_gen<DistributedTensor_::get_num_of_dimension(), 1>::type>
using ck_tile::tile_sweeper< DistributedTensor_, F_, UnpacksPerXDim_ >::DistributedTensor = remove_cvref_t<DistributedTensor_>

◆ F

template<typename DistributedTensor_ , typename F_ , typename UnpacksPerXDim_ = typename uniform_sequence_gen<DistributedTensor_::get_num_of_dimension(), 1>::type>
using ck_tile::tile_sweeper< DistributedTensor_, F_, UnpacksPerXDim_ >::F = remove_cvref_t<F_>

◆ UnpacksPerXDim

template<typename DistributedTensor_ , typename F_ , typename UnpacksPerXDim_ = typename uniform_sequence_gen<DistributedTensor_::get_num_of_dimension(), 1>::type>
using ck_tile::tile_sweeper< DistributedTensor_, F_, UnpacksPerXDim_ >::UnpacksPerXDim = remove_cvref_t<UnpacksPerXDim_>

Constructor & Destructor Documentation

◆ tile_sweeper() [1/2]

template<typename DistributedTensor_ , typename F_ , typename UnpacksPerXDim_ = typename uniform_sequence_gen<DistributedTensor_::get_num_of_dimension(), 1>::type>
CK_TILE_HOST_DEVICE ck_tile::tile_sweeper< DistributedTensor_, F_, UnpacksPerXDim_ >::tile_sweeper ( const F f_,
UnpacksPerXDim  = {} 
)
inline

◆ tile_sweeper() [2/2]

template<typename DistributedTensor_ , typename F_ , typename UnpacksPerXDim_ = typename uniform_sequence_gen<DistributedTensor_::get_num_of_dimension(), 1>::type>
CK_TILE_HOST_DEVICE ck_tile::tile_sweeper< DistributedTensor_, F_, UnpacksPerXDim_ >::tile_sweeper ( const DistributedTensor ,
const F f_,
UnpacksPerXDim  = {} 
)
inline

Member Function Documentation

◆ get_num_of_access()

template<typename DistributedTensor_ , typename F_ , typename UnpacksPerXDim_ = typename uniform_sequence_gen<DistributedTensor_::get_num_of_dimension(), 1>::type>
static constexpr CK_TILE_HOST_DEVICE index_t ck_tile::tile_sweeper< DistributedTensor_, F_, UnpacksPerXDim_ >::get_num_of_access ( )
inlinestaticconstexpr

◆ operator()() [1/2]

template<typename DistributedTensor_ , typename F_ , typename UnpacksPerXDim_ = typename uniform_sequence_gen<DistributedTensor_::get_num_of_dimension(), 1>::type>
CK_TILE_HOST_DEVICE void ck_tile::tile_sweeper< DistributedTensor_, F_, UnpacksPerXDim_ >::operator() ( ) const
inline

◆ operator()() [2/2]

template<typename DistributedTensor_ , typename F_ , typename UnpacksPerXDim_ = typename uniform_sequence_gen<DistributedTensor_::get_num_of_dimension(), 1>::type>
template<index_t i_access>
CK_TILE_HOST_DEVICE void ck_tile::tile_sweeper< DistributedTensor_, F_, UnpacksPerXDim_ >::operator() ( number< i_access >  ) const
inline

Member Data Documentation

◆ f

template<typename DistributedTensor_ , typename F_ , typename UnpacksPerXDim_ = typename uniform_sequence_gen<DistributedTensor_::get_num_of_dimension(), 1>::type>
F ck_tile::tile_sweeper< DistributedTensor_, F_, UnpacksPerXDim_ >::f

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/core/tensor/sweep_tile.hpp