Batched commitment of polys with different sizes on mpcs layer

mpcs/Cargo.toml

@@ -25,12 +25,15 @@ multilinear_extensions = { path = "../multilinear_extensions" }
 num-bigint = "0.4"
 num-integer = "0.1"
 p3 = { path = "../p3" }
+p3-field.workspace = true
+p3-goldilocks.workspace = true
 plonky2.workspace = true
 poseidon.workspace = true
 rand.workspace = true
 rand_chacha.workspace = true
 rayon = { workspace = true, optional = true }
 serde.workspace = true
+sumcheck = { path = "../sumcheck" }
 transcript = { path = "../transcript" }
 whir = { path = "../whir", features = ["ceno"] }
 witness = { path = "../witness" }

mpcs/src/basefold.rs

@@ -352,11 +352,10 @@ where
         ret
     }
 
-    fn batch_commit(
+    fn batch_commit_polys(
         pp: &Self::ProverParam,
-        rmm: witness::RowMajorMatrix<<E as ff_ext::ExtensionField>::BaseField>,
+        polys: &[DenseMultilinearExtension<E>],
     ) -> Result<Self::CommitmentWithWitness, Error> {
-        let polys = rmm.to_mles();
         // assumptions
         // 1. there must be at least one polynomial
         // 2. all polynomials must exist in the same field type
@@ -1115,8 +1114,7 @@ mod test {
     use crate::{
         basefold::Basefold,
         test_util::{
-            gen_rand_poly_base, gen_rand_poly_ext, run_batch_commit_open_verify,
-            run_commit_open_verify, run_simple_batch_commit_open_verify,
+            gen_rand_poly_base, gen_rand_poly_ext, run_batch_commit_open_verify, run_commit_open_verify, run_diff_size_batch_commit_open_verify, run_simple_batch_commit_open_verify
         },
     };
 
@@ -1216,4 +1214,15 @@ mod test {
             );
         }
     }
+
+    #[test]
+    fn batch_commit_diff_size_open_verify() {
+        let gen_rand_poly = gen_rand_poly_base;
+        run_diff_size_batch_commit_open_verify::<GoldilocksExt2, PcsGoldilocksBaseCode>(
+            gen_rand_poly,
+            20,
+            3,
+            5,
+        );
+    }
 }