Update benchmark.yaml

gizatechxyz · Jan 2, 2025 · b832325 · b832325
1 parent a6642e8
commit b832325
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/.github/workflows/benchmark.yaml b/.github/workflows/benchmark.yaml
@@ -20,10 +20,11 @@ jobs:
       - name: Checkout Code
         uses: actions/checkout@v3
 
-      - name: Setup Rust (Nightly)
-      - uses: dtolnay/rust-toolchain@master
+      - name: Setup Rust
+        uses: moonrepo/setup-rust@v0
         with:
-          toolchain: nightly-2024-12-17
+          channel: stable
+          cache-target: release
 
       - name: Cache Benchmark Data
         uses: actions/cache@v3
@@ -35,7 +36,7 @@ jobs:
             ${{ runner.os }}-bench-
 
       - name: Run Benchmarks
-        run: cargo +nightly-2024-12-17 bench --bench tensor_ops
+        run: cargo bench --bench tensor_ops
 
       - name: Process Benchmark Results
         uses: joshua-auchincloss/criterion-pages@v1