feat(generated-roles): add generated Jolt role crates

Author: markosg04

Cargo.lock

@@ -37,6 +37,9 @@ members = [
   "crates/jolt-transcript",
   "crates/jolt-witness",
   "crates/bolt",
+  "crates/jolt-kernels",
+  "crates/jolt-prover",
+  "crates/jolt-verifier",
   "crates/jolt-profiling",
   "crates/jolt-field",
   "jolt-core",
@@ -379,6 +382,8 @@ common = { path = "./common", default-features = false }
 tracer = { path = "./tracer", default-features = false }
 jolt-core = { path = "./jolt-core", default-features = false }
 jolt-crypto = { path = "./crates/jolt-crypto" }
+jolt-dory = { path = "./crates/jolt-dory" }
+jolt-hyperkzg = { path = "./crates/jolt-hyperkzg" }
 jolt-field = { path = "./crates/jolt-field" }
 jolt-openings = { path = "./crates/jolt-openings" }
 jolt-poly = { path = "./crates/jolt-poly" }
@@ -387,6 +392,9 @@ jolt-sumcheck = { path = "./crates/jolt-sumcheck" }
 jolt-r1cs = { path = "./crates/jolt-r1cs" }
 jolt-witness = { path = "./crates/jolt-witness" }
 bolt = { path = "./crates/bolt" }
+jolt-kernels = { path = "./crates/jolt-kernels" }
+jolt-prover = { path = "./crates/jolt-prover" }
+jolt-verifier = { path = "./crates/jolt-verifier" }
 jolt-riscv = { path = "./crates/jolt-riscv", default-features = false }
 jolt-program = { path = "./crates/jolt-program", default-features = false }
 jolt-lookup-tables = { path = "./crates/jolt-lookup-tables" }

Cargo.toml

@@ -426,9 +426,18 @@ fn generated_crate(
         path: format!("src/stages/{}.rs", artifact.stage.module_name()),
         source: artifact.source.source,
     }));
+    for file in &mut files {
+        file.source = normalized_generated_source(&file.source);
+    }
     GeneratedCrate { crate_name, files }
 }
 
+fn normalized_generated_source(source: &str) -> String {
+    let mut normalized = source.trim_end_matches(char::is_whitespace).to_owned();
+    normalized.push('\n');
+    normalized
+}
+
 #[derive(Clone, Copy, Debug, PartialEq, Eq)]
 enum ManifestMode<'a> {
     Standalone { dependency_root: &'a str },

crates/bolt/src/emit/rust/artifacts.rs

@@ -322,8 +322,8 @@ pub use verifier::{
     verify_jolt_through_stage6_with_programs, verify_jolt_through_stage7,
     verify_jolt_through_stage7_with_programs, verify_jolt_with_programs, JoltEvaluationProof,
     JoltEvaluationProofError, JoltNamedEval, JoltProof, JoltStage2RamAccess, JoltStage2RamData,
-    JoltStage2RamOutputLayout, JoltStageChallengeVector, JoltStageExecutionArtifacts,
-    JoltStage6BytecodeEntry, JoltStage6BytecodeReadRafData, JoltStage6VerifierData,
+    JoltStage2RamOutputLayout, JoltStage6BytecodeEntry, JoltStage6BytecodeReadRafData,
+    JoltStage6VerifierData, JoltStageChallengeVector, JoltStageExecutionArtifacts,
     JoltStageOpeningInputValue, JoltStageProof, JoltSumcheckOutput, JoltVerificationArtifacts,
     JoltVerifierInputs, JoltVerifierPrograms, JoltVerifierTarget, JoltVerifyError,
 };"
@@ -337,7 +337,7 @@ fn jolt_evaluation_role_api_extension() -> ProtocolArtifactExtension {
         required_artifact_stages: vec!["stage8".to_owned()],
         prover: ProtocolProverApiExtension {
             lib_module: jolt_prover_lib_module(),
-            imports: "#![expect(\n    clippy::too_many_arguments,\n    reason = \"generated prover helpers mirror staged protocol ABIs\"\n)]\n\nuse jolt_dory::{DoryCommitment, DoryHint, DoryProverSetup, DoryScheme};\nuse jolt_field::{Field, Fr};\nuse jolt_kernels::{stage1, stage2, stage3, stage4, stage5, stage6, stage7};\nuse jolt_openings::{AdditivelyHomomorphic, CommitmentScheme};\nuse jolt_poly::{EqPolynomial, Polynomial};\nuse jolt_transcript::{AppendToTranscript, Blake2bTranscript, LabelWithCount, Transcript};\nuse jolt_verifier::{JoltEvaluationProof, JoltNamedEval, JoltProof, JoltStage2RamAccess, JoltStage2RamData, JoltStage2RamOutputLayout, JoltStage6BytecodeEntry, JoltStage6BytecodeReadRafData, JoltStage6VerifierData, JoltStageChallengeVector, JoltStageExecutionArtifacts, JoltStageOpeningInputValue, JoltStageProof, JoltSumcheckOutput};\nuse jolt_witness::{stage4_ram_val_init_opening, CycleInput, Stage45SparseTraceWitness, Stage6BytecodeEntry as WitnessStage6BytecodeEntry, Stage6WitnessParams, Stage6WitnessPolynomials, Stage6WitnessSlices};\nuse rayon::prelude::*;\n\n".to_owned(),
+            imports: "#![expect(\n    clippy::too_many_arguments,\n    reason = \"generated prover helpers mirror staged protocol ABIs\"\n)]\n\nuse jolt_dory::{DoryCommitment, DoryHint, DoryProverSetup, DoryScheme};\nuse jolt_field::Fr;\nuse jolt_kernels::{stage1, stage2, stage3, stage4, stage5, stage6, stage7};\nuse jolt_openings::{AdditivelyHomomorphic, CommitmentScheme};\nuse jolt_poly::{EqPolynomial, Polynomial};\nuse jolt_transcript::{AppendToTranscript, Blake2bTranscript, LabelWithCount, Transcript};\nuse jolt_verifier::{JoltEvaluationProof, JoltNamedEval, JoltProof, JoltStage2RamAccess, JoltStage2RamData, JoltStage2RamOutputLayout, JoltStage6BytecodeEntry, JoltStage6BytecodeReadRafData, JoltStage6VerifierData, JoltStageChallengeVector, JoltStageExecutionArtifacts, JoltStageOpeningInputValue, JoltStageProof, JoltSumcheckOutput};\nuse jolt_witness::{stage4_ram_val_init_opening, CycleInput, Stage45SparseTraceWitness, Stage6BytecodeEntry as WitnessStage6BytecodeEntry, Stage6WitnessParams, Stage6WitnessPolynomials, Stage6WitnessSlices};\nuse rayon::prelude::*;\n\n".to_owned(),
             input_fields:
                 "    pub stage7_openings: Option<&'a [stage7::Stage7OpeningInputValue<Fr>]>,\n"
                     .to_owned(),
@@ -357,7 +357,7 @@ fn jolt_evaluation_role_api_extension() -> ProtocolArtifactExtension {
         },
         verifier: ProtocolVerifierApiExtension {
             lib_module: jolt_verifier_lib_module(),
-            imports: "use std::collections::BTreeMap;\n\nuse jolt_dory::{DoryCommitment, DoryProof, DoryScheme, DoryVerifierSetup};\nuse jolt_field::{Field, Fr};\nuse jolt_openings::{AdditivelyHomomorphic, CommitmentScheme, OpeningsError};\nuse jolt_poly::EqPolynomial;\nuse jolt_transcript::{AppendToTranscript, LabelWithCount, Transcript};\n".to_owned(),
+            imports: "use std::collections::BTreeMap;\n\nuse jolt_dory::{DoryCommitment, DoryProof, DoryScheme, DoryVerifierSetup};\nuse jolt_field::Fr;\nuse jolt_openings::{AdditivelyHomomorphic, CommitmentScheme, OpeningsError};\nuse jolt_poly::EqPolynomial;\nuse jolt_transcript::{AppendToTranscript, LabelWithCount, Transcript};\n".to_owned(),
             proof_fields: "    pub evaluation: Option<JoltEvaluationProof>,\n".to_owned(),
             proof_items: "pub type JoltStage2RamAccess = crate::stages::stage2::Stage2RamAccess;\npub type JoltStage2RamOutputLayout = crate::stages::stage2::Stage2RamOutputLayout;\npub type JoltStage2RamData<'a> = crate::stages::stage2::Stage2RamData<'a>;\npub type JoltStageChallengeVector = crate::stages::common::StageChallengeVector<Fr>;\npub type JoltStageExecutionArtifacts = crate::stages::common::StageExecutionArtifacts<Fr>;\npub type JoltStageOpeningInputValue = crate::stages::common::StageOpeningInputValue<Fr>;\n\n#[derive(Clone, Debug)]\npub struct JoltEvaluationProof {\n    pub joint_opening_proof: DoryProof,\n}\n\n".to_owned(),
             inputs_derive: Some("#[derive(Clone, Copy)]".to_owned()),

crates/bolt/src/protocols/jolt/artifacts.rs

@@ -256,7 +256,7 @@ impl CommitmentCpuProgram {
                 "use std::borrow::Cow;\n\
                  \n\
                  use jolt_dory::{DoryCommitment, DoryHint, DoryProverSetup, DoryScheme};\n\
-                 use jolt_field::{Field, Fr};\n\
+                 use jolt_field::Fr;\n\
                  use jolt_openings::CommitmentScheme as _;\n\
                  use jolt_poly::{EqPolynomial, MultilinearPoly};\n\
                  use jolt_transcript::{AppendToTranscript, Blake2bTranscript, LabelWithCount, Transcript};\n\
@@ -540,13 +540,13 @@ impl MultilinearPoly<Fr> for AddressMajorOneHotPolynomial {
         result
     }
 
-    fn is_sparse(&self) -> bool {
+    fn is_one_hot(&self) -> bool {
         true
     }
 
-    fn for_each_nonzero(&self, f: &mut dyn FnMut(usize, Fr)) {
+    fn for_each_one(&self, f: &mut dyn FnMut(usize)) {
         for flat in self.nonzero_flat_indices() {
-            f(flat, Fr::from_u64(1));
+            f(flat);
         }
     }
 }
@@ -908,6 +908,14 @@ impl Default for OneHotChunkCounts {
                 }
             }
         }
+        let input_binding = if initializers
+            .iter()
+            .any(|initializer| initializer.contains("inputs."))
+        {
+            "inputs"
+        } else {
+            "_inputs"
+        };
         let fields = fields.join("\n");
         let provider_arms = provider_arms.join("\n");
         let initializers = initializers.join("\n");
@@ -930,7 +938,7 @@ impl CommitmentInputProvider for CommitmentOracles {{
 }}
 
 pub fn build_commitment_oracles(
-    inputs: &CommitmentOracleInputs<'_>,
+    {input_binding}: &CommitmentOracleInputs<'_>,
 ) -> Result<CommitmentOracles, CommitmentPhaseError> {{
     Ok(CommitmentOracles {{
 {initializers}

crates/bolt/src/protocols/jolt/emit/rust/commitment.rs

@@ -1067,7 +1067,7 @@ impl Stage1CpuProgram {
 
     fn emit_verifier_imports() -> &'static str {
         "use super::common::append_labeled_scalar;\n\
-         use jolt_field::{Field, Fr};\n\
+         use jolt_field::Fr;\n\
          use jolt_sumcheck::{CompressedLabeledRoundPoly, LabeledRoundPoly, SumcheckClaim, SumcheckError, SumcheckVerifier};\n\
          use jolt_transcript::{Blake2bTranscript, Transcript};"
     }

crates/bolt/src/protocols/jolt/emit/rust/stage1.rs

@@ -927,7 +927,7 @@ impl Stage2CpuProgram {
 
     fn emit_verifier_imports() -> &'static str {
         "use super::common::{append_labeled_scalar, batch_claims, eval_by_name, find_batch, find_plan, pow_field, require_operand_count, reverse_slice, single_operand};\n\
-         use jolt_field::{Field, Fr};\n\
+         use jolt_field::{Field, Fr, MulPow2, MulPrimitiveInt, RingCore};\n\
          use jolt_poly::lagrange::{lagrange_evals, lagrange_kernel_eval};\n\
          use jolt_poly::{EqPolynomial, UnivariatePoly};\n\
          use jolt_sumcheck::{CompressedLabeledRoundPoly, SumcheckClaim, SumcheckError, SumcheckVerifier};\n\

crates/bolt/src/protocols/jolt/emit/rust/stage2.rs

@@ -964,7 +964,7 @@ impl Stage5CpuProgram {
 
     fn emit_verifier_imports() -> &'static str {
         "use super::common::{batch_claims, eval_by_name, find_batch, find_plan, identity_polynomial_eval, indexed_evals_by_prefix, indexed_evals_by_prefix_any, lt_polynomial_eval, normalize_instruction_read_raf_point, operand_polynomial_eval, reverse_slice, suffix_point};\n\
-         use jolt_field::{Field, Fr};\n\
+         use jolt_field::{Field, Fr, RingCore};\n\
          use jolt_lookup_tables::LookupTableKind;\n\
          use jolt_poly::EqPolynomial;\n\
          use jolt_sumcheck::SumcheckError;\n\

crates/bolt/src/protocols/jolt/emit/rust/stage5.rs

@@ -985,7 +985,7 @@ impl Stage7CpuProgram {
 
     fn emit_verifier_imports() -> &'static str {
         "use super::common::{batch_claims, eval_by_name, find_batch, find_plan, normalize_bytecode_read_raf_point, normalize_instruction_read_raf_point, reverse_slice};\n\
-         use jolt_field::{Field, Fr};\n\
+         use jolt_field::{Field, Fr, RingCore};\n\
          use jolt_poly::EqPolynomial;\n\
          use jolt_sumcheck::SumcheckError;\n\
          use jolt_transcript::{Blake2bTranscript, LabelWithCount, Transcript};"

crates/bolt/src/protocols/jolt/emit/rust/stage7.rs

@@ -61,7 +61,7 @@ impl JoltProtocolParams {
             lookups_ra_virtual_log_k_chunk,
             instruction_log_k,
             register_log_k: 7,
-            lookup_table_count: 40,
+            lookup_table_count: 41,
             instruction_d,
             instruction_ra_virtual_d,
             bytecode_d,