mirror of
https://github.com/luxfi/lattice.git
synced 2026-09-24 00:25:44 +00:00
zeekay
eaf864890e
- Update module path to github.com/luxfi/lattice/v7 - Add GPU acceleration with Metal (macOS) and CUDA backends - Enhance threshold FHE operations - Revert to vanilla Apache 2.0 license (remove tfhe) - Update Go version to 1.25.5 - Merge upstream lattigo improvements
113 lines
2.7 KiB
Go
113 lines
2.7 KiB
Go
package bootstrapping
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/luxfi/lattice/v7/core/rlwe"
|
|
|
|
"github.com/luxfi/lattice/v7/schemes/ckks"
|
|
)
|
|
|
|
func BenchmarkConcurrentBootstrap(b *testing.B) {
|
|
paramSet := DefaultParametersDense[0]
|
|
|
|
params, err := ckks.NewParametersFromLiteral(paramSet.SchemeParams)
|
|
require.NoError(b, err)
|
|
|
|
btpParams, err := NewParametersFromLiteral(params, paramSet.BootstrappingParams)
|
|
require.Nil(b, err)
|
|
|
|
kgen := rlwe.NewKeyGenerator(params)
|
|
sk := kgen.GenSecretKeyNew()
|
|
|
|
evk, _, err := btpParams.GenEvaluationKeys(sk)
|
|
require.NoError(b, err)
|
|
|
|
// Benchmark parallel bootstrapping
|
|
b.Run(ParamsToString(params, btpParams.LogMaxDimensions().Cols, "Bootstrap/"), func(b *testing.B) {
|
|
var err error
|
|
eval, err := NewEvaluator(btpParams, evk)
|
|
require.NoError(b, err)
|
|
ct1 := ckks.NewCiphertext(params, 1, 0)
|
|
b.ResetTimer()
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
_, err = eval.Bootstrap(ct1)
|
|
}
|
|
})
|
|
})
|
|
|
|
}
|
|
|
|
func BenchmarkBootstrap(b *testing.B) {
|
|
|
|
b.Skip()
|
|
paramSet := DefaultParametersDense[0]
|
|
|
|
params, err := ckks.NewParametersFromLiteral(paramSet.SchemeParams)
|
|
require.NoError(b, err)
|
|
|
|
btpParams, err := NewParametersFromLiteral(params, paramSet.BootstrappingParams)
|
|
require.Nil(b, err)
|
|
|
|
kgen := rlwe.NewKeyGenerator(params)
|
|
sk := kgen.GenSecretKeyNew()
|
|
|
|
evk, _, err := btpParams.GenEvaluationKeys(sk)
|
|
require.NoError(b, err)
|
|
|
|
b.Run(ParamsToString(params, btpParams.LogMaxDimensions().Cols, "Bootstrap/"), func(b *testing.B) {
|
|
|
|
var err error
|
|
eval, err := NewEvaluator(btpParams, evk)
|
|
require.NoError(b, err)
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
|
|
b.StopTimer()
|
|
ct := ckks.NewCiphertext(params, 1, 0)
|
|
b.StartTimer()
|
|
|
|
var t time.Time
|
|
var ct0, ct1 *rlwe.Ciphertext
|
|
|
|
// ScaleDown
|
|
t = time.Now()
|
|
ct, _, err = eval.ScaleDown(ct)
|
|
require.NoError(b, err)
|
|
b.Log("ScaleDown:", time.Since(t), ct.Level(), ct.Scale.Float64())
|
|
|
|
// ModUp ct_{Q_0} -> ct_{Q_L}
|
|
t = time.Now()
|
|
ct, err = eval.ModUp(ct)
|
|
require.NoError(b, err)
|
|
b.Log("ModUp :", time.Since(t), ct.Level(), ct.Scale.Float64())
|
|
|
|
// Part 1 : Coeffs to slots
|
|
t = time.Now()
|
|
ct0, ct1, err = eval.CoeffsToSlots(ct)
|
|
require.NoError(b, err)
|
|
b.Log("CtS :", time.Since(t), ct0.Level(), ct0.Scale.Float64())
|
|
|
|
// Part 2 : SineEval
|
|
t = time.Now()
|
|
ct0, err = eval.EvalMod(ct0)
|
|
require.NoError(b, err)
|
|
if ct1 != nil {
|
|
ct1, err = eval.EvalMod(ct1)
|
|
require.NoError(b, err)
|
|
}
|
|
b.Log("EvalMod :", time.Since(t), ct0.Level(), ct0.Scale.Float64())
|
|
|
|
// Part 3 : Slots to coeffs
|
|
t = time.Now()
|
|
ct0, err = eval.SlotsToCoeffs(ct0, ct1)
|
|
require.NoError(b, err)
|
|
b.Log("StC :", time.Since(t), ct0.Level(), ct0.Scale.Float64())
|
|
}
|
|
})
|
|
}
|