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/opencl.h | 13 ++++++++++++-
1 files changed, 12 insertions(+), 1 deletions(-)
diff --git a/src/opencl.h b/src/opencl.h
index 59efbae..cdc9e05 100644
--- a/src/opencl.h
+++ b/src/opencl.h
@@ -1,9 +1,13 @@
+#ifdef GPU
+#ifndef OPENCL_H
+#define OPENCL_H
#ifdef __APPLE__
#include <OpenCL/opencl.h>
#else
#include <CL/cl.h>
#endif
+
typedef struct {
int initialized;
cl_int error;
@@ -18,4 +22,11 @@
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);
+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);
+#endif
+#endif
--
Gitblit v1.10.0