samll fix

Layr-Labs · Jul 16, 2024 · 2a6acaa · 2a6acaa
1 parent 73efd4b
commit 2a6acaa
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/encoding/kzg/prover/parametrized_prover.go b/encoding/kzg/prover/parametrized_prover.go
@@ -19,7 +19,6 @@ type ParametrizedProver struct {
 	*rs.Encoder
 
 	*kzg.KzgConfig
-	Ks *kzg.KZGSettings
 
 	Fs         *fft.FFTSettings
 	Ks         *kzg.KZGSettings
@@ -164,7 +163,7 @@ func (g *ParametrizedProver) Encode(inputFr []fr.Element) (*bn254.G1Affine, *bn2
 			lengthCommitmentResult.Duration,
 			lengthProofResult.Duration,
 			proofsResult.Duration,
-			len(g.Srs.G2),
+			g.SRSOrder,
 			g.SRSOrder-uint64(len(inputFr)),
 		)
 	}

diff --git a/encoding/kzg/prover/prover.go b/encoding/kzg/prover/prover.go
@@ -287,9 +287,9 @@ func (g *Prover) newProver(params encoding.EncodingParams) (*ParametrizedProver,
 	return &ParametrizedProver{
 		Encoder:    encoder,
 		KzgConfig:  g.KzgConfig,
-		Srs:        g.Srs,
-		G2Trailing: g.G2Trailing,
 		Fs:         fs,
+		Ks:         ks,
+		SFs:        sfs,
 		FFTPointsT: fftPointsT,
 		UseGpu:     g.UseGpu,
 		Computer:   computer,

diff --git a/encoding/test/main.go b/encoding/test/main.go
@@ -98,7 +98,7 @@ func TestKzgRs() {
 	// create encoding object
 	p, _ := prover.NewProver(kzgConfig, true)
 
-	p.UseGpu = false
+	p.UseGpu = true
 
 	params := encoding.EncodingParams{NumChunks: numNode, ChunkLength: uint64(numSymbols) / uint64(numSys)}
 	enc, _ := p.GetKzgEncoder(params)