CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers

gemm → layout Relation

File in include/cutlass/gemmIncludes file in include/cutlass/layout
kernel / default_gemm.hlayout/matrix.h
threadblock / default_gemv_core.hlayout/matrix.h
threadblock / default_mma_core_sm50.hlayout/matrix.h
threadblock / default_mma_core_sm70.htensor_op_multiplicand_sm70.h
threadblock / default_mma_core_sm75.htensor_op_multiplicand_sm75.h
thread / gemm/thread/mma_sm50.hlayout/matrix.h
thread / gemm/thread/mma_sm60.hlayout/matrix.h
thread / gemm/thread/mma_sm61.hlayout/matrix.h
warp / mma_simt_tile_iterator.hlayout/matrix.h
warp / mma_tensor_op_tile_iterator.hlayout/matrix.h
warp / mma_tensor_op_tile_iterator.hpitch_linear.h
warp / mma_tensor_op_tile_iterator.htensor.h
warp / mma_tensor_op_tile_iterator.htensor_op_multiplicand_sm75.h
warp / mma_tensor_op_tile_iterator_sm70.hlayout/matrix.h
warp / mma_tensor_op_tile_iterator_sm70.hpitch_linear.h
warp / mma_tensor_op_tile_iterator_sm70.htensor_op_multiplicand_sm70.h