Can you please write the error you’re getting?
I’ve migrated from PyTorch 0.4 to 1.0 without any problem, but my custom functions are pretty basic. The changes I made in my c++ code were:

Using <torch/extension.h> instead of <torch/torch.h>.
AT_ASSERT now gets one parameter instead of two.