From 354b0cbdcb48a66de1af3aca9fc2f5687e46ee42 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 19 Apr 2014 20:10:59 +0000
Subject: [PATCH] GPU flag in Makefile
---
src/mini_blas.c | 88 -------------------------------------------
1 files changed, 1 insertions(+), 87 deletions(-)
diff --git a/src/mini_blas.c b/src/mini_blas.c
index bac3e22..4c7d3d0 100644
--- a/src/mini_blas.c
+++ b/src/mini_blas.c
@@ -24,7 +24,7 @@
float BETA,
float *C, int ldc)
{
- gpu_gemm( TA, TB, M, N, K, ALPHA,A,lda, B, ldb,BETA,C,ldc);
+ cpu_gemm( TA, TB, M, N, K, ALPHA,A,lda, B, ldb,BETA,C,ldc);
}
void im2row(float *image, int h, int w, int c, int size, int stride, float *matrix)
@@ -155,91 +155,5 @@
time_random_matrix(1,0,1000,100,100);
time_random_matrix(0,1,1000,100,100);
time_random_matrix(1,1,1000,100,100);
-
-
}
-void time_gpu_random_matrix(int TA, int TB, int m, int k, int n)
-{
- float *a;
- if(!TA) a = random_matrix(m,k);
- else a = random_matrix(k,m);
- int lda = (!TA)?k:m;
- float *b;
- if(!TB) b = random_matrix(k,n);
- else b = random_matrix(n,k);
- int ldb = (!TB)?n:k;
-
- float *c = random_matrix(m,n);
- int i;
- clock_t start = clock(), end;
- for(i = 0; i<1000; ++i){
- gpu_gemm(TA,TB,m,n,k,1,a,lda,b,ldb,1,c,n);
- }
- end = clock();
- printf("Matrix Multiplication %dx%d * %dx%d, TA=%d, TB=%d: %lf ms\n",m,k,k,n, TA, TB, (float)(end-start)/CLOCKS_PER_SEC);
- free(a);
- free(b);
- free(c);
-}
-
-void test_gpu_accuracy(int TA, int TB, int m, int k, int n)
-{
- srand(0);
- float *a;
- if(!TA) a = random_matrix(m,k);
- else a = random_matrix(k,m);
- int lda = (!TA)?k:m;
- float *b;
- if(!TB) b = random_matrix(k,n);
- else b = random_matrix(n,k);
- int ldb = (!TB)?n:k;
-
- float *c = random_matrix(m,n);
- float *c_gpu = random_matrix(m,n);
- memset(c, 0, m*n*sizeof(float));
- memset(c_gpu, 0, m*n*sizeof(float));
- int i;
- //pm(m,k,b);
- gpu_gemm(TA,TB,m,n,k,1,a,lda,b,ldb,1,c_gpu,n);
- //pm(m, n, c_gpu);
- cpu_gemm(TA,TB,m,n,k,1,a,lda,b,ldb,1,c,n);
- //pm(m, n, c);
- double sse = 0;
- for(i = 0; i < m*n; ++i) {
- //printf("%f %f\n", c[i], c_gpu[i]);
- sse += pow(c[i]-c_gpu[i], 2);
- }
- printf("Matrix Multiplication %dx%d * %dx%d, TA=%d, TB=%d: %g MSE\n",m,k,k,n, TA, TB, sse/(m*n));
- free(a);
- free(b);
- free(c);
-}
-
-void test_gpu_blas()
-{
- test_gpu_accuracy(0,0,17,10,10);
- test_gpu_accuracy(1,0,17,10,10);
- test_gpu_accuracy(0,1,17,10,10);
- test_gpu_accuracy(1,1,17,10,10);
-
- test_gpu_accuracy(0,0,1000,10,100);
- test_gpu_accuracy(1,0,1000,10,100);
- test_gpu_accuracy(0,1,1000,10,100);
- test_gpu_accuracy(1,1,1000,10,100);
-
- time_gpu_random_matrix(0,0,1000,1000,100);
- time_random_matrix(0,0,1000,1000,100);
-
- time_gpu_random_matrix(0,1,1000,1000,100);
- time_random_matrix(0,1,1000,1000,100);
-
- time_gpu_random_matrix(1,0,1000,1000,100);
- time_random_matrix(1,0,1000,1000,100);
-
- time_gpu_random_matrix(1,1,1000,1000,100);
- time_random_matrix(1,1,1000,1000,100);
-
-}
-
-
--
Gitblit v1.10.0