From 69eb55a529434ea128362d95e795a37e81875922 Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Wed, 18 Dec 2024 16:29:07 +0200 Subject: [PATCH] bugfix: dtype mismatch in TestAssign.test_double_assign_alt [pr] (#8310) --- test/test_assign.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_assign.py b/test/test_assign.py index fcbb4e491b..241e5a9f31 100644 --- a/test/test_assign.py +++ b/test/test_assign.py @@ -171,8 +171,8 @@ class TestAssign(unittest.TestCase): # NOTE: this is similar to the resnet failure #@unittest.expectedFailure def test_double_assign_alt(self): - a = Tensor.ones(4).contiguous().realize() - b = Tensor([1, 2, 3, 4]).realize().lazydata + a = Tensor.ones(4, dtype=dtypes.int).contiguous().realize() + b = Tensor([1, 2, 3, 4], dtype=dtypes.int).realize().lazydata a1 = a.lazydata.assign(b) a2 = a.lazydata.assign(b) sched = create_schedule([a1, a2])