mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
41 lines
999 B
C
41 lines
999 B
C
#ifndef GEMM_H
|
|
#define GEMM_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
|
|
#ifndef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
void gemm_ongpu(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);
|
|
#ifndef __cplusplus
|
|
}
|
|
#endif
|
|
#endif
|
|
#endif
|