gemm_internal.cpp File Reference

#include "compute/matmul/gemm_internal.h"

Include dependency graph for gemm_internal.cpp:

Functions
template<typename T >
bool	magmadnn::internal::gemm_check (Tensor< T > *A, Tensor< T > *B, Tensor< T > *C, unsigned int &M, unsigned int &N, unsigned int &K)
template<>
void	magmadnn::internal::gemm_full (int alpha, Tensor< int > *A, Tensor< int > *B, int beta, Tensor< int > *C)
template<>
void	magmadnn::internal::gemm_full (float alpha, Tensor< float > *A, Tensor< float > *B, float beta, Tensor< float > *C)
template<>
void	magmadnn::internal::gemm_full (double alpha, Tensor< double > *A, Tensor< double > *B, double beta, Tensor< double > *C)

Detailed Description

Author

Daniel Nichols

Version

1.0

Date

2019-02-22

Copyright

Copyright (c) 2019

Function Documentation

gemm_check()

template<typename T >

bool magmadnn::internal::gemm_check	(	Tensor< T > *	A,
		Tensor< T > *	B,
		Tensor< T > *	C,
		unsigned int &	M,
		unsigned int &	N,
		unsigned int &	K
	)

Returns true if A, B, C are valid parameters for gemm_full. It also sets M, N, K to A.get_shape(0), B.get_shape(1), and A.get_shape(1), respectively.

Template Parameters

T

Parameters

A
B
C
M
N
K

Returns

true

false

Here is the call graph for this function:

Here is the caller graph for this function: