17e18750dd0b981460903ae631580d1d200509b5,test/lazy/test_block_diag_lazy_tensor.py,TestBlockDiagLazyTensor,test_batch_matmul,#TestBlockDiagLazyTensor#,54

Before Change

    def test_batch_matmul(self):
        rhs = torch.randn(2, 4 * 4, 4)
        rhs_tensor = torch.tensor(rhs, requires_grad=True)
        rhs_tensor_copy = torch.tensor(rhs, requires_grad=True)

        block_tensor = torch.tensor(self.blocks, requires_grad=True)
        block_tensor_copy = torch.tensor(self.blocks, requires_grad=True)

After Change

    def test_batch_matmul(self):
        rhs_tensor = torch.randn(2, 4 * 4, 4, requires_grad=True)
        rhs_tensor_copy = rhs_tensor.clone().detach().requires_grad_(True)
        block_tensor = self.blocks.clone().requires_grad_(True)
        block_tensor_copy = self.blocks.clone().requires_grad_(True)

        actual_block_diag = torch.zeros(2, 16, 16)

Instances