Thanks for your quick response!
I have tried to use gelu_backward_out_cuda and softmax_backward_cuda_out . However, the same errors still appear. I was wondering if the arguments for the two functions have changed as well?
My code:
return at::native::gelu_backward_cuda_out(grad_output, input);
...
return at::native::softmax_backward_cuda_out(grad_output, output, dim, self);

The errors:
error: ‘gelu_backward_cuda_out’ is not a member of ‘at::native’
error: ‘softmax_backward_cuda_out’ is not a member of ‘at::native’