00001 #ifndef GEMM_H 00002 #define GEMM_H 00003 00004 void gemm_bin(int M, int N, int K, float ALPHA, 00005 char *A, int lda, 00006 float *B, int ldb, 00007 float *C, int ldc); 00008 00009 void gemm(int TA, int TB, int M, int N, int K, float ALPHA, 00010 float *A, int lda, 00011 float *B, int ldb, 00012 float BETA, 00013 float *C, int ldc); 00014 00015 void gemm_cpu(int TA, int TB, int M, int N, int K, float ALPHA, 00016 float *A, int lda, 00017 float *B, int ldb, 00018 float BETA, 00019 float *C, int ldc); 00020 00021 #ifdef GPU 00022 void gemm_ongpu(int TA, int TB, int M, int N, int K, float ALPHA, 00023 float *A_gpu, int lda, 00024 float *B_gpu, int ldb, 00025 float BETA, 00026 float *C_gpu, int ldc); 00027 00028 void gemm_gpu(int TA, int TB, int M, int N, int K, float ALPHA, 00029 float *A, int lda, 00030 float *B, int ldb, 00031 float BETA, 00032 float *C, int ldc); 00033 #endif 00034 #endif