Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
The storage in shared memory.
#include <gemm_traits.h>
Public Attributes | |
MainLoopSharedStorage | main_loop |
Epilogue::SharedStorage | epilogue |
Epilogue::SharedStorage cutlass::gemm::GemmTraits< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::SharedStorage::epilogue |
MainLoopSharedStorage cutlass::gemm::GemmTraits< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::SharedStorage::main_loop |