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()