6ff31af0d8af54c6b3c853cdc37281d635b84637,gpytorch/lazy/added_diag_lazy_variable.py,AddedDiagLazyVariable,_preconditioner,#AddedDiagLazyVariable#,51

Before Change

                tensor, self._piv_chol_self, self._woodbury_cache, self._diag_var.diag()
            )

        return precondition_closure

    def inv_quad_log_det(self, inv_quad_rhs=None, log_det=False):
        inv_quad_term, log_det_term = super(
            AddedDiagLazyVariable, self

After Change

        // log_det correction
        if not hasattr(self, "_precond_log_det_cache"):
            lr_flipped = self._piv_chol_self.matmul(
                self._piv_chol_self.transpose(-2, -1).
                div(self._diag_var.diag().unsqueeze(1))
            )
            lr_flipped = lr_flipped + torch.eye(
                n=lr_flipped.size(0), dtype=lr_flipped.dtype, device=lr_flipped.device
            )
            ld_one = lr_flipped.potrf().diag().log().sum() * 2
            ld_two = self._diag_var.diag().data.log().sum()
            self._precond_log_det_cache = ld_one + ld_two

        return precondition_closure, self._precond_log_det_cache

Instances