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 | 23 ++++++++---------------
1 files changed, 8 insertions(+), 15 deletions(-)
diff --git a/src/opencl.c b/src/opencl.c
index fc7310c..50a03a6 100644
--- a/src/opencl.c
+++ b/src/opencl.c
@@ -4,7 +4,10 @@
#include <string.h>
#include <time.h>
#include <unistd.h>
-//#include <clBLAS.h>
+
+#ifdef CLBLAS
+#include <clBLAS.h>
+#endif
#include "opencl.h"
#include "utils.h"
@@ -81,7 +84,7 @@
}
int index = getpid()%num_devices;
- index = 1;
+ index = 0;
printf("%d rand, %d devices, %d index\n", getpid(), num_devices, index);
info.device = devices[index];
fprintf(stderr, "Found %d device(s)\n", num_devices);
@@ -95,24 +98,14 @@
check_error(info);
info.queue = clCreateCommandQueue(info.context, info.device, 0, &info.error);
check_error(info);
- for(i = 0; i < NUM_QUEUES; ++i){
- info.queues[i] = clCreateCommandQueue(info.context, info.device, 0, &info.error);
- check_error(info);
- }
- //info.error = clblasSetup();
+ #ifdef CLBLAS
+ info.error = clblasSetup();
+ #endif
check_error(info);
info.initialized = 1;
return info;
}
-void wait_for_queues()
-{
- int i;
- for(i = 0; i < NUM_QUEUES; ++i){
- clFinish(cl.queues[i]);
- }
-}
-
cl_program cl_fprog(char *filename, char *options, cl_info info)
{
size_t srcsize;
--
Gitblit v1.10.0