Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
Public Member Functions | List of all members
cutlass::gemm::RowMajorBlockSwizzle< groupRows, swDirection > Struct Template Reference

#include <threadblock_swizzle.h>

Public Member Functions

CUTLASS_HOST_DEVICE RowMajorBlockSwizzle ()
 Ctor. More...
 
CUTLASS_DEVICE dim3 swizzle ()
 Swizzle the block index. More...
 
CUTLASS_HOST_DEVICE dim3 get_grid_layout (GemmCoord const &problem_size, Coord< 3 > const &OutputTile)
 
CUTLASS_DEVICE Coord< 3 > get_threadblock_offset (Coord< 3 > const &OutputTile)
 
CUTLASS_DEVICE int get_batch_id ()
 
CUTLASS_DEVICE bool is_last_partition ()
 check if at the last partition More...
 
CUTLASS_DEVICE Coord< 3 > get_threadblock_bounds (GemmCoord const &problem_size, int partitionK_range)
 

Constructor & Destructor Documentation

◆ RowMajorBlockSwizzle()

template<int groupRows, enum swizzleDirection::Kind swDirection>
CUTLASS_HOST_DEVICE cutlass::gemm::RowMajorBlockSwizzle< groupRows, swDirection >::RowMajorBlockSwizzle ( )
inline

Member Function Documentation

◆ get_batch_id()

template<int groupRows, enum swizzleDirection::Kind swDirection>
CUTLASS_DEVICE int cutlass::gemm::RowMajorBlockSwizzle< groupRows, swDirection >::get_batch_id ( )
inline

◆ get_grid_layout()

template<int groupRows, enum swizzleDirection::Kind swDirection>
CUTLASS_HOST_DEVICE dim3 cutlass::gemm::RowMajorBlockSwizzle< groupRows, swDirection >::get_grid_layout ( GemmCoord const &  problem_size,
Coord< 3 > const &  OutputTile 
)
inline

◆ get_threadblock_bounds()

template<int groupRows, enum swizzleDirection::Kind swDirection>
CUTLASS_DEVICE Coord<3> cutlass::gemm::RowMajorBlockSwizzle< groupRows, swDirection >::get_threadblock_bounds ( GemmCoord const &  problem_size,
int  partitionK_range 
)
inline

◆ get_threadblock_offset()

template<int groupRows, enum swizzleDirection::Kind swDirection>
CUTLASS_DEVICE Coord<3> cutlass::gemm::RowMajorBlockSwizzle< groupRows, swDirection >::get_threadblock_offset ( Coord< 3 > const &  OutputTile)
inline

◆ is_last_partition()

template<int groupRows, enum swizzleDirection::Kind swDirection>
CUTLASS_DEVICE bool cutlass::gemm::RowMajorBlockSwizzle< groupRows, swDirection >::is_last_partition ( )
inline

◆ swizzle()

template<int groupRows, enum swizzleDirection::Kind swDirection>
CUTLASS_DEVICE dim3 cutlass::gemm::RowMajorBlockSwizzle< groupRows, swDirection >::swizzle ( )
inline

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