CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Shared memory storage structure.
#include <gemm_splitk_parallel.h>
Public Attributes | |
Mma::SharedStorage | main_loop |
Epilogue::SharedStorage | epilogue |
Epilogue::SharedStorage cutlass::gemm::kernel::GemmSplitKParallel< Mma_, Epilogue_, ThreadblockSwizzle_ >::SharedStorage::epilogue |
Mma::SharedStorage cutlass::gemm::kernel::GemmSplitKParallel< Mma_, Epilogue_, ThreadblockSwizzle_ >::SharedStorage::main_loop |