From 787d5345609459f21fd65d2d8b4fcd55201e21a1 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Mon, 13 Oct 2014 07:31:10 +0000 Subject: [PATCH] Convolutional working on GPU --- 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