Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
#include <gemm_global_tile.h>
Public Types | |
typedef Shape< Threads_::kD, Threads_::kH *Threads_::kW/Tile_::kW, Tile_::kW, 1 > | Threads |
typedef Shape<Threads_::kD, Threads_::kH * Threads_::kW / Tile_::kW, Tile_::kW, 1> cutlass::gemm::ReshapeThreads< Tile_, Threads_, true >::Threads |