Joseph Redmon
2014-10-28 af4e4f92dc9e5da160eb6c6870a7b38b863f1c6c
src/im2col.c
@@ -91,12 +91,10 @@
        width_col = 1 + (width-1) / stride;
    }
    size_t global_size[2];
    global_size[0] = batch*channels_col;
    global_size[1] = height_col*width_col;
    size_t global_size = batch*channels_col*height_col*width_col;
    clEnqueueNDRangeKernel(queue, im2col_kernel, 2, 0,
            global_size, 0, 0, 0, 0);
    clEnqueueNDRangeKernel(queue, im2col_kernel, 1, 0,
            &global_size, 0, 0, 0, 0);
    check_error(cl);
}