add cu bench recipe

Author: buffalojoec

.github/workflows/main.yml

@@ -130,6 +130,23 @@ jobs:
       - name: Build Program
         run: make build-program
 
+  build_program:
+    name: Test Program
+    runs-on: ubuntu-latest
+    steps:
+      - name: Git Checkout
+        uses: actions/checkout@v4
+
+      - name: Setup Environment
+        uses: ./.github/actions/setup
+        with:
+          toolchain: build
+          cargo-cache-key: cargo-program
+          solana: true
+
+      - name: Build Program
+        run: make build-program
+
   test_program:
     name: Test Program
     runs-on: ubuntu-latest
@@ -147,6 +164,38 @@ jobs:
       - name: Test Program
         run: make test-program
 
+  bench_program_compute_units:
+    name: Benchmark Program Compute Units
+    runs-on: ubuntu-latest
+    needs: build_program # Cargo Bench won't build the SBPF binary...
+    steps:
+      - name: Git Checkout
+        uses: actions/checkout@v4
+
+      - name: Setup Environment
+        uses: ./.github/actions/setup
+        with:
+          cargo-cache-key: cargo-program-benches
+          cargo-cache-fallback-key: cargo-program
+          solana: true
+      
+      - name: Restore Program Builds
+        uses: actions/cache/restore@v4
+        with:
+          path: ./**/*.so
+          key: ${{ runner.os }}-builds-${{ github.sha }}
+
+      - name: Benchmark Compute Units
+        run: make bench-program-compute-units
+      
+      - name: Check Working Directory
+        run: |
+          if [ -n "$(git status --porcelain)" ]; then
+            test -z "$(git status --porcelain)"
+            echo "CU usage has changed. Please run `cargo bench` and commit the new results.";
+            exit 1;
+          fi
+
   generate_clients:
     name: Check Client Generation
     runs-on: ubuntu-latest

Makefile

@@ -32,6 +32,9 @@ test-%:
 build-program:
 	cargo test-sbf --manifest-path program/Cargo.toml --features bpf-entrypoint
 
+bench-program-compute-units:
+	cargo bench --manifest-path program/Cargo.toml
+
 format-js:
 	cd ./clients/js && pnpm install && pnpm format