1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
| void pm(int M, int N, float *A);
| 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);
| float *random_matrix(int rows, int cols);
| void im2row(float *image, int h, int w, int c, int size, int stride, float *matrix);
| void im2col(float *image, int h, int w, int c, int size, int stride, float *matrix);
| void im2col_cpu(float* data_im, const int channels,
| const int height, const int width, const int ksize, const int stride,
| float* data_col);
| void col2im_cpu(float* data_col, const int channels,
| const int height, const int width, const int ksize, const int stride,
| float* data_im);
| void test_blas();
|
| void gpu_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 cpu_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 test_gpu_blas();
|
|
