From 1cd40941c844707aa531769887c55d8c60f09760 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Fri, 3 Oct 2025 10:35:12 +0800 Subject: [PATCH] delete junk --- tinygrad/codegen/__init__.py | 3 --- tinygrad/schedule/rangeify.py | 12 ------------ 2 files changed, 15 deletions(-) diff --git a/tinygrad/codegen/__init__.py b/tinygrad/codegen/__init__.py index 86157aaf6b..516a128a3d 100644 --- a/tinygrad/codegen/__init__.py +++ b/tinygrad/codegen/__init__.py @@ -77,9 +77,6 @@ def _get_rewrites_for_renderer(opts:Renderer, optimize:bool, linearizer:bool, _Q # ** expander (expand_rewrite) ** ret.append(RewriteStep(sym+migrate_indexing+pm_group_for_reduce, name="postopt symbolic")) - # add locals - #ret.append(RewriteStep(pm_fix_bufferize, name="fix bufferize")) - # add locals ret.append(RewriteStep(pm_add_buffers+rangeify_codegen+pm_flatten_range, name="add local buffers")) diff --git a/tinygrad/schedule/rangeify.py b/tinygrad/schedule/rangeify.py index a49c3555cb..b3f6856392 100644 --- a/tinygrad/schedule/rangeify.py +++ b/tinygrad/schedule/rangeify.py @@ -576,18 +576,6 @@ pm_add_buffers = PatternMatcher([ (UPat(Ops.BUFFERIZE, name="x"), bufferize_to_store), ])+_pm_add_buffers -def fix_bufferize(x:UOp, y:UOp): - local_end_ranges = [] - for r in x.ranges: - if r.arg[-1] in {AxisType.LOCAL, AxisType.WARP}: - local_end_ranges.append(r) - if len(local_end_ranges) == 0: return None - return UOp(Ops.INDEX, y.dtype, (x.replace(src=x.src+tuple(local_end_ranges)),)+y.src[1:]+tuple(local_end_ranges), arg=y.arg) - -pm_fix_bufferize = PatternMatcher([ - (UPat(Ops.INDEX, src=(UPat(Ops.BUFFERIZE, name="x"),), name="y", allow_any_len=True), fix_bufferize), -]) - # ***************** # 5. split into kernels