pytorch-sqrtm icon indicating copy to clipboard operation
pytorch-sqrtm copied to clipboard

Published 20 hours ago • verified publisher icon steveli

How to accelerate the running speed

Open ZrW00 opened this issue 3 years ago • 2 comments

I try to calculate sqrtm by this method, but soon I found it so slow to run, How can I accelerate the running speed?

ZrW00 avatar Dec 04 '21 13:12 ZrW00

I dont think we can do much here. Most of the latency s behould due to converting your data from GPU to cpu back and forth and not necessarily suboptimal code. If you know a better implementation please let meknow.

h-gokul avatar Apr 22 '22 21:04 h-gokul

I find discussions in this thread useful: https://github.com/pytorch/pytorch/issues/25481

steveli avatar Apr 23 '22 03:04 steveli