编译psroialign_cuda CHECK_CUDA(x); ‘CHECK_INPUT error: command 'gcc' failed with exit status 1

psroialign_cuda.cpp: In function ‘int psroialign_backward_cuda(at::Tensor, at::Tensor, at::Tensor, at::Tensor, float, int, int)’:
psroialign_cuda.cpp:25:80: error: ‘AT_CHECK’ was not declared in this scope
 #define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x, " must be a CUDAtensor ")
                                                                                ^
psroialign_cuda.cpp:29:3: note: in expansion of macro ‘CHECK_CUDA’
   CHECK_CUDA(x);       \
   ^
psroialign_cuda.cpp:66:5: note: in expansion of macro ‘CHECK_INPUT’
     CHECK_INPUT(top_diff);
     ^
error: command 'gcc' failed with exit status 1