github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

fuzz_linearizer: fix uops and add to test.yml (#3588)

Browse Source
This commit is contained in:
Francis Lam
2024-03-02 15:03:42 -08:00
committed by GitHub
parent 8978488565
commit 162dfb07d9
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.github/workflows/test.yml vendored
View File

@@ -303,6 +303,9 @@ jobs:
run: METAL=1 DEBUG=3 python test/test_ops.py TestOps.test_big_gemm
- name: Test LLaMA compile speed
run: PYTHONPATH="." METAL=1 python test/external/external_test_speed_llama.py
- name: Fuzz Test linearizer
run: PYTHONPATH="." METAL=1 CACHELEVEL=0 FUZZ_BEAM=1 DEPTH=2 FUZZ_N=48 FUZZ_MAX_SIZE=10000000 python test/external/fuzz_linearizer.py
# testwebgl:
# name: WebGL Tests

2
test/external/fuzz_linearizer.py vendored
View File

@@ -103,7 +103,7 @@ def fuzz_linearizer(lin: Linearizer):
if not FUZZ_BEAM and test_lin.applied_opts: print(f"applied opts: {test_lin.applied_opts}")
# stop if kernel uops repeat
tuops = tuplize_uops(test_lin.linearize().uops)
tuops = tuplize_uops(test_lin.linearize().uops.uops)
if tuops in seen_uops:
continue
seen_uops[tuops] = tuple(test_lin.applied_opts)