Torch.potrs perform on CPU or GPU?


(linyu) #1

A = torch.potrs(B, torch.potrf©)
Now, I perform the above calculation on pytorch, and both B and C are in GPU. Does pytorch perform this under GPU or CPU? If it is performed on GPU, in my tests, this command takes up a lot of CPU, why?
Thank you for your attention.