I wonder if there’s a batch version of torch.eig or calling it in a async way on a GPU?
For example,
aaa = torch.randn((5, 5))
mat = torch.mm(torch.t(aaa), aaa)
mat = mat.pin_memory().cuda(async=True)
for ii in range(1000):
    torch.eig(mat, eigenvectors=True)

How can I run these 1000 torch.eig computations simultaneously on GPU?