add gc load

backend/groth16/bls12-377/icicle/icicle.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"math/big"
 	"math/bits"
+	"runtime"
 	"sync"
 	"time"
 	"unsafe"
@@ -50,6 +51,9 @@ func (pk *ProvingKey) setupDevicePointers() error {
 	if pk.deviceInfo != nil {
 		return nil
 	}
+
+	runtime.GC()
+
 	pk.deviceInfo = &deviceInfo{}
 	n := int(pk.Domain.Cardinality)
 	sizeBytes := n * fr.Bytes
@@ -111,6 +115,7 @@ func (pk *ProvingKey) setupDevicePointers() error {
 
 	// free
 	pk.Domain = fft.Domain{Cardinality: pk.Domain.Cardinality}
+	runtime.GC()
 
 	/*************************  Start G1 Device Setup  ***************************/
 	/*************************     A      ***************************/
@@ -119,13 +124,15 @@ func (pk *ProvingKey) setupDevicePointers() error {
 	go iciclegnark.CopyPointsToDevice(pk.G1.A, pointsBytesA, copyADone) // Make a function for points
 	pk.G1Device.A = <-copyADone
 	pk.G1.A = nil
+	runtime.GC()
 
 	/*************************     B      ***************************/
 	pointsBytesB := len(pk.G1.B) * fp.Bytes * 2
 	copyBDone := make(chan unsafe.Pointer, 1)
 	go iciclegnark.CopyPointsToDevice(pk.G1.B, pointsBytesB, copyBDone) // Make a function for points
 	pk.G1Device.B = <-copyBDone
 	pk.G1.B = nil
+	runtime.GC()
 
 	/*************************     K      ***************************/
 	var pointsNoInfinity []curve.G1Affine
@@ -142,20 +149,23 @@ func (pk *ProvingKey) setupDevicePointers() error {
 	go iciclegnark.CopyPointsToDevice(pointsNoInfinity, pointsBytesK, copyKDone) // Make a function for points
 	pk.G1Device.K = <-copyKDone
 	pk.G1.K = nil
+	runtime.GC()
 
 	/*************************     Z      ***************************/
 	pointsBytesZ := len(pk.G1.Z) * fp.Bytes * 2
 	copyZDone := make(chan unsafe.Pointer, 1)
 	go iciclegnark.CopyPointsToDevice(pk.G1.Z, pointsBytesZ, copyZDone) // Make a function for points
 	pk.G1Device.Z = <-copyZDone
 	pk.G1.Z = make([]curve.G1Affine, 1)
+	runtime.GC()
 
 	/*************************  Start G2 Device Setup  ***************************/
 	pointsBytesB2 := len(pk.G2.B) * fp.Bytes * 4
 	copyG2BDone := make(chan unsafe.Pointer, 1)
 	go iciclegnark.CopyG2PointsToDevice(pk.G2.B, pointsBytesB2, copyG2BDone) // Make a function for points
 	pk.G2Device.B = <-copyG2BDone
 	pk.G2.B = nil
+	runtime.GC()
 
 	return nil
 }
@@ -225,13 +235,13 @@ func Prove(r1cs *cs.R1CS, pk *ProvingKey, fullWitness witness.Witness, opts ...b
 			solver.OverrideHint(r1cs.GkrInfo.ProveHintID, cs.GkrProveHint(r1cs.GkrInfo.HashName, &gkrData)))
 	}
 
-	//solveLock.Lock()
+	solveLock.Lock()
 	_solution, err := r1cs.Solve(fullWitness, solverOpts...)
 	if err != nil {
-		//solveLock.Unlock()
+		solveLock.Unlock()
 		return nil, err
 	}
-	//solveLock.Unlock()
+	solveLock.Unlock()
 
 	solution := _solution.(*cs.R1CSSolution)
 	wireValues := []fr.Element(solution.W)

backend/groth16/groth16.go

@@ -224,7 +224,9 @@ func Setup(r1cs constraint.ConstraintSystem) (ProvingKey, VerifyingKey, error) {
 			return nil, nil, err
 		}
 		if icicle_bls12377.HasIcicle {
-			return &icicle_bls12377.ProvingKey{ProvingKey: &pk}, &vk, nil
+			iciclePk := &icicle_bls12377.ProvingKey{ProvingKey: &pk}
+			err := icicle_bls12377.SetupDevicePointers(iciclePk)
+			return iciclePk, &vk, err
 		}
 		return &pk, &vk, nil
 	case *cs_bls12381.R1CS: