Joseph Redmon
2014-10-26 edbccdfcaf46f11e631afe98796f3e6e170da5d0
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);
}