Nice code

[Naruto-Sasuke]

I am wondering how you implement your code 'cuda_utils.cu'.
I am suffering from how to write efficient cuda code to make my custom operation work.
Could you give me some advices ?
BTW, I will check your code and may make some PRs to refine your code.

[luanfujun]

Thanks!

Maybe you can start by writing some small util functions based on NVIDIA tutorials. Say this one:

https://devblogs.nvidia.com/even-easier-introduction-cuda/