chore: add fast_mul benchmark

Author: 0xLucqs

crates/math/benches/polynomials/polynomial.rs

@@ -1,8 +1,10 @@
-use super::utils::{rand_field_elements, rand_poly, FE};
+use super::utils::{rand_complex_mersenne_poly, rand_field_elements, rand_poly, FE};
 use const_random::const_random;
 use core::hint::black_box;
 use criterion::Criterion;
-use lambdaworks_math::polynomial::Polynomial;
+use lambdaworks_math::{
+    field::fields::mersenne31::extensions::Degree2ExtensionField, polynomial::Polynomial,
+};
 
 pub fn polynomial_benchmarks(c: &mut Criterion) {
     let mut group = c.benchmark_group("Polynomial");
@@ -42,6 +44,25 @@ pub fn polynomial_benchmarks(c: &mut Criterion) {
         let y_poly = rand_poly(order);
         bench.iter(|| black_box(&x_poly) * black_box(&y_poly));
     });
+    group.bench_function("fast mul", |bench| {
+        let x_poly = rand_complex_mersenne_poly(order as u32);
+        let y_poly = rand_complex_mersenne_poly(order as u32);
+        bench.iter(|| black_box(&x_poly) * black_box(&y_poly));
+    });
+
+    let big_order = 9;
+    let x_poly = rand_complex_mersenne_poly(big_order);
+    let y_poly = rand_complex_mersenne_poly(big_order);
+    group.bench_function("fast_mul big poly", |bench| {
+        bench.iter(|| {
+            black_box(&x_poly)
+                .fast_multiplication::<Degree2ExtensionField>(black_box(&y_poly))
+                .unwrap()
+        });
+    });
+    group.bench_function("slow mul big poly", |bench| {
+        bench.iter(|| black_box(&x_poly) * black_box(&y_poly));
+    });
 
     group.bench_function("div", |bench| {
         let x_poly = rand_poly(order);

crates/math/benches/polynomials/utils.rs

@@ -1,6 +1,12 @@
 use const_random::const_random;
 use lambdaworks_math::{
-    field::fields::u64_prime_field::{U64FieldElement, U64PrimeField},
+    field::{
+        element::FieldElement,
+        fields::{
+            mersenne31::{extensions::Degree2ExtensionField, field::Mersenne31Field},
+            u64_prime_field::{U64FieldElement, U64PrimeField},
+        },
+    },
     polynomial::{
         dense_multilinear_poly::DenseMultilinearPolynomial,
         sparse_multilinear_poly::SparseMultilinearPolynomial, Polynomial,
@@ -36,6 +42,28 @@ pub fn rand_field_elements(order: u64) -> Vec<FE> {
 pub fn rand_poly(order: u64) -> Polynomial<FE> {
     Polynomial::new(&rand_field_elements(order))
 }
+#[allow(dead_code)]
+#[inline(never)]
+#[export_name = "u64_utils::rand_complex_mersenne_field_elements"]
+pub fn rand_complex_mersenne_field_elements(
+    order: u32,
+) -> Vec<FieldElement<Degree2ExtensionField>> {
+    let mut result = Vec::with_capacity(1 << order);
+    for _ in 0..result.capacity() {
+        result.push(FieldElement::<Degree2ExtensionField>::new([
+            FieldElement::<Mersenne31Field>::new(random()),
+            FieldElement::<Mersenne31Field>::new(random()),
+        ]));
+    }
+    result
+}
+
+#[allow(dead_code)]
+#[inline(never)]
+#[export_name = "u64_utils::rand_complex_mersenne_poly"]
+pub fn rand_complex_mersenne_poly(order: u32) -> Polynomial<FieldElement<Degree2ExtensionField>> {
+    Polynomial::new(&rand_complex_mersenne_field_elements(order))
+}
 
 #[allow(dead_code)]
 #[inline(never)]