Thanks for your reply.
sorry i am late.
I can make .cu file include <Aten.h>.
but, still there are error if i include <torch.h>
at::Tensor can not save gradient right?
so, i need torch::Tensor. Is it possible to use torch::Tensor in cuda(.cu file)