sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 8 Activity

bugfix: dtype mismatch in TestAssign.test_double_assign_alt [pr] (#8310)

Browse Source
This commit is contained in:
qazal
2024-12-18 22:29:07 +08:00
committed by GitHub
parent 32df46cd73
commit 69eb55a529
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_assign.py
+2 -2
View File
@@ -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])