Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
Classes | Namespaces | Functions
gemm/threadblock_swizzle.h File Reference

Defies functors for mapping blockIdx to partitions of the GEMM computation. More...

#include "cutlass/coord.h"
#include "cutlass/gemm/gemm_coord.h"

Go to the source code of this file.

Classes

struct  cutlass::gemm::swizzleDirection
 
struct  cutlass::gemm::IdentityBlockSwizzle
 
struct  cutlass::gemm::ColumnMajorBlockSwizzle< groupCols, swDirection >
 
struct  cutlass::gemm::RowMajorBlockSwizzle< groupRows, swDirection >
 

Namespaces

 cutlass
 
 cutlass::gemm
 

Functions

template<enum swizzleDirection::Kind >
CUTLASS_DEVICE int cutlass::gemm::getLinearIdx (int groups)
 
template<>
CUTLASS_DEVICE int cutlass::gemm::getLinearIdx< swizzleDirection::Boustrophedon > (int groups)