delete junk

This commit is contained in:
2025-10-03 10:35:12 +08:00
parent 9a607e69e1
commit 1cd40941c8
2 changed files with 0 additions and 15 deletions
-3
View File
@@ -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"))
-12
View File
@@ -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