From b13ad6d5fd23f68f506c14ede4282126d893702b Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 05 Nov 2014 22:49:58 +0000
Subject: [PATCH] Can validate on imagenet now
---
src/axpy.cl | 18 ++++++++++++++++++
1 files changed, 18 insertions(+), 0 deletions(-)
diff --git a/src/axpy.cl b/src/axpy.cl
index e69de29..901a826 100644
--- a/src/axpy.cl
+++ b/src/axpy.cl
@@ -0,0 +1,18 @@
+__kernel void axpy(int N, float ALPHA, __global float *X, int OFFX, int INCX, __global float *Y, int OFFY, int INCY)
+{
+ int i = get_global_id(0);
+ Y[OFFY+i*INCY] += ALPHA*X[OFFX+i*INCX];
+}
+
+__kernel void scal(int N, float ALPHA, __global float *X, int INCX)
+{
+ int i = get_global_id(0);
+ X[i*INCX] *= ALPHA;
+}
+
+__kernel void copy(int N, __global float *X, int OFFX, int INCX, __global float *Y, int OFFY, int INCY)
+{
+ int i = get_global_id(0);
+ Y[i*INCY + OFFY] = X[i*INCX + OFFX];
+}
+
--
Gitblit v1.10.0