github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-08 22:48:25 -05:00
Code Issues Packages Projects Releases Wiki Activity

RANGEIFY=1 test_jit (#12254)

Browse Source 
* RANGEIFY=1 test_jit

* don't do any of that

* disk

* simple disk tensor

* more work

* run more tests

* it also doesn't copy everytime

* skip tests that hang everything
This commit is contained in:
qazal
2025-09-20 17:34:32 +03:00
committed by GitHub
parent 393c6b236c
commit 57c7e0a8f8
4 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_jit.py
View File

@@ -8,7 +8,7 @@ from tinygrad.tensor import Tensor
from tinygrad.engine.jit import TinyJit, GraphRunner, MultiGraphRunner, graph_class
from tinygrad.engine.realize import CompiledRunner, BufferCopy, BufferXfer
from tinygrad.device import Device
from tinygrad.helpers import Context, JIT, GlobalCounters, getenv
from tinygrad.helpers import Context, JIT, RANGEIFY, GlobalCounters, getenv
from tinygrad.dtype import dtypes
from extra.models.unet import ResBlock
@@ -605,6 +605,7 @@ class TestJitPrune(unittest.TestCase):
assert len(w2_prune.captured.jit_cache) == 1, "prune should have removed the copy"
class TestJitFree(unittest.TestCase):
@unittest.skipIf(RANGEIFY, "needs a rewrite")
def test_free_intermediates(self):
ext_tensor = Tensor([1,24,23,45,1])
@TinyJit