.

.github/workflows/ci.yml

@@ -70,3 +70,42 @@ jobs:
       - name: Run test
         run: |
           bash examples/test.sh
+
+  benchmark:
+    runs-on: ubuntu-20.04
+
+    steps:
+      - uses: actions/checkout@v3
+      - uses: actions/cache@v3
+        with:
+          path: |
+            ~/.cargo/registry
+            ~/.cargo/git
+            target
+          key: ${{ runner.os }}-cargo-${{ hashFiles('**/Cargo.lock') }}-1
+
+      - name: Install Rust
+        run: |
+          rustup update ${{ matrix.rust }} --no-self-update
+          rustup default ${{ matrix.rust }}
+          rustup target add wasm32-unknown-unknown
+
+      - name: Run benchmark test
+        run: |
+          bash benchmarks/benchmark_ci.sh
+
+  checks-pass:
+    # Always run this job!
+    if: always()
+    needs: [build, examples, benchmark]
+    runs-on: ubuntu-20.04
+    steps:
+      - name: check build result
+        if: ${{ needs.build.result != 'success' }}
+        run: exit 1
+      - name: check examples result
+        if: ${{ needs.examples.result != 'success' }}
+        run: exit 1
+      - name: check benchmark result
+        if: ${{ needs.benchmark.result != 'success' }}
+        run: exit 1

Cargo.toml

@@ -47,6 +47,6 @@ members = [
 ]
 
 [workspace.dependencies]
-candid = "0.9.5"
+candid = "0.9.11"
 ic-cdk = "0.10.0"
 ic-cdk-macros = "0.7.1"

benchmark-canisters/src/memory_manager.rs

@@ -56,3 +56,19 @@ pub fn memory_manager_overhead() -> BenchResult {
         }
     })
 }
+
+/// Benchmarks the `MemoryManager`'s `grow` method.
+#[ic_cdk_macros::query]
+pub fn memory_manager_grow() -> BenchResult {
+    let mem_mgr = MemoryManager::init_with_bucket_size(DefaultMemoryImpl::default(), 1);
+
+    let buckets_per_memory = 32000;
+
+    let memory = mem_mgr.get(MemoryId::new(0));
+
+    crate::benchmark(|| {
+        for _ in 0..buckets_per_memory {
+            memory.grow(1);
+        }
+    })
+}

benchmarks/benchmark.rs

@@ -22,6 +22,7 @@ lazy_static::lazy_static! {
         // MemoryManager benchmarks
         "memory_manager_baseline",
         "memory_manager_overhead",
+        "memory_manager_grow",
 
         // BTree benchmarks
         "btreemap_insert_10mib_values",

benchmarks/benchmark_ci.sh

@@ -0,0 +1,16 @@
+#!/usr/bin/env bash
+set -Eexuo pipefail
+
+BENCH_OUTPUT=$(cargo bench)
+
+set +e
+REGRESSIONS=$( echo "$BENCH_OUTPUT" |  grep -c "regressed by" )
+set -e
+
+if [[ $REGRESSIONS != 0 ]]; then
+  echo "FAIL! Performance regressions are detected. 
+        Please run \"cargo bench -- --persist\" to update \"results.yml\""
+  exit 1
+fi
+
+echo "SUCCESS! Performance regressions are not detected."