0cf477cb0fc956527b63d11638372c03dce1d1e6,test/util/test_linear_cg.py,TestLinearCG,test_batch_cg,#TestLinearCG#,52

Before Change

        solves = linear_cg(matrix.matmul, rhs=rhs, max_iter=size)

        // Check cg
        matrix_chol = torch.cat([matrix[i].potrf().unsqueeze(0) for i in range(5)])
        actual = torch.cat(
            [torch.potrs(rhs[i], matrix_chol[i]).unsqueeze(0) for i in range(5)]
        )
        self.assertTrue(approx_equal(solves, actual))

    def test_batch_cg_with_tridiag(self):
        batch = 5

After Change

        solves = linear_cg(matrix.matmul, rhs=rhs, max_iter=size)

        // Check cg
        matrix_chol = batch_potrf(matrix)
        actual = batch_potrs(rhs, matrix_chol)
        self.assertTrue(approx_equal(solves, actual))

    def test_batch_cg_with_tridiag(self):
        batch = 5

Instances