[update] ckks example

README.md

@@ -43,20 +43,23 @@ int main() {
     int precision_bits = 30;
     auto params = ckks::create_params(4096, precision_bits);
     CkksSk sk(params);
-    auto relin_key = get_relin_key(sk);
+    auto relin_key = get_relin_key(sk, params.additional_mod);
 
-    auto pt = ckks::encode(1, params);
-    auto ct_sum = ckks::encrypt(pt, sk);
-    for (int i = 2; i <= 100000; i++) {
+    CkksCt ct_sum;
+    for (int i = 1; i <= 100000; i++) {
         auto pt = ckks::encode(1.0 / i, params);
         auto ct = ckks::encrypt(pt, sk);
         auto ct_squared = ckks::mult(ct, ct, relin_key);
-        ct_sum = ckks::add(ct_sum, ct_squared);
+
+        if (i == 1) {
+            ct_sum = ct_squared;
+        } else {
+            ct_sum = ckks::add(ct_sum, ct_squared);
+        }
     }
 
     double sum = ckks::decode(ckks::decrypt(ct_sum, sk));
-    std::cout << "(" << sum << ", " 
-              << M_PI * M_PI / 6 << ")" << std::end;
+    std::cout << "(" << sum << ", " << M_PI * M_PI / 6 << ")" << std::endl;
 }
 
 ```

examples/ckks_example.cpp

@@ -1,6 +1,7 @@
 #include "ckks/ckks.h"
 #include <cmath>
 #include <iostream>
+#include <numeric>
 
 using namespace hehub;
 
@@ -10,13 +11,17 @@ int main() {
     CkksSk sk(params);
     auto relin_key = get_relin_key(sk, params.additional_mod);
 
-    auto pt = ckks::encode(1, params);
-    auto ct_sum = ckks::encrypt(pt, sk);
-    for (int i = 2; i <= 100000; i++) {
+    CkksCt ct_sum;
+    for (int i = 1; i <= 100000; i++) {
         auto pt = ckks::encode(1.0 / i, params);
         auto ct = ckks::encrypt(pt, sk);
         auto ct_squared = ckks::mult(ct, ct, relin_key);
-        ct_sum = ckks::add(ct_sum, ct_squared);
+
+        if (i == 1) {
+            ct_sum = ct_squared;
+        } else {
+            ct_sum = ckks::add(ct_sum, ct_squared);
+        }
     }
 
     double sum = ckks::decode(ckks::decrypt(ct_sum, sk));

src/fhe/bgv/basics.cpp

@@ -1,5 +1,6 @@
 #include "bgv.h"
 #include "common/mod_arith.h"
+#include "common/ntt.h"
 #include "common/rns_transform.h"
 #include <algorithm>
 #include <cmath>

src/fhe/primitives/keys.cpp

@@ -1,5 +1,6 @@
 #include "keys.h"
 #include "common/mod_arith.h"
+#include "common/ntt.h"
 #include "common/rns_transform.h"
 
 namespace hehub {

tests/bgv_t.cpp

@@ -1,5 +1,6 @@
 #include "catch2/catch.hpp"
 #include "common/mod_arith.h"
+#include "common/ntt.h"
 #include "common/sampling.h"
 #include "bgv/bgv.h"
 #include <iostream>

tests/ckks_t.cpp

@@ -2,6 +2,7 @@
 #include "ckks/ckks.h"
 #include "common/bigint.h"
 #include "common/mod_arith.h"
+#include "common/ntt.h"
 #include "common/permutation.h"
 #include "common/sampling.h"
 #include <type_traits>