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

check Program from HEAD [run_process_replay] (#4996)

Browse Source 
* use the same prg [run_process_replay]

* put var back
This commit is contained in:
qazal
2024-06-16 20:12:30 +03:00
committed by GitHub
parent 2b07847f2b
commit 71aad183fd
3 changed files with 5 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/external/replay_codegen.py vendored
View File

@@ -2,7 +2,7 @@
import difflib, pickle
from tqdm import tqdm
from tinygrad.codegen.linearizer import Linearizer
from tinygrad.helpers import colored, db_connection, VERSION
from tinygrad.helpers import colored, db_connection, VERSION, to_function_name
page_size = 100
conn = db_connection()
@@ -11,12 +11,10 @@ row_count = cur.execute(f"select count(*) from 'process_replay_{VERSION}'").fetc
for offset in tqdm(range(0, row_count, page_size)):
cur.execute(f"SELECT val FROM 'process_replay_{VERSION}' LIMIT ? OFFSET ?", (page_size, offset))
for row in cur.fetchall():
compare_k: Linearizer = pickle.loads(row[0])
compare_src = compare_k.opts.render("test", compare_k.uops)
compare_k, compare_src = pickle.loads(row[0])
k = Linearizer(*compare_k.ast, opts=compare_k.opts)
for opt in compare_k.applied_opts: k.apply_opt(opt)
good_uops = k.linearize().uops
good_src = k.opts.render("test", good_uops)
good_src = k.opts.render(to_function_name(compare_k.name), k.linearize().uops)
try: assert compare_src == good_src
except AssertionError as e:
print("PROCESS REPLAY FAILED")