raise assert [run_process_replay] (#4879)

This commit is contained in:
qazal 2024-06-08 20:31:44 +08:00 committed by GitHub
parent d19f39d4dd
commit 1e3325f369
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions

View File

@ -18,12 +18,11 @@ for offset in tqdm(range(0, row_count, page_size)):
good_uops = k.linearize().uops
good_src = k.opts.render("test", good_uops)
try: assert compare_src == good_src
except AssertionError:
except AssertionError as e:
print("PROCESS REPLAY FAILED")
print(compare_k.ast)
print(compare_k.applied_opts)
diff = list(difflib.unified_diff(good_src.splitlines(), compare_src.splitlines()))
for line in diff:
print(colored(line, "red" if line.startswith("-") else "green" if line.startswith("+") else None))
# TODO: fix nondeterminism in ASTs with Variable 4860
#raise e
raise e