return ctIn when n=1, batchSize=k*N

tuneinsight · Nov 21, 2024 · b67c72b · b67c72b
1 parent 141c402
commit b67c72b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/schemes/bgv/evaluator.go b/schemes/bgv/evaluator.go
@@ -1515,6 +1515,13 @@ func (eval Evaluator) InnerSum(ctIn *rlwe.Ciphertext, batchSize, n int, opOut *r
 	l := n * batchSize
 
 	if l%N == 0 {
+		if n == 1 {
+			if ctIn != opOut {
+				opOut.Copy(ctIn)
+			}
+			return
+		}
+
 		if err = eval.Evaluator.InnerSum(ctIn, batchSize, n/2, opOut); err != nil {
 			return
 		}