[update] benchmarks

Author: ppppbamzy

README.md

@@ -67,14 +67,14 @@ int main() {
 ## Benchmarks
 We tested the performance of HEhub compiled with Clang-12.0.5 and run on an Intel i7-9750H @ 2.60GHz. _Note: The code for benchmark is still incomplete since our effort is limited currently. We will list more benchmark results later._
 
-| parameter set |  NTT  |  INTT  |
-| ------------- |  ---  |  ----  |
-| N = 1024      |  7 us |   9 us |
-| N = 2048      | 15 us |  19 us |
-| N = 4096      | 31 us | 42 us |
-| N = 8192      | 69 us | 97 us |
-| N = 16384     | 144 us| 196 us |
-| N = 32768     | 313 us| 470 us |
+| parameter set |  NTT  |  INTT  | CKKS<br>encode |
+| ------------- |  ---  |  ----  | -------------- |
+| N = 1024      |  7 us |   9 us | |
+| N = 2048      | 14 us |  19 us |  |
+| N = 4096      | 30 us | 37 us |         138 us |
+| N = 8192      | 68 us | 85 us |         309 us |
+| N = 16384     | 142 us| 195 us |         838 us |
+| N = 32768     | 330 us| 406 us |        3952 us |
 
 ## How to contribute
 If you want to contribute to this project, feel free to create an issue at our [Issue](https://github.com/primihub/primihub/issues) page (e.g., documentation, new idea and proposal).

bench/CMakeLists.txt

@@ -1,4 +1,4 @@
-add_executable(benchmarks benchmarks.cpp mod_arith_bm.cpp ntt_bm.cpp)
+add_executable(benchmarks benchmarks.cpp mod_arith_bm.cpp ntt_bm.cpp ckks_bm.cpp)
 add_definitions(-DCATCH_CONFIG_ENABLE_BENCHMARKING)
 target_link_libraries(benchmarks PUBLIC hehub)
 target_include_directories(benchmarks PUBLIC ${PROJECT_SOURCE_DIR}/src)

bench/ckks_bm.cpp

@@ -0,0 +1,21 @@
+#include "catch2/catch.hpp"
+#include "ckks/ckks.h"
+#include "common/ntt.h"
+
+using namespace hehub;
+using namespace std;
+
+TEST_CASE("benchmark ckks") {
+ for (auto [LOGN, scaling_bits] :
+         {pair{12, 36}, pair{13, 43}, pair{14, 48}, pair{15, 55}}) {
+ u64 N = 1 << LOGN;
+ auto params = ckks::create_params(N, scaling_bits);
+ cache_ntt_factors_strict(LOGN, params.moduli);
+
+        std::vector<cc_double> data(N / 2);
+ BENCHMARK(string("CKKS encode / N=") + to_string(N) +
+ string(" / scaling=2^") + to_string(scaling_bits)) {
+ return ckks::simd_encode(data, params);
+        };
+    }
+}