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 (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_bin (int M, int N, int K, float ALPHA, char *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)
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)
float *	random_matrix (int rows, int cols)
void	time_random_matrix (int TA, int TB, int m, int k, int n)

---

Function Documentation

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.

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.

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 141 of file gemm.c.

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.

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 90 of file gemm.c.

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 107 of file gemm.c.

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 123 of file gemm.c.

float* random_matrix	(	int	rows,
		int	cols
	)

Definition at line 30 of file gemm.c.

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

Definition at line 40 of file gemm.c.