From a79832b01f4a85e3970d07ad64cd1e050bfed978 Mon Sep 17 00:00:00 2001 From: chenyu Date: Mon, 27 Oct 2025 12:38:13 -0400 Subject: [PATCH] control_flow.py -> linearizer.py [pr] (#12948) --- test/external/external_benchmark_schedule.py | 2 +- tinygrad/codegen/__init__.py | 2 +- tinygrad/codegen/late/{control_flow.py => linearizer.py} | 0 3 files changed, 2 insertions(+), 2 deletions(-) rename tinygrad/codegen/late/{control_flow.py => linearizer.py} (100%) diff --git a/test/external/external_benchmark_schedule.py b/test/external/external_benchmark_schedule.py index 40f6a2114b..86879e7489 100644 --- a/test/external/external_benchmark_schedule.py +++ b/test/external/external_benchmark_schedule.py @@ -3,7 +3,7 @@ from tinygrad import Tensor, nn, Device from tinygrad.helpers import Profiling, Timing, getenv from tinygrad.uop.ops import Ops from tinygrad.codegen import full_rewrite_to_sink -from tinygrad.codegen.late.control_flow import linearize +from tinygrad.codegen.late.linearizer import linearize from tinygrad.uop.spec import type_verify, program_spec if __name__ == "__main__": diff --git a/tinygrad/codegen/__init__.py b/tinygrad/codegen/__init__.py index 55caa9ef4d..6819a393d9 100644 --- a/tinygrad/codegen/__init__.py +++ b/tinygrad/codegen/__init__.py @@ -17,7 +17,7 @@ from tinygrad.codegen.late.devectorizer import load_store_folding, load_store_in from tinygrad.codegen.opt.postrange import apply_opts from tinygrad.codegen.simplify import pm_simplify_ranges, pm_flatten_range, pm_split_ranges, pm_load_collapse from tinygrad.schedule.rangeify import pm_add_buffers_local, rangeify_codegen -from tinygrad.codegen.late.control_flow import CFGContext, pm_split_ends, pm_add_control_flow, linearize +from tinygrad.codegen.late.linearizer import CFGContext, pm_split_ends, pm_add_control_flow, linearize def full_rewrite_to_sink(sink:UOp, ren:Renderer|None=None, optimize:bool=True) -> UOp: if ren is None: ren = Renderer() diff --git a/tinygrad/codegen/late/control_flow.py b/tinygrad/codegen/late/linearizer.py similarity index 100% rename from tinygrad/codegen/late/control_flow.py rename to tinygrad/codegen/late/linearizer.py