Unresolved reference 'nd_linear_sample_cuda' 

Thanks for your great work!
How can I solve the import error follows
  `import torch
  from torch.autograd import Function
  from .. import nd_linear_sample_cuda`

error: Unresolved reference 'nd_linear_sample_cuda' ?