0cf477cb0fc956527b63d11638372c03dce1d1e6,test/util/test_linear_cg.py,TestLinearCG,test_batch_cg_with_tridiag,#TestLinearCG#,70

Before Change

        // 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))

        // Check tridiag
        for i in range(5):

After Change

        )

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

        // Check tridiag
        for i in range(5):

Instances