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