22361f98c2c8683715a507f59a5884071d672826,gpytorch/utils/lincg.py,LinearCG,solve,#LinearCG#Any#Any#Any#,34

Before Change

        if isinstance(A, Variable) or isinstance(b, Variable):
            raise RuntimeError("LinearCG is not intended to operate directly on Variables or be used with autograd.")

        if not isinstance(A, torch.Tensor) or not isinstance(b, torch.Tensor):
            raise RuntimeError("LinearCG is intended to operate on tensors.")

        if x is None:
            x = self.precondition_closure(b)

        residual = b - A.mv(x)

After Change

            mv_closure = lambda v: A.mv(v)
        else:
            // Probably fairly difficult to implement a default preconditioner for an arbitrary mv closure.
            if self.precondition_closure is None:
                self.precondition_closure = lambda v: v
            mv_closure = A
            self._reset_precond = True

        if b.ndimension() > 1:
            return self._solve_batch(A, b, x)

Instances