Joseph Redmon
2014-12-16 d6cdea49eb509af852e6279922863cf63adb4a56
Small fixes
2 files modified
5 ■■■■ changed files
src/cnn.c 2 ●●● patch | view | raw | blame | history
src/opencl.c 3 ●●●●● patch | view | raw | blame | history
src/cnn.c
@@ -562,7 +562,7 @@
int find_arg(int argc, char* argv[], char *arg)
{
    int i;
    for(i = 0; i < argc-1; ++i) if(0==strcmp(argv[i], arg)) {
    for(i = 0; i < argc; ++i) if(0==strcmp(argv[i], arg)) {
        del_arg(argc, argv, i);
        return 1;
    }
src/opencl.c
@@ -159,6 +159,7 @@
void cl_read_array(cl_mem mem, float *x, int n)
{
    if(gpu_index < 0) return;
    cl.error = clEnqueueReadBuffer(cl.queue, mem, CL_TRUE, 0, sizeof(float)*n,x,0,0,0);
    check_error(cl);
}
@@ -175,6 +176,7 @@
void cl_write_array(cl_mem mem, float *x, int n)
{
    if(gpu_index < 0) return;
    cl.error = clEnqueueWriteBuffer(cl.queue, mem, CL_TRUE, 0,sizeof(float)*n,x,0,0,0);
    check_error(cl);
}
@@ -209,6 +211,7 @@
cl_mem cl_make_int_array(int *x, int n)
{
    if(gpu_index < 0) return 0;
    cl_mem mem = clCreateBuffer(cl.context,
            CL_MEM_READ_WRITE|CL_MEM_COPY_HOST_PTR,
            sizeof(int)*n, x, &cl.error);