From a3a139317ed2e511adce54ef0b099b105f0f070a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9ment=20=27birdy=27=20Danjou?= <clement@danjou.io>
Date: Tue, 9 Jan 2024 20:08:53 +0100
Subject: [PATCH] feat() add benchmarks

---
 Makefile                 |  4 +++
 fhevm/benchmarks_test.go | 70 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 74 insertions(+)
 create mode 100644 fhevm/benchmarks_test.go

diff --git a/Makefile b/Makefile
index d83dff6..b630fca 100644
--- a/Makefile
+++ b/Makefile
@@ -6,6 +6,10 @@ build: build-tfhe-rs-capi
 test: build-tfhe-rs-capi
 	cd fhevm && go test -v .
 
+.PHONY: benchmarks
+benchmarks: build-tfhe-rs-capi
+	cd fhevm && go test -v . -run Benchmarks
+
 .PHONY: build-tfhe-rs-capi
 build-tfhe-rs-capi:
 	cd tfhe-rs && RUSTFLAGS="" make build_c_api_experimental_deterministic_fft \
diff --git a/fhevm/benchmarks_test.go b/fhevm/benchmarks_test.go
new file mode 100644
index 0000000..4458395
--- /dev/null
+++ b/fhevm/benchmarks_test.go
@@ -0,0 +1,70 @@
+// Copyright 2014 The go-ethereum Authors
+// This file is part of the go-ethereum library.
+//
+// The go-ethereum library is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Lesser General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// The go-ethereum library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public License
+// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
+
+package fhevm
+
+import (
+	"testing"
+	"time"
+)
+
+type operation func(FheUintType)
+
+
+
+func convertInGas(t *testing.T, name string, elapsed [3]time.Duration) {
+	sum := int64(0)
+	for i := 0; i < 3; i++ {
+			sum += int64(elapsed[i])
+	}
+		
+	avg := sum / 3
+	gasUsed := avg / 1000 // 1s = 1 000 000 gas
+	gasUsed = gasUsed / 7  * 10 // 1s = 100k
+	gasUsed = gasUsed / 1000 // divide to round it
+  t.Logf("%s in %s => %d", name, elapsed, gasUsed * 1000)
+}
+
+func runTest(t *testing.T, name string, fn operation, bits string, fheUintType FheUintType) {
+	var elapsed [3]time.Duration
+	n := 0
+	for n < 3 {
+		start := time.Now()
+		fn(fheUintType)
+  	elapsed[n] = time.Since(start)
+		n += 1
+	}
+	convertInGas(t, name + bits, elapsed)
+}
+
+func benchTests(t *testing.T, name string, fn operation) {
+	runTest(t, name, fn, "8", FheUint8)
+	runTest(t, name, fn, "16", FheUint16)
+	runTest(t, name, fn, "32", FheUint32)
+}
+
+func TestBenchmarks(t *testing.T) {
+	benchTests(t, "add", func(fheUintType FheUintType) { FheAdd(t, fheUintType, false) })
+	benchTests(t, "ScalarAdd", func(fheUintType FheUintType) { FheAdd(t, fheUintType, true) })
+
+	benchTests(t, "sub", func(fheUintType FheUintType) { FheSub(t, fheUintType, false) })
+	benchTests(t, "ScalarSub", func(fheUintType FheUintType) { FheSub(t, fheUintType, true) })
+
+	benchTests(t, "mul", func(fheUintType FheUintType) { FheMul(t, fheUintType, false) })
+	benchTests(t, "ScalarMul", func(fheUintType FheUintType) { FheMul(t, fheUintType, true) })
+
+	benchTests(t, "ScalarDiv", func(fheUintType FheUintType) { FheDiv(t, fheUintType, true) })
+}