gemm.c File Reference

#include "gemm.h"
#include "utils.h"
#include "cuda.h"
#include <stdlib.h>
#include <stdio.h>
#include <math.h>

Include dependency graph for gemm.c:

Go to the source code of this file.

Functions
void	gemm_bin (int M, int N, int K, float ALPHA, char *A, int lda, float *B, int ldb, float *C, int ldc)
float *	random_matrix (int rows, int cols)
void	time_random_matrix (int TA, int TB, int m, int k, int n)
void	gemm (int TA, int TB, int M, int N, int K, float ALPHA, float *A, int lda, float *B, int ldb, float BETA, float *C, int ldc)
void	gemm_nn (int M, int N, int K, float ALPHA, float *A, int lda, float *B, int ldb, float *C, int ldc)
void	gemm_nt (int M, int N, int K, float ALPHA, float *A, int lda, float *B, int ldb, float *C, int ldc)
void	gemm_tn (int M, int N, int K, float ALPHA, float *A, int lda, float *B, int ldb, float *C, int ldc)
void	gemm_tt (int M, int N, int K, float ALPHA, float *A, int lda, float *B, int ldb, float *C, int ldc)
void	gemm_cpu (int TA, int TB, int M, int N, int K, float ALPHA, float *A, int lda, float *B, int ldb, float BETA, float *C, int ldc)

Function Documentation

gemm()

void gemm	(	int	TA,
		int	TB,
		int	M,
		int	N,
		int	K,
		float	ALPHA,
		float *	A,
		int	lda,
		float *	B,
		int	ldb,
		float	BETA,
		float *	C,
		int	ldc
	)

Definition at line 65 of file gemm.c.

gemm_bin()

void gemm_bin	(	int	M,
		int	N,
		int	K,
		float	ALPHA,
		char *	A,
		int	lda,
		float *	B,
		int	ldb,
		float *	C,
		int	ldc
	)

Definition at line 8 of file gemm.c.

gemm_cpu()

void gemm_cpu	(	int	TA,
		int	TB,
		int	M,
		int	N,
		int	K,
		float	ALPHA,
		float *	A,
		int	lda,
		float *	B,
		int	ldb,
		float	BETA,
		float *	C,
		int	ldc
	)

Definition at line 145 of file gemm.c.

gemm_nn()

void gemm_nn	(	int	M,
		int	N,
		int	K,
		float	ALPHA,
		float *	A,
		int	lda,
		float *	B,
		int	ldb,
		float *	C,
		int	ldc
	)

Definition at line 74 of file gemm.c.

gemm_nt()

void gemm_nt	(	int	M,
		int	N,
		int	K,
		float	ALPHA,
		float *	A,
		int	lda,
		float *	B,
		int	ldb,
		float *	C,
		int	ldc
	)

Definition at line 91 of file gemm.c.

gemm_tn()

void gemm_tn	(	int	M,
		int	N,
		int	K,
		float	ALPHA,
		float *	A,
		int	lda,
		float *	B,
		int	ldb,
		float *	C,
		int	ldc
	)

Definition at line 109 of file gemm.c.

gemm_tt()

void gemm_tt	(	int	M,
		int	N,
		int	K,
		float	ALPHA,
		float *	A,
		int	lda,
		float *	B,
		int	ldb,
		float *	C,
		int	ldc
	)

Definition at line 126 of file gemm.c.

random_matrix()

float* random_matrix	(	int	rows,
		int	cols
	)

Definition at line 30 of file gemm.c.

time_random_matrix()

void time_random_matrix	(	int	TA,
		int	TB,
		int	m,
		int	k,
		int	n
	)

Definition at line 40 of file gemm.c.