where is the implementation details of border_align_forward

[muyihao]

I want read the implementation of border_align_forward, but can't open the _C in "from cvpods import _C" in the file of border_align.py

[Maycbj]

Thank you for your attention. https://github.com/Megvii-BaseDetection/cvpods/blob/master/cvpods/layers/border_align.py https://github.com/Megvii-BaseDetection/cvpods/tree/master/cvpods/layers/csrc/border_align

[fengqian-wei]

@Maycbj Thank you for you reply. I have some problem after reading the code. The block parameter set 128. why the grid parameter set output_size /64 instead of output_size /128

dim3 grid(std::min(THCCeilDiv((long)output_size, 64L), 4096L)); dim3 block(128, 4);

https://github.com/Megvii-BaseDetection/cvpods/blob/d1b1de5260eebc37f0f68c351a5e1963e91051a5/cvpods/layers/csrc/border_align/border_align_kernel.cu#L179