CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers

reduction → layout Relation

File in include/cutlass/reductionIncludes file in include/cutlass/layout
kernel / reduce_split_k.hlayout/matrix.h