mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

failing test from application

Browse source
This commit is contained in:
George Hotz 2026-02-02 16:48:24 +08:00
commit 8a3d451ee4
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
test/test_linearizer_failures.py
View file

@ -1,11 +1,21 @@
# ruff: noqa: E501
import unittest
from tinygrad import Tensor
from tinygrad.uop.ops import UOp, Ops, AxisType
from tinygrad.dtype import dtypes
from tinygrad.engine.realize import get_program
from tinygrad.device import Device
class TestLinearizerFailures(unittest.TestCase):
def test_cumsum_repeat_reshape_multiply(self):
# cumsum + repeat + reshape + multiply fails when step > 512
step, num_steps = 513, 10
t = Tensor.arange(step).float()
phase = t.cumsum()
tiled = phase.repeat((num_steps,)).reshape(num_steps, step)
pattern = Tensor([1,0,0,1,0,0,0,0,1,0]).reshape(num_steps, 1)
result = (tiled * pattern).flatten()
result.numpy() # should not raise AssertionError in CFGContext
def test_fail_1(self):
c0 = UOp(Ops.DEFINE_GLOBAL, dtypes.float.ptr(64), arg=0, src=())
c1 = UOp.range(UOp.const(dtypes.index, 2), 1, AxisType.LOOP)