@tom Hi Thomas, thanks for your reply!
I am using at::native::. For example,
at::native::gelu_backward_out_cuda(grad_output, input);

However, it will result in an error with PyTorch 1.10 as mentioned before. But with PyTorch 1.9 and lower versions, I can call at::native::gelu_backward_cuda(grad_output, input); successfully.
Kind regards.