CUDA Templates for Linear Algebra Subroutines and Solvers
Public Member Functions | List of all members
cutlass::gemm::threadblock::GemmSplitKIdentityThreadblockSwizzle Struct Reference

Threadblock swizzling function for split-K GEMMs.

#include <threadblock_swizzle.h>

Public Member Functions

CUTLASS_HOST_DEVICE GemmCoord get_tiled_shape (GemmCoord problem_size, GemmCoord tile_size, int partitions) const
 Returns the shape of the problem in units of logical tiles. More...
CUTLASS_HOST_DEVICE dim3 get_grid_shape (GemmCoord tiled_shape) const
 Computes CUDA grid dimensions given a size in units of logical tiles. More...
CUTLASS_DEVICE GemmCoord get_tile_offset () const
 Obtains the threadblock offset (in units of threadblock-scoped tiles) More...

Member Function Documentation

CUTLASS_HOST_DEVICE dim3 cutlass::gemm::threadblock::GemmSplitKIdentityThreadblockSwizzle::get_grid_shape ( GemmCoord  tiled_shape) const
CUTLASS_DEVICE GemmCoord cutlass::gemm::threadblock::GemmSplitKIdentityThreadblockSwizzle::get_tile_offset ( ) const
CUTLASS_HOST_DEVICE GemmCoord cutlass::gemm::threadblock::GemmSplitKIdentityThreadblockSwizzle::get_tiled_shape ( GemmCoord  problem_size,
GemmCoord  tile_size,
int  partitions 
) const

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