CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
Public Member Functions | List of all members
cutlass::reference::device::Gemm< ElementA, LayoutA, ElementB, LayoutB, ElementC, LayoutC, ScalarType, AccumulatorType, arch::OpMultiplyAddSaturate > Struct Template Reference

Partial specialization for multiply-add-saturate.

#include <gemm.h>

Public Member Functions

void operator() (gemm::GemmCoord problem_size, ScalarType alpha, TensorRef< ElementA, LayoutA > tensor_a, TensorRef< ElementB, LayoutB > tensor_b, ScalarType beta, TensorRef< ElementC, LayoutC > tensor_c, AccumulatorType initial_accum=AccumulatorType(0))
 
void operator() (gemm::GemmCoord problem_size, ScalarType alpha, TensorRef< ElementA, LayoutA > tensor_a, TensorRef< ElementB, LayoutB > tensor_b, ScalarType beta, TensorRef< ElementC, LayoutC > tensor_c, TensorRef< ElementC, LayoutC > tensor_d, AccumulatorType initial_accum=AccumulatorType(0))
 

Member Function Documentation

template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementC , typename LayoutC , typename ScalarType , typename AccumulatorType >
void cutlass::reference::device::Gemm< ElementA, LayoutA, ElementB, LayoutB, ElementC, LayoutC, ScalarType, AccumulatorType, arch::OpMultiplyAddSaturate >::operator() ( gemm::GemmCoord  problem_size,
ScalarType  alpha,
TensorRef< ElementA, LayoutA >  tensor_a,
TensorRef< ElementB, LayoutB >  tensor_b,
ScalarType  beta,
TensorRef< ElementC, LayoutC >  tensor_c,
AccumulatorType  initial_accum = AccumulatorType(0) 
)
inline
template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementC , typename LayoutC , typename ScalarType , typename AccumulatorType >
void cutlass::reference::device::Gemm< ElementA, LayoutA, ElementB, LayoutB, ElementC, LayoutC, ScalarType, AccumulatorType, arch::OpMultiplyAddSaturate >::operator() ( gemm::GemmCoord  problem_size,
ScalarType  alpha,
TensorRef< ElementA, LayoutA >  tensor_a,
TensorRef< ElementB, LayoutB >  tensor_b,
ScalarType  beta,
TensorRef< ElementC, LayoutC >  tensor_c,
TensorRef< ElementC, LayoutC >  tensor_d,
AccumulatorType  initial_accum = AccumulatorType(0) 
)
inline

The documentation for this struct was generated from the following file: