Cuda on amd gpu. What is the AMD equivalent to the following command? torch.