From 2c8ad1b419e12b91f16e701aac2f6736bc13ae50 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Wed, 19 Nov 2025 16:10:00 -0800 Subject: [PATCH] _apply_movement_op cache --- test/external/external_uop_gc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/external/external_uop_gc.py b/test/external/external_uop_gc.py index f54c24e7c5..e99d71691e 100644 --- a/test/external/external_uop_gc.py +++ b/test/external/external_uop_gc.py @@ -1,7 +1,7 @@ import gc from tinygrad import Tensor, UOp, Device, nn from tinygrad.engine.realize import method_cache, get_program -from tinygrad.schedule.indexing import apply_movement_op +from tinygrad.schedule.indexing import _apply_movement_op from tinygrad.uop.divandmod import fold_divmod_general from test.test_tiny import TestTiny @@ -69,7 +69,7 @@ if __name__ == "__main__": # these caches will keep uops alive method_cache.clear() - apply_movement_op.cache_clear() + _apply_movement_op.cache_clear() fold_divmod_general.cache_clear() Tensor._device_seeds.clear() Tensor._device_rng_counters.clear()