github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

match kernelize API in process replay (#9948)

Browse Source
This commit is contained in:
qazal
2025-04-21 00:23:41 +03:00
committed by GitHub
parent 343a5eb588
commit 1d90be2cff
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/external/process_replay/process_replay.py vendored
View File

@@ -35,7 +35,7 @@ class ProcessReplayWarning(Warning): pass
def recreate_sched(big_sink:UOp) -> list[UOp]:
becomes_map = get_becomes_map(big_sink)
sched_sink = UOp.sink(*[becomes_map.get(x,x) for x in big_sink.src])
sched_sink = big_sink.substitute(becomes_map)
return dedup(u.arg.ast for u in sched_sink.toposort if u.op is Ops.KERNEL)
def recreate_kernel(ast:UOp, opts:Renderer, applied_opts:list[Opt], name:str, _) -> str: