Revert "Delete pairing benchmark"

This reverts commit 18c1146.

Author: Avaneesh-axiom

benchmarks/guest/pairing/Cargo.toml

@@ -0,0 +1,27 @@
+[workspace]
+[package]
+name = "openvm-pairing-program"
+version = "0.0.0"
+edition = "2021"
+
+[dependencies]
+openvm = { path = "../../../crates/toolchain/openvm", features = ["std"] }
+openvm-algebra-guest = { path = "../../../extensions/algebra/guest", default-features = false }
+openvm-ecc-guest = { path = "../../../extensions/ecc/guest", default-features = false }
+openvm-pairing-guest = { path = "../../../extensions/pairing/guest", default-features = false, features = [
+    "bn254",
+] }
+hex = { version = "0.4.3", default-features = false, features = ["alloc"] }
+
+[features]
+default = []
+halo2curves = ["openvm-pairing-guest/halo2curves"]
+
+[profile.release]
+panic = "abort"
+lto = "thin"    # faster compile time
+
+[profile.profiling]
+inherits = "release"
+debug = 2
+strip = false

benchmarks/guest/pairing/elf/openvm-pairing-program.elf

@@ -0,0 +1,25 @@
+[app_vm_config.rv32i]
+[app_vm_config.rv32m]
+[app_vm_config.io]
+[app_vm_config.modular]
+supported_modulus = [
+    # bn254 (alt bn128)
+    "21888242871839275222246405745257275088696311157297823662689037894645226208583", # coordinate field
+    "21888242871839275222246405745257275088548364400416034343698204186575808495617", # scalar field
+]
+
+[app_vm_config.fp2]
+supported_modulus = [
+    # bn254 (alt bn128)
+    "21888242871839275222246405745257275088696311157297823662689037894645226208583",
+]
+
+[app_vm_config.pairing]
+supported_curves = ["Bn254"]
+
+# bn254 (alt bn128)
+[[app_vm_config.ecc.supported_curves]]
+modulus = "21888242871839275222246405745257275088696311157297823662689037894645226208583"
+scalar = "21888242871839275222246405745257275088548364400416034343698204186575808495617"
+a = "0"
+b = "3"

benchmarks/guest/pairing/openvm.toml

@@ -0,0 +1,77 @@
+use openvm_algebra_guest::IntMod;
+use openvm_ecc_guest::AffinePoint;
+#[allow(unused_imports)]
+use openvm_pairing_guest::{
+    bn254::{Bn254, Bn254G1Affine, Fp, Fp2},
+    pairing::PairingCheck,
+};
+
+openvm_algebra_guest::moduli_macros::moduli_init! {
+    "0x30644e72e131a029b85045b68181585d97816a916871ca8d3c208c16d87cfd47", // Bn254Fp Coordinate field
+    "0x30644e72e131a029b85045b68181585d2833e84879b9709143e1f593f0000001", // Bn254 Scalar
+}
+
+openvm_ecc_guest::sw_macros::sw_init! {
+    Bn254G1Affine
+}
+
+openvm_algebra_guest::complex_macros::complex_init! {
+    Bn254Fp2 { mod_idx = 0 },
+}
+
+const PAIR_ELEMENT_LEN: usize = 32 * (2 + 4); // 1 G1Affine (2 Fp), 1 G2Affine (4 Fp)
+
+pub fn main() {
+    // copied from https://github.com/bluealloy/revm/blob/9e39df5dbc5fdc98779c644629b28b8bee75794a/crates/precompile/src/bn128.rs#L395
+    let input = hex::decode(
+        "\
+            1c76476f4def4bb94541d57ebba1193381ffa7aa76ada664dd31c16024c43f59\
+            3034dd2920f673e204fee2811c678745fc819b55d3e9d294e45c9b03a76aef41\
+            209dd15ebff5d46c4bd888e51a93cf99a7329636c63514396b4a452003a35bf7\
+            04bf11ca01483bfa8b34b43561848d28905960114c8ac04049af4b6315a41678\
+            2bb8324af6cfc93537a2ad1a445cfd0ca2a71acd7ac41fadbf933c2a51be344d\
+            120a2a4cf30c1bf9845f20c6fe39e07ea2cce61f0c9bb048165fe5e4de877550\
+            111e129f1cf1097710d41c4ac70fcdfa5ba2023c6ff1cbeac322de49d1b6df7c\
+            2032c61a830e3c17286de9462bf242fca2883585b93870a73853face6a6bf411\
+            198e9393920d483a7260bfb731fb5d25f1aa493335a9e71297e485b7aef312c2\
+            1800deef121f1e76426a00665e5c4479674322d4f75edadd46debd5cd992f6ed\
+            090689d0585ff075ec9e99ad690c3395bc4b313370b38ef355acdadcd122975b\
+            12c85ea5db8c6deb4aab71808dcb408fe3d1e7690c43d37b4ce6cc0166fa7daa",
+    )
+    .unwrap();
+
+    let elements = input.len() / PAIR_ELEMENT_LEN;
+
+    let mut p = Vec::with_capacity(elements);
+    let mut q = Vec::with_capacity(elements);
+
+    // read points
+    for idx in 0..elements {
+        // At each idx, there is (G1, G2) which is 6 Fp points
+        let read_fq_at = |n: usize| {
+            debug_assert!(n < PAIR_ELEMENT_LEN / 32);
+            let start = idx * PAIR_ELEMENT_LEN + n * 32;
+            // SAFETY: We're reading `6 * 32 == PAIR_ELEMENT_LEN` bytes from `input[idx..]`
+            // per iteration. This is guaranteed to be in-bounds.
+            let slice = unsafe { input.get_unchecked(start..start + 32) };
+            Fp::from_be_bytes(&slice[..32])
+        };
+        // https://eips.ethereum.org/EIPS/eip-197, Fp2 is encoded as (a, b) where a * i + b
+        let g1_x = read_fq_at(0);
+        let g1_y = read_fq_at(1);
+        let g2_x_c1 = read_fq_at(2);
+        let g2_x_c0 = read_fq_at(3);
+        let g2_y_c1 = read_fq_at(4);
+        let g2_y_c0 = read_fq_at(5);
+
+        let g1 = AffinePoint::new(g1_x, g1_y);
+        let g2_x = Fp2::new(g2_x_c0, g2_x_c1);
+        let g2_y = Fp2::new(g2_y_c0, g2_y_c1);
+        let g2 = AffinePoint::new(g2_x, g2_y);
+
+        p.push(g1);
+        q.push(g2);
+    }
+    let success = Bn254::pairing_check(&p, &q).is_ok();
+    assert!(success);
+}

benchmarks/guest/pairing/src/main.rs

@@ -0,0 +1,41 @@
+use clap::Parser;
+use eyre::Result;
+use openvm_algebra_circuit::{Fp2Extension, ModularExtension};
+use openvm_benchmarks_prove::util::BenchmarkCli;
+use openvm_circuit::arch::SystemConfig;
+use openvm_ecc_circuit::WeierstrassExtension;
+use openvm_pairing_circuit::{PairingCurve, PairingExtension};
+use openvm_pairing_guest::bn254::{BN254_COMPLEX_STRUCT_NAME, BN254_MODULUS, BN254_ORDER};
+use openvm_sdk::{config::SdkVmConfig, Sdk, StdIn};
+use openvm_stark_sdk::bench::run_with_metric_collection;
+
+fn main() -> Result<()> {
+    let args = BenchmarkCli::parse();
+
+    let elf = args.build_bench_program("pairing")?;
+    let vm_config = SdkVmConfig::builder()
+        .system(SystemConfig::default().with_continuations().into())
+        .rv32i(Default::default())
+        .rv32m(Default::default())
+        .io(Default::default())
+        .keccak(Default::default())
+        .modular(ModularExtension::new(vec![
+            BN254_MODULUS.clone(),
+            BN254_ORDER.clone(),
+        ]))
+        .fp2(Fp2Extension::new(vec![(
+            BN254_COMPLEX_STRUCT_NAME.to_string(),
+            BN254_MODULUS.clone(),
+        )]))
+        .ecc(WeierstrassExtension::new(vec![
+            PairingCurve::Bn254.curve_config()
+        ]))
+        .pairing(PairingExtension::new(vec![PairingCurve::Bn254]))
+        .build();
+    let sdk = Sdk::new();
+    let exe = sdk.transpile(elf, vm_config.transpiler()).unwrap();
+
+    run_with_metric_collection("OUTPUT_PATH", || -> Result<()> {
+        args.bench_from_exe("pairing", vm_config, exe, StdIn::default())
+    })
+}