From 158bb1bee9951875dbe3474d84c6663431e18301 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 21 Oct 2014 21:49:18 +0000
Subject: [PATCH] softmax on gpu
---
src/mini_blas.c | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/mini_blas.c b/src/mini_blas.c
index 0227b37..4d92971 100644
--- a/src/mini_blas.c
+++ b/src/mini_blas.c
@@ -41,7 +41,7 @@
float *c = random_matrix(m,n);
int i;
clock_t start = clock(), end;
- for(i = 0; i<1000; ++i){
+ for(i = 0; i<10; ++i){
gemm_cpu(TA,TB,m,n,k,1,a,lda,b,ldb,1,c,n);
}
end = clock();
--
Gitblit v1.10.0