While this might be not the preferred way of @fxia22 in order to implement gpu support, i just want to share it for other people who might have interests in this.
Since I wrote these codes without careful reviews, I would appreciate if you understand any mistake I might have made.
Hi,
I added cuda support based on following references; 1) https://github.com/pytorch/extension-ffi/issues/3 2) https://discuss.pytorch.org/t/compiling-an-extension-with-cuda-files/302/9 3) https://github.com/longcw/yolo2-pytorch/blob/master/layers/reorg/build.py
While this might be not the preferred way of @fxia22 in order to implement gpu support, i just want to share it for other people who might have interests in this.
Since I wrote these codes without careful reviews, I would appreciate if you understand any mistake I might have made.
I'm working on test script in advance.
Anyway, thank you for sharing your work.
Best regards,
Jae Hyun