Added Go template by Sergey Gomenyuk

Co-authored-by: Sergey Gomenyuk <[email protected]>
fairmath · nkaskov · Apr 10, 2024 · Mar 22, 2024 · Mar 22, 2024 · Apr 1, 2024
commit f7b8ab362f44e50dff5a313003c3aeadf186b5c6
diff --git a/cpp/include/cifar/lattigo/config.json b/cpp/include/cifar/lattigo/config.json
@@ -0,0 +1,6 @@
+{
+ "log_n": 17,
+ "log_q": [60, 50, 50, 50, 50, 50, 50, 50, 50, 50, 50, 50, 50],
+ "log_p": [56, 55, 55, 55],
+ "log_default_scale": 50
+}
diff --git a/cpp/include/cifar/lattigo/go.sum b/cpp/include/cifar/lattigo/go.sum
@@ -0,0 +1,35 @@
+github.com/ALTree/bigfloat v0.0.0-20220102081255-38c8b72a9924 h1:DG4UyTVIujioxwJc8Zj8Nabz1L1wTgQ/xNBSQDfdP3I=
+github.com/ALTree/bigfloat v0.0.0-20220102081255-38c8b72a9924/go.mod h1:+NaH2gLeY6RPBPPQf4aRotPPStg+eXc8f9ZaE4vRfD4=
+github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/google/go-cmp v0.5.8 h1:e6P7q2lk1O+qJJb4BtCQXlK8vWEO8V1ZeuEdJNOqZyg=
+github.com/google/go-cmp v0.5.8/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
+github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=
+github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=
+github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
+github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
+github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
+github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.8.0 h1:pSgiaMZlXftHpm5L7V1+rVB+AZJydKsMxsQBIJw4PKk=
+github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
+github.com/tuneinsight/lattigo/v5 v5.0.2 h1:g+WmQK0G04nldAPnthgBFsfMtCJRLcfwME+cR9hLuIg=
+github.com/tuneinsight/lattigo/v5 v5.0.2/go.mod h1:mmynIHGOeVRGYzuHRdZNbDLJWe8WHDYCgYkl9WZrHN0=
+golang.org/x/crypto v0.0.0-20220926161630-eccd6366d1be h1:fmw3UbQh+nxngCAHrDCCztao/kbYFnWjoqop8dHx05A=
+golang.org/x/crypto v0.0.0-20220926161630-eccd6366d1be/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=
+golang.org/x/exp v0.0.0-20230321023759-10a507213a29 h1:ooxPy7fPvB4kwsA2h+iBNHkAbp/4JxTSwCmvdjEYmug=
+golang.org/x/exp v0.0.0-20230321023759-10a507213a29/go.mod h1:CxIveKay+FTh1D0yPZemJVgC/95VzuuOLq5Qi4xnoYc=
+golang.org/x/sys v0.1.0 h1:kunALQeHf1/185U1i0GOB/fy1IPRDDpuoOOqRReG57U=
+golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
+gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
diff --git a/cpp/include/cifar/lattigo/internal/solution/solution.go b/cpp/include/cifar/lattigo/internal/solution/solution.go
@@ -0,0 +1,19 @@
+package solution
+
+import (
+ "io"
+
+ "github.com/tuneinsight/lattigo/v5/core/rlwe"
+ "github.com/tuneinsight/lattigo/v5/he/hefloat"
+)
+
+func SolveTestcase(
+ params *hefloat.Parameters,
+ evl *rlwe.MemEvaluationKeySet,
+ pk *rlwe.PublicKey,
+ cipher *rlwe.Ciphertext,
+ out io.Writer,
+) (*rlwe.Ciphertext, error) {
+ // Put your solution here
+ return nil, nil
+}
diff --git a/cpp/include/cifar/lattigo/main.go b/cpp/include/cifar/lattigo/main.go
@@ -0,0 +1,128 @@
+package main
+
+import (
+ "errors"
+ "flag"
+ "io"
+ "log"
+ "os"
+
+ "github.com/tuneinsight/lattigo/v5/core/rlwe"
+ "github.com/tuneinsight/lattigo/v5/he/hefloat"
+
+ "app/internal/solution"
+)
+
+func main() {
+ ccFile := flag.String("cc", "", "")
+ evalFile := flag.String("key_eval", "", "")
+ inputFile := flag.String("input", "", "")
+ outputFile := flag.String("output", "", "")
+ pkFile := flag.String("key_public", "", "")
+
+ flag.Parse()
+
+ params, err := readContext(ccFile)
+ if err != nil {
+ log.Fatalf("read context error: %s", err.Error())
+ }
+
+ evl, err := readEvaluationKeySet(evalFile)
+ if err != nil {
+ log.Fatalf("read evaluation key set error: %s", err.Error())
+ }
+
+ pk, err := readPublicKey(pkFile)
+ if err != nil {
+ log.Fatalf("read public key error: %s", err.Error())
+ }
+
+ in, err := readInputFile(inputFile)
+ if err != nil {
+ log.Fatalf("read input file error: %s", err.Error())
+ }
+
+ out, err := os.Create(*outputFile)
+ if err != nil {
+ log.Fatalf("create output file error: %s", err.Error())
+ }
+
+ defer out.Close()
+
+ res, err := solution.SolveTestcase(params, evl, pk, in, out)
+ if err != nil {
+ log.Fatalf("solution error: %s", err)
+ }
+
+ if _, err = res.WriteTo(out); err != nil {
+ log.Fatalf("can't write resulted ciphertext: %s", err.Error())
+ }
+}
+
+func readContext(ccFile *string) (*hefloat.Parameters, error) {
+ if ccFile == nil {
+ return nil, errors.New("ccFile is nil")
+ }
+
+ ccData, err := os.ReadFile(*ccFile)
+ if err != nil {
+ return nil, errors.New("can't read context")
+ }
+
+ params := &hefloat.Parameters{}
+ if err := params.UnmarshalBinary(ccData); err != nil {
+ return nil, errors.New("can't unmarshal context")
+ }
+
+ return params, nil
+}
+
+func readEvaluationKeySet(evalFile *string) (*rlwe.MemEvaluationKeySet, error) {
+ if evalFile == nil {
+ return nil, errors.New("evalFile is nil")
+ }
+
+ evalF, err := os.Open(*evalFile)
+ if err != nil {
+ return nil, errors.New("can't open evaluation key set file")
+ }
+ defer evalF.Close()
+
+ evl := &rlwe.MemEvaluationKeySet{}
+
+ if _, err = evl.ReadFrom(evalF); err != nil && !errors.Is(err, io.EOF) {
+ return nil, errors.New("can't read evaluation key set file")
+ }
+
+ return evl, nil
+}
+
+func readPublicKey(pkFile *string) (*rlwe.PublicKey, error) {
+ pkF, err := os.Open(*pkFile)
+ if err != nil {
+ return nil, errors.New("can't open public key file")
+ }
+ defer pkF.Close()
+
+ pk := &rlwe.PublicKey{}
+ if _, err = pk.ReadFrom(pkF); err != nil && !errors.Is(err, io.EOF) {
+ return nil, errors.New("can't read public key file")
+ }
+
+ return pk, nil
+}
+
+func readInputFile(inputFile *string) (*rlwe.Ciphertext, error) {
+ inputF, err := os.Open(*inputFile)
+ if err != nil {
+ return nil, errors.New("can't open input ciphertext")
+ }
+ defer inputF.Close()
+
+ in := &rlwe.Ciphertext{}
+ if _, err := in.ReadFrom(inputF); err != nil && !errors.Is(err, io.EOF) {
+ return nil, errors.New("can't read input ciphertext")
+ }
+
+ return in, nil
+}
diff --git a/cpp/include/cifar/cifar_ckks.h → cpp/include/cifar/openfhe/cifar_ckks.h b/cpp/include/cifar/cifar_ckks.h → cpp/include/cifar/openfhe/cifar_ckks.h
diff --git a/cpp/include/cifar/config.json → cpp/include/cifar/openfhe/config.json b/cpp/include/cifar/config.json → cpp/include/cifar/openfhe/config.json
diff --git a/cpp/include/parity/lattigo/config.json b/cpp/include/parity/lattigo/config.json
@@ -0,0 +1,6 @@
+{
+ "log_n": 17,
+ "log_q": [60, 50, 50, 50, 50, 50, 50, 50, 50, 50, 50, 50, 50],
+ "log_p": [56, 55, 55, 55],
+ "log_default_scale": 50
+}
diff --git a/cpp/include/parity/lattigo/go.sum b/cpp/include/parity/lattigo/go.sum
@@ -0,0 +1,35 @@
+github.com/ALTree/bigfloat v0.0.0-20220102081255-38c8b72a9924 h1:DG4UyTVIujioxwJc8Zj8Nabz1L1wTgQ/xNBSQDfdP3I=
+github.com/ALTree/bigfloat v0.0.0-20220102081255-38c8b72a9924/go.mod h1:+NaH2gLeY6RPBPPQf4aRotPPStg+eXc8f9ZaE4vRfD4=
+github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/google/go-cmp v0.5.8 h1:e6P7q2lk1O+qJJb4BtCQXlK8vWEO8V1ZeuEdJNOqZyg=
+github.com/google/go-cmp v0.5.8/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
+github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=
+github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=
+github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
+github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
+github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
+github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.8.0 h1:pSgiaMZlXftHpm5L7V1+rVB+AZJydKsMxsQBIJw4PKk=
+github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
+github.com/tuneinsight/lattigo/v5 v5.0.2 h1:g+WmQK0G04nldAPnthgBFsfMtCJRLcfwME+cR9hLuIg=
+github.com/tuneinsight/lattigo/v5 v5.0.2/go.mod h1:mmynIHGOeVRGYzuHRdZNbDLJWe8WHDYCgYkl9WZrHN0=
+golang.org/x/crypto v0.0.0-20220926161630-eccd6366d1be h1:fmw3UbQh+nxngCAHrDCCztao/kbYFnWjoqop8dHx05A=
+golang.org/x/crypto v0.0.0-20220926161630-eccd6366d1be/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=
+golang.org/x/exp v0.0.0-20230321023759-10a507213a29 h1:ooxPy7fPvB4kwsA2h+iBNHkAbp/4JxTSwCmvdjEYmug=
+golang.org/x/exp v0.0.0-20230321023759-10a507213a29/go.mod h1:CxIveKay+FTh1D0yPZemJVgC/95VzuuOLq5Qi4xnoYc=
+golang.org/x/sys v0.1.0 h1:kunALQeHf1/185U1i0GOB/fy1IPRDDpuoOOqRReG57U=
+golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
+gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
diff --git a/cpp/include/parity/lattigo/internal/solution/solution.go b/cpp/include/parity/lattigo/internal/solution/solution.go
@@ -0,0 +1,19 @@
+package solution
+
+import (
+ "io"
+
+ "github.com/tuneinsight/lattigo/v5/core/rlwe"
+ "github.com/tuneinsight/lattigo/v5/he/hefloat"
+)
+
+func SolveTestcase(
+ params *hefloat.Parameters,
+ evl *rlwe.MemEvaluationKeySet,
+ pk *rlwe.PublicKey,
+ cipher *rlwe.Ciphertext,
+ out io.Writer,
+) (*rlwe.Ciphertext, error) {
+ // Put your solution here
+ return nil, nil
+}
diff --git a/cpp/include/parity/lattigo/main.go b/cpp/include/parity/lattigo/main.go
@@ -0,0 +1,128 @@
+package main
+
+import (
+ "errors"
+ "flag"
+ "io"
+ "log"
+ "os"
+
+ "github.com/tuneinsight/lattigo/v5/core/rlwe"
+ "github.com/tuneinsight/lattigo/v5/he/hefloat"
+
+ "app/internal/solution"
+)
+
+func main() {
+ ccFile := flag.String("cc", "", "")
+ evalFile := flag.String("key_eval", "", "")
+ inputFile := flag.String("input", "", "")
+ outputFile := flag.String("output", "", "")
+ pkFile := flag.String("key_public", "", "")
+
+ flag.Parse()
+
+ params, err := readContext(ccFile)
+ if err != nil {
+ log.Fatalf("read context error: %s", err.Error())
+ }
+
+ evl, err := readEvaluationKeySet(evalFile)
+ if err != nil {
+ log.Fatalf("read evaluation key set error: %s", err.Error())
+ }
+
+ pk, err := readPublicKey(pkFile)
+ if err != nil {
+ log.Fatalf("read public key error: %s", err.Error())
+ }
+
+ in, err := readInputFile(inputFile)
+ if err != nil {
+ log.Fatalf("read input file error: %s", err.Error())
+ }
+
+ out, err := os.Create(*outputFile)
+ if err != nil {
+ log.Fatalf("create output file error: %s", err.Error())
+ }
+
+ defer out.Close()
+
+ res, err := solution.SolveTestcase(params, evl, pk, in, out)
+ if err != nil {
+ log.Fatalf("solution error: %s", err)
+ }
+
+ if _, err = res.WriteTo(out); err != nil {
+ log.Fatalf("can't write resulted ciphertext: %s", err.Error())
+ }
+}
+
+func readContext(ccFile *string) (*hefloat.Parameters, error) {
+ if ccFile == nil {
+ return nil, errors.New("ccFile is nil")
+ }
+
+ ccData, err := os.ReadFile(*ccFile)
+ if err != nil {
+ return nil, errors.New("can't read context")
+ }
+
+ params := &hefloat.Parameters{}
+ if err := params.UnmarshalBinary(ccData); err != nil {
+ return nil, errors.New("can't unmarshal context")
+ }
+
+ return params, nil
+}
+
+func readEvaluationKeySet(evalFile *string) (*rlwe.MemEvaluationKeySet, error) {
+ if evalFile == nil {
+ return nil, errors.New("evalFile is nil")
+ }
+
+ evalF, err := os.Open(*evalFile)
+ if err != nil {
+ return nil, errors.New("can't open evaluation key set file")
+ }
+ defer evalF.Close()
+
+ evl := &rlwe.MemEvaluationKeySet{}
+
+ if _, err = evl.ReadFrom(evalF); err != nil && !errors.Is(err, io.EOF) {
+ return nil, errors.New("can't read evaluation key set file")
+ }
+
+ return evl, nil
+}
+
+func readPublicKey(pkFile *string) (*rlwe.PublicKey, error) {
+ pkF, err := os.Open(*pkFile)
+ if err != nil {
+ return nil, errors.New("can't open public key file")
+ }
+ defer pkF.Close()
+
+ pk := &rlwe.PublicKey{}
+ if _, err = pk.ReadFrom(pkF); err != nil && !errors.Is(err, io.EOF) {
+ return nil, errors.New("can't read public key file")
+ }
+
+ return pk, nil
+}
+
+func readInputFile(inputFile *string) (*rlwe.Ciphertext, error) {
+ inputF, err := os.Open(*inputFile)
+ if err != nil {
+ return nil, errors.New("can't open input ciphertext")
+ }
+ defer inputF.Close()
+
+ in := &rlwe.Ciphertext{}
+ if _, err := in.ReadFrom(inputF); err != nil && !errors.Is(err, io.EOF) {
+ return nil, errors.New("can't read input ciphertext")
+ }
+
+ return in, nil
+}
diff --git a/cpp/include/parity/config.json → cpp/include/parity/openfhe/config.json b/cpp/include/parity/config.json → cpp/include/parity/openfhe/config.json
diff --git a/cpp/include/parity/parity_ckks.h → cpp/include/parity/openfhe/parity_ckks.h b/cpp/include/parity/parity_ckks.h → cpp/include/parity/openfhe/parity_ckks.h
diff --git a/cpp/include/relu/config.json b/cpp/include/relu/config.json
@@ -0,0 +1,9 @@
+{
+ "indexes_for_rotation_key": [
+ 1,
+ 2,
+ 3,
+ 4,
+ 5
+ ]
+ }