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

fixup ast in kernel to be MetaOps.SINK [run_process_replay] (#5424)

Browse Source 
* fixup ast in kernel to be MetaOps.SINK [run_process_replay]

* fix tests

* fix more tests
This commit is contained in:
George Hotz
2024-07-12 14:01:03 -07:00
committed by GitHub
parent b055ece550
commit 94599c0637
8 changed files with 35 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -17,7 +17,7 @@ for offset in tqdm(range(0, row_count, page_size)):
with Context(**{k:v for k,v in ctx.items() if k in ContextVar._cache}):
# try linearize
try:
k = Linearizer(*ast, opts=opts)
k = Linearizer(ast, opts=opts)
for opt in applied_opts: k.apply_opt(opt)
good_src = k.opts.render(name, k.linearize().uops)
except Exception as e: