cublas.h File Reference

Functions
void	cuspmul (curmat &y, const cusp &A, const curmat &x, long ncol, char trans, Real alpha, stream_t &stream)
template<typename T >
cublasStatus_t	cublasGdgmm (cublasHandle_t handle, cublasSideMode_t mode, int m, int n, const T *A, int lda, const T *x, int incx, T *C, int ldc)
template<typename T >
void	cudgmm (NumArray< T, Gpu >C, NumArray< T, Gpu > A, NumArray< T, Gpu >Sd, stream_t &stream)
template<typename T >
cublasStatus_t	cublasGgemm (cublasHandle_t handle, cublasOperation_t transa, cublasOperation_t transb, int m, int n, int k, const T *alpha, const T *A, int lda, const T *B, int ldb, const T *beta, T *C, int ldc)
template<typename T >
void	cugemm (NumArray< T, Gpu > &C, T beta, const NumArray< T, Gpu > &A, const NumArray< T, Gpu > &B, const char trans[2], T alpha, stream_t &stream)

Function Documentation

cublasGdgmm()

template<typename T >

cublasStatus_t cublasGdgmm	(	cublasHandle_t	handle,
		cublasSideMode_t	mode,
		int	m,
		int	n,
		const T *	A,
		int	lda,
		const T *	x,
		int	incx,
		T *	C,
		int	ldc
	)

< C=A*diag(x)

cudgmm()

template<typename T >

void cudgmm	(	NumArray< T, Gpu >	C,
		NumArray< T, Gpu >	A,
		NumArray< T, Gpu >	Sd,
		stream_t &	stream
	)

C=A*S Sd=diagonal(S)

cublasGgemm()

template<typename T >

cublasStatus_t cublasGgemm	(	cublasHandle_t	handle,
		cublasOperation_t	transa,
		cublasOperation_t	transb,
		int	m,
		int	n,
		int	k,
		const T *	alpha,
		const T *	A,
		int	lda,
		const T *	B,
		int	ldb,
		const T *	beta,
		T *	C,
		int	ldc
	)

C=alpha*op(A)*op(B)+beta*C

cugemm()

template<typename T >

void cugemm	(	NumArray< T, Gpu > &	C,
		T	beta,
		const NumArray< T, Gpu > &	A,
		const NumArray< T, Gpu > &	B,
		const char	trans[2],
		T	alpha,
		stream_t &	stream
	)

Computes C = beta * C + alpha * op(A) * B ;