diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 52f659fd6c..59cc95f684 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -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 diff --git a/test/external/fuzz_linearizer.py b/test/external/fuzz_linearizer.py index 12830b8e7f..15c6041453 100644 --- a/test/external/fuzz_linearizer.py +++ b/test/external/fuzz_linearizer.py @@ -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)