From 741ada451cc7fee1b9a4c3deaec6af87a2af7497 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 17 Aug 2015 16:31:25 +0000
Subject: [PATCH] Added Darknet reference model

---
 src/blas.h |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/src/blas.h b/src/blas.h
index 90f1a9b..9909925 100644
--- a/src/blas.h
+++ b/src/blas.h
@@ -22,7 +22,7 @@
 void copy_ongpu(int N, float * X, int INCX, float * Y, int INCY);
 void copy_ongpu_offset(int N, float * X, int OFFX, int INCX, float * Y, int OFFY, int INCY);
 void scal_ongpu(int N, float ALPHA, float * X, int INCX);
-void mask_ongpu(int N, float * X, float * mask);
+void mask_ongpu(int N, float * X, float mask_num, float * mask);
 void const_ongpu(int N, float ALPHA, float *X, int INCX);
 void pow_ongpu(int N, float ALPHA, float *X, int INCX, float *Y, int INCY);
 void mul_ongpu(int N, float *X, int INCX, float *Y, int INCY);

--
Gitblit v1.10.0