ATen/cuda/CUDATypeConversion.cuh: No such file or directory

I’m trying to install custom pytorch kernels from https://github.com/thuyen/multicrop
I’m getting following error:

src/extract_glimpses_cuda.cu:6:44: fatal error: ATen/cuda/CUDATypeConversion.cuh: No such file or directory

My gcc version is 5.2.0 and Pytorch version is 0.4.1, I know that ATen/cuda/CUDATypeConversion.cuh was removed in Pytorch 0.4.1 But I don’t know how to modify the current multicrop code to get it working for Pytorch 0.4.1

Please help.