Update benchmark command

ami-iit · Feb 28, 2025 · 1cdd9d3 · 1cdd9d3
1 parent cafe87c
commit 1cdd9d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/gpu_benchmark.yml b/.github/workflows/gpu_benchmark.yml
@@ -29,7 +29,7 @@ jobs:
         uses: actions/checkout@v4
 
       - name: Run benchmark and store result
-        run: pytest bench.py --benchmark-json output.json
+        run: pytest tests/test_benchmark.py --benchmark-only --benchmark-json output.json
 
       - name: Download previous benchmark data
         uses: actions/cache@v4