From 60dd9a162c92ce1e2e8db34c2367b2f7bc3eebde Mon Sep 17 00:00:00 2001 From: nimlgen <138685161+nimlgen@users.noreply.github.com> Date: Thu, 28 Aug 2025 14:07:18 +0300 Subject: [PATCH] memory: tiny tlsf cleanup (#11887) --- tinygrad/runtime/support/memory.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tinygrad/runtime/support/memory.py b/tinygrad/runtime/support/memory.py index 96db45250a..0c74ea4127 100644 --- a/tinygrad/runtime/support/memory.py +++ b/tinygrad/runtime/support/memory.py @@ -77,11 +77,10 @@ class TLSFAllocator: if self.lv1_entries[l1] == 0: continue for l2 in range(self.lv2(size) if l1 == size.bit_length() else 0, (1 << self.l2_cnt)): if len(self.storage[l1][l2]) > 0: - nsize = self.blocks[self.storage[l1][l2][0]][0] - assert nsize >= size, "block must be larger" - # Block start address. start = self.storage[l1][l2][0] + nsize = self.blocks[start][0] + assert nsize >= size, "block must be larger" # If request contains alignment, split the block into two parts. if (new_start:=round_up(start, align)) != start: