91b0d220c8e816766fd4565e1d2f5115d3afbefe,test/functions/test_inv_matmul.py,TestInvMatmulBatch,setUp,#TestInvMatmulBatch#,53

Before Change

class TestInvMatmulBatch(unittest.TestCase):
    def setUp(self):
        mats = [[[3, -1, 0], [-1, 3, 0], [0, 0, 3]], [[10, -2, 1], [-2, 10, 0], [1, 0, 10]]]
        vecs = torch.randn(2, 3, 4, requires_grad=True)

        self.mats_var = torch.tensor(mats, dtype=torch.float, requires_grad=True)
        self.mats_var_copy = torch.tensor(mats, dtype=torch.float, requires_grad=True)
        self.vecs_var = vecs
        self.vecs_var_copy = vecs.clone().detach().requires_grad_(True)

After Change

        if os.getenv("unlock_seed") is None or os.getenv("unlock_seed").lower() == "false":
            self.rng_state = torch.get_rng_state()
            torch.manual_seed(0)
            if torch.cuda.is_available():
                torch.cuda.manual_seed_all(0)
            random.seed(0)

        mats = torch.randn(2, 8, 8)
        mats = mats @ mats.transpose(-1, -2)
        vecs = torch.randn(2, 8, 4)
        self.mats = mats.detach().clone().requires_grad_(True)
        self.mats_copy = mats.detach().clone().requires_grad_(True)
        self.vecs = vecs.detach().clone().requires_grad_(True)
        self.vecs_copy = vecs.detach().clone().requires_grad_(True)

Instances