b24e13ad3ca193a2e3b84350d19199bdf17df45f,gpytorch/functions/_inv_matmul.py,,_solve,#Any#Any#Any#,8

Before Change

def _solve(lazy_tsr, rhs, preconditioner):
    if settings.fast_computations.solves.off() or lazy_tsr.matrix_shape.numel() <= settings.max_cholesky_numel.value():
        return lazy_tsr._cholesky()._cholesky_solve(rhs)
    else:
        return lazy_tsr._solve(rhs, preconditioner)

After Change

def _solve(lazy_tsr, rhs, preconditioner):
    if settings.fast_computations.solves.off() or lazy_tsr.size(-1) <= settings.max_cholesky_size.value():
        return lazy_tsr._cholesky()._cholesky_solve(rhs)
    else:
        return lazy_tsr._solve(rhs, preconditioner)

Instances