diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml
index ed2de584..cd766cb8 100644
--- a/.github/workflows/benchmark.yml
+++ b/.github/workflows/benchmark.yml
@@ -7,7 +7,7 @@ on:
 jobs:
   benchmark:
     if: ${{ github.repository_owner == 'deepmodeling' }}
-    runs-on: codspeed-macro
+    runs-on: ubuntu-latest
     steps:
     - uses: actions/checkout@v5
     - name: Set up Python
@@ -23,7 +23,8 @@ jobs:
     - name: Install dependencies
       run: uv pip install --system .[test,amber,ase,pymatgen,benchmark] rdkit openbabel-wheel
     - name: Run benchmarks
-      uses: CodSpeedHQ/action@v3
+      uses: CodSpeedHQ/action@v4
       with:
         token: ${{ secrets.CODSPEED_TOKEN }}
+        mode: walltime
         run: pytest benchmark/ --codspeed