From d9f1b0b16edeb59281355a855e18a8be343fc33c Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Fri, 08 Aug 2014 19:04:15 +0000 Subject: [PATCH] probably how maxpool layers should be --- src/opencl.h | 11 ++++++++++- 1 files changed, 10 insertions(+), 1 deletions(-) diff --git a/src/opencl.h b/src/opencl.h index 59efbae..9cf3acd 100644 --- a/src/opencl.h +++ b/src/opencl.h @@ -1,3 +1,6 @@ +#ifdef GPU +#ifndef OPENCL_H +#define OPENCL_H #ifdef __APPLE__ #include <OpenCL/opencl.h> #else @@ -18,4 +21,10 @@ void cl_setup(); void check_error(cl_info info); cl_kernel get_kernel(char *filename, char *kernelname, char *options); - +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); +void cl_copy_array(cl_mem src, cl_mem dst, int n); +cl_mem cl_sub_array(cl_mem src, int offset, int size); +#endif +#endif -- Gitblit v1.10.0