Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
#include <igemm_epilogue.h>
Public Types | |
typedef Fragment< float, kElements_ > | InputFragment |
The input fragment. More... | |
typedef Fragment< int8_t, kElements_ > | OutputFragment |
The output fragment. More... | |
Public Member Functions | |
CUTLASS_DEVICE | IgemmFloatToInt8Converter () |
Ctor. More... | |
CUTLASS_DEVICE void | transform (InputFragment const &src, OutputFragment &dst) |
Transform a fragment. More... | |
template<typename Fragment_ > | |
CUTLASS_DEVICE void | transform (Fragment_ const &src, int offset, OutputFragment &dst) |
Transform a fragment. More... | |
typedef Fragment<float, kElements_> cutlass::gemm::IgemmFloatToInt8Converter< kElements_ >::InputFragment |
typedef Fragment<int8_t, kElements_> cutlass::gemm::IgemmFloatToInt8Converter< kElements_ >::OutputFragment |
|
inline |
|
inline |
|
inline |