84fafd4599e29c68e63b90a38b56fcc65f289288,test/functions/test_inv_quad_log_det.py,TestInvQuadLogDetNonBatch,test_inv_quad_logdet_vector,#TestInvQuadLogDetNonBatch#,37

Before Change

    def test_inv_quad_logdet_vector(self):
        // Forward pass
        actual_inv_quad = self.mat_clone.inverse().matmul(self.vec_clone).mul(self.vec_clone).sum()
        actual_logdet = self.mat_clone.logdet()
        with gpytorch.settings.num_trace_samples(1000):
            non_lazy_tsr = NonLazyTensor(self.mat)
            res_inv_quad, res_logdet = non_lazy_tsr.inv_quad_logdet(inv_quad_rhs=self.vec, logdet=True)

        self.assertAlmostEqual(res_inv_quad.item(), actual_inv_quad.item(), places=1)
        self.assertAlmostEqual(res_logdet.item(), actual_logdet.item(), places=1)

        // Backward
        actual_inv_quad.backward()
        actual_logdet.backward()
        res_inv_quad.backward(retain_graph=True)
        res_logdet.backward()

        self.assertLess(torch.max((self.mat_clone.grad - self.mat.grad).abs()).item(), 1e-1)
        self.assertLess(torch.max((self.vec_clone.grad - self.vec.grad).abs()).item(), 1e-1)

After Change

        self.assertTrue(linear_cg_mock.called)

    def test_inv_quad_logdet_vector(self):
        rhs = torch.randn(self.matrix_shape[-1])
        self._test_inv_quad_logdet(inv_quad_rhs=rhs, logdet=True)

    def test_inv_quad_only_vector(self):
        rhs = torch.randn(self.matrix_shape[-1])
        self._test_inv_quad_logdet(inv_quad_rhs=rhs, logdet=False)

Instances