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