Marlin Mma#
-
namespace MARLIN_NAMESPACE_NAME
Functions
- template<trt_edgellm::marlin_dtypes::ScalarTypeId type_id, int k_size = 16> inline __device__ void mma (typename MarlinScalarType< type_id >::FragA const &a_frag, typename MarlinScalarType< type_id >::FragB const &frag_b, typename MarlinScalarType< type_id >::FragC &frag_c, int idx=0)
- template<trt_edgellm::marlin_dtypes::ScalarTypeId type_id, int k_size = 16> inline __device__ void mma_trans (typename MarlinScalarType< type_id >::FragA const &a_frag, typename MarlinScalarType< type_id >::FragB const &frag_b, typename MarlinScalarType< type_id >::FragB const &frag_b2, typename MarlinScalarType< type_id >::FragC &frag_c)