From eff7c5fd2c466a9b3bf62cd086ccf83bcdcaf598 Mon Sep 17 00:00:00 2001 From: chenyu Date: Thu, 25 Jul 2024 14:35:11 -0400 Subject: [PATCH] halve kernel counts in metal Fuzz Test linearizer (#5716) the test time has increased to 3 minutes --- .github/workflows/test.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index a1dcef8a66..cbd7ab7b2e 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -312,7 +312,7 @@ jobs: - name: Test Beam Search run: PYTHONPATH="." METAL=1 IGNORE_BEAM_CACHE=1 python3 -m pytest extra/optimization/test_beam_search.py - name: Fuzz Test linearizer - run: PYTHONPATH="." METAL=1 CACHELEVEL=0 FUZZ_ALL_ACTIONS=1 DEPTH=2 FUZZ_N=48 FUZZ_MAX_SIZE=10000000 python test/external/fuzz_linearizer.py + run: PYTHONPATH="." METAL=1 FUZZ_ALL_ACTIONS=1 DEPTH=2 FUZZ_N=24 FUZZ_MAX_SIZE=10000000 python test/external/fuzz_linearizer.py - name: Fuzz Test models schedule run: FUZZ_SCHEDULE=1 FUZZ_SCHEDULE_MAX_PATHS=5 python -m pytest test/models/test_train.py test/models/test_end2end.py - name: Run TRANSCENDENTAL math