mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-09 15:08:02 -05:00
583553f467
* split ranges * simpler * split ranges * range str * fix test * oops * faster * no group 2 * tests * dont_sub_ranges_for_image * revert that
24 lines
1.3 KiB
Python
24 lines
1.3 KiB
Python
# ruff: noqa: E501
|
|
import unittest
|
|
from tinygrad.uop.ops import UOp, Ops, AxisType
|
|
from tinygrad.dtype import dtypes
|
|
from tinygrad.engine.realize import get_program
|
|
|
|
class TestLinearizerFailures(unittest.TestCase):
|
|
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)
|
|
c2 = UOp.range(UOp.const(dtypes.index, 32), 2, AxisType.LOOP)
|
|
c3 = ((c1*UOp.const(dtypes.index, 32))+c2)
|
|
c4 = UOp(Ops.DEFINE_GLOBAL, dtypes.float.ptr(163840), arg=1, src=())
|
|
c5 = UOp.range(UOp.const(dtypes.index, 2560), 0, AxisType.REDUCE)
|
|
c6 = c4.index(((((((c5//UOp.const(dtypes.index, 8))%UOp.const(dtypes.index, 8))*UOp.const(dtypes.index, 8))+(c5%UOp.const(dtypes.index, 8)))+(((c2*UOp.const(dtypes.index, 40))+(c5//UOp.const(dtypes.index, 64)))*UOp.const(dtypes.index, 64)))+(c1*UOp.const(dtypes.index, 81920)))).load()
|
|
c7 = UOp(Ops.DEFINE_GLOBAL, dtypes.float.ptr(64), arg=2, src=())
|
|
c8 = c7.index(c3).load()
|
|
c9 = ((((c6+(c8*UOp.const(dtypes.float, -1.0)))*(c6+(c8*UOp.const(dtypes.float, -1.0)))).reduce(c5, arg=Ops.ADD)*UOp.const(dtypes.float, 0.000390625))+UOp.const(dtypes.float, 1e-05)).sqrt().reciprocal()
|
|
c10 = c0.index(c3).store(c9, c1, c2)
|
|
ast = c10.sink()
|
|
get_program(ast)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |