build generify parallelization

Consensys · Dec 19, 2024 · be02fb6 · be02fb6
1 parent dd5cac2
commit be02fb6
Show file tree

Hide file tree

Showing 7 changed files with 77 additions and 70 deletions.
diff --git a/backend/groth16/bls12-377/mpcsetup/phase2.go b/backend/groth16/bls12-377/mpcsetup/phase2.go
diff --git a/backend/groth16/bls12-381/mpcsetup/phase2.go b/backend/groth16/bls12-381/mpcsetup/phase2.go
diff --git a/backend/groth16/bls24-315/mpcsetup/phase2.go b/backend/groth16/bls24-315/mpcsetup/phase2.go
diff --git a/backend/groth16/bls24-317/mpcsetup/phase2.go b/backend/groth16/bls24-317/mpcsetup/phase2.go
diff --git a/backend/groth16/bw6-633/mpcsetup/phase2.go b/backend/groth16/bw6-633/mpcsetup/phase2.go
diff --git a/backend/groth16/bw6-761/mpcsetup/phase2.go b/backend/groth16/bw6-761/mpcsetup/phase2.go
diff --git a/internal/generator/backend/template/zkpschemes/groth16/mpcsetup/phase2.go.tmpl b/internal/generator/backend/template/zkpschemes/groth16/mpcsetup/phase2.go.tmpl
@@ -2,6 +2,7 @@ import (
 	"bytes"
 	"crypto/sha256"
 	"errors"
+	"github.com/consensys/gnark/internal/utils"
 	"math/big"
 	"fmt"
 	"slices"
@@ -115,14 +116,18 @@ func (p *Phase2) update(delta *fr.Element, sigma []fr.Element) {
 			panic("unknown type")
 		}
 	}
+	scaleG1Slice := func(s []curve.G1Affine) {
+		utils.Parallelize(len(s), func(start, end int) {
+			for i := start; i < end; i++ {
+				s[i].ScalarMultiplication(&s[i], &I)
+			}
+		})
+	}
 
 	for i := range sigma {
 		sigma[i].BigInt(&I)
-		s := p.Parameters.G1.SigmaCKK[i]
-		for j := range s {
-			scale(&s[j])
-		}
 		scale(&p.Parameters.G2.Sigma[i])
+		scaleG1Slice(p.Parameters.G1.SigmaCKK[i])
 	}
 
 	delta.BigInt(&I)
@@ -131,12 +136,8 @@ func (p *Phase2) update(delta *fr.Element, sigma []fr.Element) {
 
 	delta.Inverse(delta)
 	delta.BigInt(&I)
-	for i := range p.Parameters.G1.Z {
-		scale(&p.Parameters.G1.Z[i])
-	}
-	for i := range p.Parameters.G1.PKK {
-		scale(&p.Parameters.G1.PKK[i])
-	}
+	scaleG1Slice(p.Parameters.G1.Z)
+	scaleG1Slice(p.Parameters.G1.PKK)
 }
 
 func (p *Phase2) Contribute() {