33ef02d09c16aaf5191115d0a6a3d3a9b6242ba1,gpytorch/utils/lanczos.py,,lanczos_tridiag_to_diag,#Any#,154

Before Change

    if t_mat.size(-1) < 64:
        orig_device = t_mat.device
        retr = torch.symeig(t_mat.cpu(), eigenvectors=True)
        return (r.to(orig_device) for r in retr)
    else:
        return torch.symeig(t_mat, eigenvectors=True)

After Change

    evals, evecs = retr
    mask = evals.ge(0)
    evecs = evecs * mask.type_as(evecs).unsqueeze(-2)
    evals = evals.masked_fill_(~mask, 1)

    return evals.to(orig_device), evecs.to(orig_device)

Instances