Detection is back, baby\!

Joseph Redmon

2015-02-11 0f645836f193e75c4c3b718369e6fab15b5d19c5

 src/col2im_kernels.cu

@@ -3,7 +3,7 @@
#include "cuda.h"
}

__global__ void col2im_kernel(float *data_col, int offset,
__global__ void col2im_kernel(float *data_col,
        int channels, int height, int width,
        int ksize, int stride, int pad, float *data_im)
{
@@ -46,17 +46,17 @@
            val += part;
        }
    }
    data_im[index+offset] = val;
    data_im[index] = val;
}


extern "C" void col2im_ongpu(float *data_col, int offset,
extern "C" void col2im_ongpu(float *data_col,
        int channels,  int height,  int width,
        int ksize,  int stride,  int pad, float *data_im)
{

    size_t n = channels*height*width;

    col2im_kernel<<<cuda_gridsize(n), BLOCK>>>(data_col, offset, channels, height, width, ksize, stride, pad, data_im);
    col2im_kernel<<<cuda_gridsize(n), BLOCK>>>(data_col, channels, height, width, ksize, stride, pad, data_im);
    check_error(cudaPeekAtLastError());
}