Merge pull request #48 from 0xmozak/sai/enable_serialize_poseidon2_gates

Enable serialization of Poseidon2 gates
0xmozak · Dec 7, 2023 · 1017c30 · 1017c30
2 parents ec95756 + e19f123
commit 1017c30
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/plonky2/src/gates/poseidon2.rs b/plonky2/src/gates/poseidon2.rs
@@ -392,8 +392,8 @@ impl<F: RichField + Extendable<D>, const D: usize> Gate<F, D> for Poseidon2Gate<
     }
 }
 
-#[derive(Debug)]
-struct Poseidon2Generator<F: RichField + Extendable<D> + Poseidon2, const D: usize> {
+#[derive(Debug, Default)]
+pub struct Poseidon2Generator<F: RichField + Extendable<D> + Poseidon2, const D: usize> {
     row: usize,
     _phantom: PhantomData<F>,
 }

diff --git a/plonky2/src/util/serialization/gate_serialization.rs b/plonky2/src/util/serialization/gate_serialization.rs
@@ -103,6 +103,7 @@ pub mod default {
     use crate::gates::multiplication_extension::MulExtensionGate;
     use crate::gates::noop::NoopGate;
     use crate::gates::poseidon::PoseidonGate;
+    use crate::gates::poseidon2::Poseidon2Gate;
     use crate::gates::poseidon_mds::PoseidonMdsGate;
     use crate::gates::public_input::PublicInputGate;
     use crate::gates::random_access::RandomAccessGate;
@@ -127,6 +128,7 @@ pub mod default {
             NoopGate,
             PoseidonMdsGate<F, D>,
             PoseidonGate<F, D>,
+            Poseidon2Gate<F, D>,
             PublicInputGate,
             RandomAccessGate<F, D>,
             ReducingExtensionGate<D>,

diff --git a/plonky2/src/util/serialization/generator_serialization.rs b/plonky2/src/util/serialization/generator_serialization.rs
@@ -112,6 +112,7 @@ pub mod default {
     use crate::gates::lookup_table::LookupTableGenerator;
     use crate::gates::multiplication_extension::MulExtensionGenerator;
     use crate::gates::poseidon::PoseidonGenerator;
+    use crate::gates::poseidon2::Poseidon2Generator;
     use crate::gates::poseidon_mds::PoseidonMdsGenerator;
     use crate::gates::random_access::RandomAccessGenerator;
     use crate::gates::reducing::ReducingGenerator;
@@ -152,6 +153,7 @@ pub mod default {
             MulExtensionGenerator<F, D>,
             NonzeroTestGenerator,
             PoseidonGenerator<F, D>,
+            Poseidon2Generator<F, D>,
             PoseidonMdsGenerator<D>,
             QuotientGeneratorExtension<D>,
             RandomAccessGenerator<F, D>,