From b77a8f39874c05a1ed0cabd8d85134126ac2da47 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 28 Nov 2014 18:38:26 +0000
Subject: [PATCH] stable
---
src/axpy.cl | 6 ++++++
1 files changed, 6 insertions(+), 0 deletions(-)
diff --git a/src/axpy.cl b/src/axpy.cl
index 901a826..396389d 100644
--- a/src/axpy.cl
+++ b/src/axpy.cl
@@ -10,6 +10,12 @@
X[i*INCX] *= ALPHA;
}
+__kernel void mask(int n, __global float *x, __global float *mask, int mod)
+{
+ int i = get_global_id(0);
+ x[i] = (mask[(i/mod)*mod]) ? x[i] : 0;
+}
+
__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);
--
Gitblit v1.10.0