Added aligned_array

virtualsecureplatform · Jul 18, 2024 · faa0592 · faa0592
1 parent a6462b3
commit faa0592
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 4 deletions.
diff --git a/include/detwfa.hpp b/include/detwfa.hpp
@@ -28,9 +28,9 @@ TRGSWFFT<P> TRGSWFFTOneGen()
     return ApplyFFT2trgsw<P>(trgsw);
 }
 
-alignas(32) const TRGSWFFT<lvl1param> trgswonelvl1 =
+alignas(64) const TRGSWFFT<lvl1param> trgswonelvl1 =
     TRGSWFFTOneGen<lvl1param>();
-alignas(32) const TRGSWFFT<lvl2param> trgswonelvl2 =
+alignas(64) const TRGSWFFT<lvl2param> trgswonelvl2 =
     TRGSWFFTOneGen<lvl2param>();
 
 template <class bkP>

diff --git a/include/params.hpp b/include/params.hpp
@@ -8,6 +8,8 @@
 
 namespace TFHEpp {
 
+template<class T, size_t N> struct alignas( 64 ) aligned_array : public std::array<T,N> { };
+
 enum class ErrorDistribution { ModularGaussian, CenteredBinomial };
 
 // Use old 80bit security parameters. It is faster, but not recommended.
@@ -106,7 +108,7 @@ using TRLWERAINTT = std::array<PolynomialRAINTT<P>, P::k + 1>;
 template <class P>
 using TRGSW = std::array<TRLWE<P>, (P::k + 1) * P::l>;
 template <class P>
-using TRGSWFFT = std::array<TRLWEInFD<P>, (P::k + 1) * P::l>;
+using TRGSWFFT = aligned_array<TRLWEInFD<P>, (P::k + 1) * P::l>;
 template <class P>
 using TRGSWNTT = std::array<TRLWENTT<P>, (P::k + 1) * P::l>;
 template <class P>

diff --git a/include/utils.hpp b/include/utils.hpp
@@ -214,4 +214,5 @@ inline void Automorphism(Polynomial<P> &res, const Polynomial<P> &poly,
             res[index & Nmask] += poly[i];
     }
 }
+
 }  // namespace TFHEpp
diff --git a/test/circuitbootstrapping.cpp b/test/circuitbootstrapping.cpp
@@ -38,7 +38,7 @@ int main()
     for (int i = 0; i < num_test; i++) pones[i] = true;
     alignas(64) std::vector<TFHEpp::TRLWE<typename privksP::targetP>> ca(num_test);
     alignas(64) std::vector<TFHEpp::TLWE<typename iksP::domainP>> cones(num_test);
-    std::vector<TFHEpp::TRGSWFFT<typename privksP::targetP>,TFHEpp::AlignedAllocator<TFHEpp::TRGSWFFT<typename privksP::targetP>,64>> bootedTGSW(
+    std::vector<TFHEpp::TRGSWFFT<typename privksP::targetP>> bootedTGSW(
         num_test);
 
     for (int i = 0; i < num_test; i++)
-Original file line number
+Diff line change
@@ Expand Up @@
                 res[index & Nmask] += poly[i];
         }
     }
     }  // namespace TFHEpp