From edbccdfcaf46f11e631afe98796f3e6e170da5d0 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sun, 26 Oct 2014 05:04:34 +0000
Subject: [PATCH] Maybe something changed?
---
src/gemm.c | 12 ++++++------
1 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/src/gemm.c b/src/gemm.c
index 2e53b31..63c2950 100644
--- a/src/gemm.c
+++ b/src/gemm.c
@@ -176,13 +176,13 @@
float BETA,
cl_mem C_gpu, int ldc)
{
- /*
+/*
cl_setup();
cl_command_queue queue = cl.queue;
cl_event event;
cl.error = clblasSgemm(clblasRowMajor, TA?clblasTrans:clblasNoTrans, TB?clblasTrans:clblasNoTrans,M, N, K,ALPHA, A_gpu, 0, lda,B_gpu, 0, ldb,BETA, C_gpu, 0, ldc,1, &queue, 0, NULL, &event);
- */
+*/
gemm_ongpu_new(TA, TB, M, N, K, ALPHA, A_gpu, lda, B_gpu, ldb, BETA, C_gpu, ldc);
}
@@ -327,7 +327,7 @@
void time_ongpu(int TA, int TB, int m, int k, int n)
{
- int iter = 100;
+ int iter = 128;
float *a = random_matrix(m,k);
float *b = random_matrix(k,n);
@@ -345,10 +345,10 @@
for(i = 0; i<iter; ++i){
gemm_ongpu(TA,TB,m,n,k,1,a_cl,lda,b_cl,ldb,1,c_cl,n);
}
- int flop = m*n*(2*k+3)*iter;
- float gflop = flop/pow(10., 9);
+ double flop = m*n*(2.*k+3.)*iter;
+ double gflop = flop/pow(10., 9);
end = clock();
- float seconds = sec(end-start);
+ double seconds = sec(end-start);
printf("Matrix Multiplication %dx%d * %dx%d, TA=%d, TB=%d: %lf s, %lf GFLOPS\n",m,k,k,n, TA, TB, seconds, gflop/seconds);
clReleaseMemObject(a_cl);
clReleaseMemObject(b_cl);
--
Gitblit v1.10.0