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

alloc new output in fuzz_linearizer between baseline and real one (#3859)

Browse Source 
if the kernel is an assign `a += 1`, the rawbufs[0] is updated twice and gives false compare_error
This commit is contained in:
chenyu
2024-03-21 11:36:05 -04:00
committed by GitHub
parent b78352b423
commit 6bf0b82267
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/external/fuzz_linearizer.py vendored
View File

@@ -73,6 +73,7 @@ def compare_linearizer(lin: Linearizer, rawbufs=None, var_vals=None, ground_trut
return ("BASELINE_ERROR", rawbufs, var_vals, ground_truth,)
ground_truth = np.frombuffer(rawbufs[0].as_buffer(), rawbufs[0].dtype.np).copy()
rawbufs[0] = get_fuzz_rawbuf_like(rawbufs[0], zero=True) # get a new output buffer
if (run_msg := run_linearizer(lin, rawbufs, var_vals)) != "PASS":
return (run_msg, rawbufs, var_vals, ground_truth,)
result = np.frombuffer(rawbufs[0].as_buffer(), rawbufs[0].dtype.np)