Tune test_perf [perf]

ragulpr · Jan 19, 2025 · 1d601ff · 1d601ff
1 parent 38f63c0
commit 1d601ff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -30,4 +30,4 @@ jobs:
 
       - name: Run performance tests
         if: contains(github.event.head_commit.message, '[perf]')
-        run: python test_performance.py --repeats 1000 --time-limit 120
+        run: python test_performance.py
diff --git a/test_performance.py b/test_performance.py
@@ -7,7 +7,7 @@
 
 # python -m cProfile -s cumtime test_performance.py --repeats  10
 parser = argparse.ArgumentParser(description='')
-parser.add_argument('--repeats', type=int, default=1000000, metavar='N')
+parser.add_argument('--repeats', type=int, default=100000, metavar='N')
 parser.add_argument('--n-features', type=int, default=512, metavar='N')
 parser.add_argument('--batch-size', type=int, default=1024, metavar='N')
 parser.add_argument('--no-cuda', action='store_true',