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

ignore beam cache in test_tiny for stateless beam (#8771)

Browse Source
This commit is contained in:
Ignacio Sica
2025-01-27 12:56:30 -05:00
committed by GitHub
parent 3ed146a5ff
commit ed1b573868
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_tiny.py
+2 -2
View File
@@ -63,7 +63,7 @@ class TestTiny(unittest.TestCase):
# *** BEAM (for Kernel speed) ***
def test_beam(self):
with Context(BEAM=1): self.test_plus()
with Context(BEAM=1, IGNORE_BEAM_CACHE=1): self.test_plus()
# *** symbolic (to allow less recompilation) ***
@@ -107,7 +107,7 @@ class TestTiny(unittest.TestCase):
with Context(IMAGE=2): self.test_gemm(out_dtype=dtypes.imagef((4, 1, 4)))
def test_beam_image(self):
with Context(BEAM=1): self.test_image()
with Context(BEAM=1, IGNORE_BEAM_CACHE=1): self.test_image()
if __name__ == '__main__':
unittest.main()
tinygrad/helpers.py
+1 -1
View File
@@ -109,7 +109,7 @@ USE_TC, TC_OPT, AMX, TRANSCENDENTAL = ContextVar("TC", 1), ContextVar("TC_OPT",
FUSE_ARANGE, FUSE_CONV_BW = ContextVar("FUSE_ARANGE", 0), ContextVar("FUSE_CONV_BW", 0)
SPLIT_REDUCEOP, NO_MEMORY_PLANNER, RING = ContextVar("SPLIT_REDUCEOP", 1), ContextVar("NO_MEMORY_PLANNER", 0), ContextVar("RING", 1)
PICKLE_BUFFERS, PROFILE, LRU = ContextVar("PICKLE_BUFFERS", 1), ContextVar("PROFILE", getenv("VIZ")), ContextVar("LRU", 1)
CACHELEVEL = ContextVar("CACHELEVEL", 2)
CACHELEVEL, IGNORE_BEAM_CACHE = ContextVar("CACHELEVEL", 2), ContextVar("IGNORE_BEAM_CACHE", 0)
@dataclass(frozen=True)
class Metadata: