test(benchmarks): add benchmarks for e2e actions

privacy-scaling-explorations · Jan 22, 2024 · 256fe63 · 256fe63
1 parent 77a424d
commit 256fe63
Show file tree

Hide file tree

Showing 8 changed files with 407 additions and 1 deletion.
diff --git a/.github/workflows/benchmarks.yml b/.github/workflows/benchmarks.yml
@@ -33,5 +33,21 @@ jobs:
         run: |
           pnpm build
 
+      - name: Download rapidsnark (1c137)
+        run: |
+          mkdir -p ~/rapidsnark/build
+          wget -qO ~/rapidsnark/build/prover https://maci-devops-zkeys.s3.ap-northeast-2.amazonaws.com/rapidsnark-linux-amd64-1c137
+          chmod +x ~/rapidsnark/build/prover
+
+      - name: Download circom Binary v2.1.6
+        run: |
+          wget -qO ${{ github.workspace }}/circom https://github.com/iden3/circom/releases/download/v2.1.6/circom-linux-amd64
+          chmod +x ${{ github.workspace }}/circom
+          sudo mv ${{ github.workspace }}/circom /bin/circom
+
+      - name: Download zkeys
+        run: |
+          pnpm download:test-zkeys
+
       - name: Benchmarks
         run: pnpm benchmarks
diff --git a/circuits/ts/proofs.ts b/circuits/ts/proofs.ts
@@ -51,6 +51,10 @@ export const genProof = async ({
     }
 
     const { proof, publicSignals } = await groth16.fullProve(inputs, wasmPath, zkeyPath);
+
+    // ensure we clean up all threads spawned by snarkjs
+    await cleanThreads();
+
     return { proof, publicSignals };
   }
 

diff --git a/cli/package.json b/cli/package.json
@@ -13,6 +13,7 @@
   ],
   "scripts": {
     "watch": "tsc --watch",
+    "benchmarks": "ts-node tests/benchmarks/benchmarks.ts",
     "build": "tsc -p tsconfig.build.json",
     "postbuild": "cp package.json ./build && mkdir -p ./zkeys",
     "types": "tsc -p tsconfig.json --noEmit",

diff --git a/cli/tests/e2e/e2e.test.ts b/cli/tests/e2e/e2e.test.ts
@@ -315,7 +315,7 @@ describe("e2e tests", function test() {
     });
   });
 
-  describe("5 signups, 1 message", () => {
+  describe("9 signups, 1 message", () => {
     after(() => {
       cleanVanilla();
     });

diff --git a/integrationTests/package.json b/integrationTests/package.json
@@ -11,6 +11,7 @@
   "scripts": {
     "watch": "tsc --watch",
     "build": "tsc",
+    "benchmarks": "ts-node ts/__benchmarks__/benchmarks.ts",
     "types": "tsc -p tsconfig.json --noEmit",
     "test": "ts-mocha --exit ./ts/__tests__/**.test.ts",
     "test:integration": "NODE_OPTIONS=--max-old-space-size=4096 ts-mocha --exit  ./ts/__tests__/integration.test.ts",
@@ -22,6 +23,7 @@
     "@types/chai-as-promised": "^7.1.8",
     "@types/mocha": "^10.0.6",
     "@types/node": "^20.11.2",
+    "benny": "^3.7.1",
     "chai": "^4.3.10",
     "chai-as-promised": "^7.1.1",
     "hardhat": "^2.19.2",