How to write cuda extension in version 0.4?

Hello. I wrote cuda extesion in 0.2 and 0,3, and it works.
However, when upgrade to 0.4, the compling script install.bash can’t work… This error show up:

/torch/lib/include/TH/THAllocator.h:6:28: fatal error: ATen/Allocator.h: No such file or directory

any suggestion ? Thanks.