From aa5996d58e68edfbefe51061856aecd549dd09c4 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 13 Jan 2015 01:27:08 +0000
Subject: [PATCH] Faster

---
 src/opencl.h |    6 +++++-
 1 files changed, 5 insertions(+), 1 deletions(-)

diff --git a/src/opencl.h b/src/opencl.h
index 9cf3acd..ceb6e56 100644
--- a/src/opencl.h
+++ b/src/opencl.h
@@ -1,12 +1,14 @@
-#ifdef GPU
 #ifndef OPENCL_H
 #define OPENCL_H
+extern int gpu_index;
+#ifdef GPU
 #ifdef __APPLE__
 #include <OpenCL/opencl.h>
 #else
 #include <CL/cl.h>
 #endif
 
+
 typedef struct {
     int initialized;
     cl_int error;
@@ -24,7 +26,9 @@
 void cl_read_array(cl_mem mem, float *x, int n);
 void cl_write_array(cl_mem mem, float *x, int n);
 cl_mem cl_make_array(float *x, int n);
+cl_mem cl_make_int_array(int *x, int n);
 void cl_copy_array(cl_mem src, cl_mem dst, int n);
 cl_mem cl_sub_array(cl_mem src, int offset, int size);
+float cl_checksum(cl_mem mem, int n);
 #endif
 #endif

--
Gitblit v1.10.0