You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
964 B
35 lines
964 B
#pragma once |
|
#ifndef GEMM_H |
|
#define GEMM_H |
|
#include "define_inc.h" |
|
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(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_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); |
|
|
|
#ifdef GPU |
|
void gemm_gpu(int TA, int TB, int M, int N, int K, float ALPHA, |
|
float *A_gpu, int lda, |
|
float *B_gpu, int ldb, |
|
float BETA, |
|
float *C_gpu, int ldc); |
|
|
|
void gemm_gpu(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); |
|
#endif |
|
#endif
|
|
|