CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers

gemm → epilogue Relation

File in include/cutlass/gemmIncludes file in include/cutlass/epilogue
kernel / default_gemm.hthreadblock / default_epilogue_simt.h
kernel / default_gemm.hthreadblock / default_epilogue_tensor_op.h
kernel / default_gemm.hthreadblock / default_epilogue_volta_tensor_op.h
kernel / default_gemm.hthreadblock / epilogue.h
kernel / default_gemm.hthread / linear_combination.h
device / default_gemm_configuration.hthread / linear_combination.h
device / default_gemm_configuration.hthread / linear_combination_clamp.h
device / device/gemm_splitk_parallel.hthread / conversion_op.h