From cf0c28d5ae2664e87f731668f4f5ce2ad79413c1 Mon Sep 17 00:00:00 2001 From: George Hotz <72895+geohot@users.noreply.github.com> Date: Tue, 16 Dec 2025 19:35:50 -0400 Subject: [PATCH] all tests pass on strix halo (#13728) --- test/test_rangeify.py | 2 +- test/unit/test_viz.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_rangeify.py b/test/test_rangeify.py index ab6018cfe8..9dd0dde998 100644 --- a/test/test_rangeify.py +++ b/test/test_rangeify.py @@ -339,7 +339,7 @@ class TestRangeify(unittest.TestCase): def test_transformer_ffn(self): from tinygrad.apps.llm import TransformerBlock from tinygrad import nn - blk = TransformerBlock(1024, 4096, 1, 1, 1e-5) + blk = TransformerBlock(1024, 4096, 1, 1, 1e-5, head_dim=1024, rope_theta=10000.0) for p in nn.state.get_parameters(blk): p.replace(Tensor.empty(p.shape)) x = Tensor.empty(128, 1024) diff --git a/test/unit/test_viz.py b/test/unit/test_viz.py index b7edb9d2c6..6cd5673b46 100644 --- a/test/unit/test_viz.py +++ b/test/unit/test_viz.py @@ -232,7 +232,7 @@ import gc def bufs_allocated() -> int: gc.collect() - return sum([isinstance(x, Buffer) for x in gc.get_objects()]) + return sum([type(x).__name__ == "Buffer" and type(x).__module__ == "tinygrad.device" for x in gc.get_objects()]) class TestVizGC(BaseTestViz): def test_gc(self):