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.c | 27 ++++++++++++++++++++++++---
1 files changed, 24 insertions(+), 3 deletions(-)
diff --git a/src/opencl.c b/src/opencl.c
index bcc0f09..50a03a6 100644
--- a/src/opencl.c
+++ b/src/opencl.c
@@ -1,11 +1,16 @@
#ifdef GPU
-#include "opencl.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>
+#ifdef CLBLAS
+#include <clBLAS.h>
+#endif
+
+#include "opencl.h"
+#include "utils.h"
cl_info cl = {0};
@@ -79,9 +84,9 @@
}
int index = getpid()%num_devices;
+ index = 0;
printf("%d rand, %d devices, %d index\n", getpid(), num_devices, index);
- //info.device = devices[index];
- info.device = devices[0];
+ info.device = devices[index];
fprintf(stderr, "Found %d device(s)\n", num_devices);
check_error(info);
@@ -93,6 +98,10 @@
check_error(info);
info.queue = clCreateCommandQueue(info.context, info.device, 0, &info.error);
check_error(info);
+ #ifdef CLBLAS
+ info.error = clblasSetup();
+ #endif
+ check_error(info);
info.initialized = 1;
return info;
}
@@ -103,6 +112,7 @@
char src[64*1024];
memset(src, 0, 64*1024);
FILE *fil=fopen(filename,"r");
+ if(fil == 0) file_error(filename);
srcsize=fread(src, sizeof src, 1, fil);
fclose(fil);
const char *srcptr[]={src};
@@ -124,6 +134,7 @@
void cl_setup()
{
if(!cl.initialized){
+ printf("initializing\n");
cl = cl_init();
}
}
@@ -178,4 +189,14 @@
return mem;
}
+cl_mem cl_make_int_array(int *x, int n)
+{
+ cl_setup();
+ cl_mem mem = clCreateBuffer(cl.context,
+ CL_MEM_READ_WRITE|CL_MEM_COPY_HOST_PTR,
+ sizeof(int)*n, x, &cl.error);
+ check_error(cl);
+ return mem;
+}
+
#endif
--
Gitblit v1.10.0