From cbbc2fdea50fe22f388190e15430f916bc30cf1e Mon Sep 17 00:00:00 2001 From: chenyu Date: Wed, 11 Feb 2026 15:02:44 -0500 Subject: [PATCH] update test_assign_slice_then_read (#14687) passes locally now --- test/unit/test_assign.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/test/unit/test_assign.py b/test/unit/test_assign.py index 1c19525799..5e41c36908 100644 --- a/test/unit/test_assign.py +++ b/test/unit/test_assign.py @@ -535,16 +535,10 @@ class TestAssign(unittest.TestCase): This is the KV cache pattern from llm.py. """ v_pos = Variable("pos", 0, 3).bind(0) - - # without .realize() after assign, the read doesn't see the assigned values cache = Tensor.zeros(4, 4).contiguous().realize() cache[v_pos:v_pos+1, :].assign(Tensor.ones(1, 4)) - self.assertEqual(cache.sum().item(), 0.0) # should be 4.0! + self.assertEqual(cache.sum().item(), 4.0) - # TODO: remove .realize() workaround once assign-read dependency is fixed - cache2 = Tensor.zeros(4, 4).contiguous().realize() - cache2[v_pos:v_pos+1, :].assign(Tensor.ones(1, 4)).realize() - self.assertEqual(cache2.sum().item(), 4.0) class TestAssignOrdering(unittest.TestCase): """Tests for complex assign orderings that could differ between lazy and eager execution.