Interesting. Yes, I’m getting the hanging now. I’ve opened an issue on the github with the bug: https://github.com/pytorch/pytorch/issues/3619 13 – we are looking into it.
Also torch.potrf is a thing and it probably calls what you’re using under the hood.