multilinear_batching_flavor.hpp

Go to the documentation of this file.

// === AUDIT STATUS ===
// internal:    { status: Planned, auditors: [], commit: }
// external_1:  { status: not started, auditors: [], commit: }
// external_2:  { status: not started, auditors: [], commit: }
// =====================
 
#pragma once
#include <utility>
 
#include "barretenberg/commitment_schemes/kzg/kzg.hpp"
#include "barretenberg/common/ref_vector.hpp"
#include "barretenberg/flavor/flavor.hpp"
#include "barretenberg/flavor/flavor_macros.hpp"
#include "barretenberg/flavor/relation_definitions.hpp"
#include "barretenberg/relations/multilinear_batching/multilinear_batching_relation.hpp"
#include "barretenberg/transcript/transcript.hpp"
 
namespace bb {
 
class MultilinearBatchingFlavor {
  public:
    using Curve = curve::BN254;
    using FF = Curve::ScalarField;
    using GroupElement = Curve::Element;
    using Commitment = Curve::AffineElement;
    using PCS = KZG<Curve>;
    using Polynomial = bb::Polynomial<FF>;
    using CommitmentKey = bb::CommitmentKey<Curve>;
    using VerifierCommitmentKey = bb::VerifierCommitmentKey<Curve>;
    using Transcript = NativeTranscript;
 
    // An upper bound on the size of the MultilinearBatching-circuits. `CONST_FOLDING_LOG_N` bounds the log circuit
    // sizes in the Chonk context.
    static constexpr size_t VIRTUAL_LOG_N = CONST_FOLDING_LOG_N;
    static constexpr bool USE_SHORT_MONOMIALS = false;
    // Indicates that this flavor runs with non-ZK Sumcheck.
    static constexpr bool HasZK = false;
    // Indicates that this flavor runs with Multilinear Batching.
    static constexpr bool IS_MULTILINEAR_BATCHING = true;
    // To achieve fixed proof size and that the recursive verifier circuit is constant, we are using padding in Sumcheck
    // and Shplemini
    static constexpr bool USE_PADDING = true;
    static constexpr size_t NUM_WIRES = 4;
    // The number of multivariate polynomials on which a sumcheck prover sumcheck operates (including shifts). We often
    // need containers of this size to hold related data, so we choose a name more agnostic than `NUM_POLYNOMIALS`.
    static constexpr size_t NUM_ALL_ENTITIES = 6;
    // The total number of witness entities not including shifts.
    static constexpr size_t NUM_WITNESS_ENTITIES = 4;
    // The number of shifted witness entities including derived witness entities
    static constexpr size_t NUM_SHIFTED_ENTITIES = 2;
 
    // define the tuple of Relations that comprise the Sumcheck relation
    // Note: made generic for use in MegaRecursive.
    template <typename FF>
    using Relations_ =
        std::tuple<bb::MultilinearBatchingAccumulatorRelation<FF>, bb::MultilinearBatchingInstanceRelation<FF>>;
    using Relations = Relations_<FF>;
 
    static constexpr size_t MAX_PARTIAL_RELATION_LENGTH = compute_max_partial_relation_length<Relations>();
    // BATCHED_RELATION_PARTIAL_LENGTH = algebraic degree of sumcheck relation *after* multiplying by the `pow_zeta`
    // random polynomial e.g. For \sum(x) [A(x) * B(x) + C(x)] * PowZeta(X), relation length = 2 and random relation
    // length = 3
    static constexpr size_t BATCHED_RELATION_PARTIAL_LENGTH = MAX_PARTIAL_RELATION_LENGTH + 1;
    static constexpr size_t NUM_RELATIONS = std::tuple_size_v<Relations>;
 
    // A challenge whose powers are used to batch subrelation contributions during Sumcheck
    static constexpr size_t NUM_SUBRELATIONS = compute_number_of_subrelations<Relations>();
    using SubrelationSeparator = FF;
 
    // Whether or not the first row of the execution trace is reserved for 0s to enable shifts
    static constexpr bool has_zero_row = false;
 
    static constexpr size_t num_frs_comm = FrCodec::calc_num_fields<Commitment>();
    static constexpr size_t num_frs_fr = FrCodec::calc_num_fields<FF>();
 
    static constexpr size_t PROOF_LENGTH_WITHOUT_PUB_INPUTS()
    {
        return /*accumulator commitments*/ (NUM_WITNESS_ENTITIES / 2 * num_frs_comm) +
               /*multivariate challenges*/ (VIRTUAL_LOG_N * num_frs_fr) +
               /*witness evaluations*/ (NUM_WITNESS_ENTITIES / 2 * num_frs_fr) +
               /*sumcheck univariates*/ (VIRTUAL_LOG_N * BATCHED_RELATION_PARTIAL_LENGTH * num_frs_fr) +
               /*sumcheck evaluations*/ (NUM_ALL_ENTITIES * num_frs_fr);
    }
 
    // WireEntities for basic witness entities
    template <typename DataType> class WireEntities {
      public:
        DEFINE_FLAVOR_MEMBERS(DataType,
                              batched_unshifted_accumulator, // column 0: batched unshifted poly for accumulator
                              batched_unshifted_instance,    // column 1: batched unshifted poly for instance
                              eq_accumulator,                // column 2: eq(u, r_acc) - selects accumulator eval point
                              eq_instance);                  // column 3: eq(u, r_inst) - selects instance eval point
    };
 
    template <typename DataType> class WitnessEntities : public WireEntities<DataType> {
      public:
        DEFINE_FLAVOR_MEMBERS(DataType,
                              batched_unshifted_accumulator, // column 0: batched unshifted poly for accumulator
                              batched_unshifted_instance,    // column 1: batched unshifted poly for instance
                              eq_accumulator,                // column 2: eq(u, r_acc) - selects accumulator eval point
                              eq_instance);                  // column 3: eq(u, r_inst) - selects instance eval point
    };
 
    template <typename DataType> class ShiftedEntities {
      public:
        DEFINE_FLAVOR_MEMBERS(DataType,
                              batched_shifted_accumulator, // column 0: batched shifted poly for accumulator
                              batched_shifted_instance     // column 1: batched shifted poly for instance
        );
        auto get_shifted() { return RefArray{ batched_shifted_accumulator, batched_shifted_instance }; };
    };
 
    template <typename DataType>
    class AllEntities : public WitnessEntities<DataType>, public ShiftedEntities<DataType> {
      public:
        DEFINE_COMPOUND_GET_ALL(WitnessEntities<DataType>, ShiftedEntities<DataType>)
 
        auto get_unshifted() { return WitnessEntities<DataType>::get_all(); };
        auto get_witness() { return WitnessEntities<DataType>::get_all(); };
        auto get_witness() const { return WitnessEntities<DataType>::get_all(); };
        auto get_shifted() { return ShiftedEntities<DataType>::get_all(); };
    };
 
    class AllValues : public AllEntities<FF> {
      public:
        using Base = AllEntities<FF>;
        using Base::Base;
    };
 
    class ProverPolynomials : public AllEntities<Polynomial> {
      public:
        // Define all operations as default, except copy construction/assignment
        ProverPolynomials() = default;
        // fully-formed constructor
        ProverPolynomials(size_t circuit_size)
        {
            BB_BENCH_NAME("ProverPolynomials(size_t)");
 
            // catch-all with fully formed polynomials
            for (auto& poly : get_unshifted()) {
                if (poly.is_empty()) {
                    // Not set above
                    poly = Polynomial{ /*memory size*/ circuit_size, /*largest possible index*/ 1 << VIRTUAL_LOG_N };
                }
            }
 
            for (auto& poly : get_shifted()) {
                poly = Polynomial{ /*memory size*/ circuit_size, /*largest possible index*/ 1 << VIRTUAL_LOG_N };
            }
        }
        [[nodiscard]] size_t get_polynomial_size() const { return batched_unshifted_accumulator.size(); }
        void increase_polynomials_virtual_size(const size_t size_in)
        {
            for (auto& polynomial : this->get_all()) {
                polynomial.increase_virtual_size(size_in);
            }
        }
    };
 
    class ProvingKey {
      public:
        ProverPolynomials polynomials; // storage for all polynomials evaluated by the prover
        std::vector<FF> accumulator_challenge;
        std::vector<FF> instance_challenge;
        std::vector<FF> accumulator_evaluations;
        std::vector<FF> instance_evaluations;
        size_t circuit_size;
        ProvingKey(ProverPolynomials&& polynomials,
                   std::vector<FF> accumulator_challenge,
                   std::vector<FF> instance_challenge,
                   std::vector<FF> accumulator_evaluations,
                   std::vector<FF> instance_evaluations)
            : polynomials(std::move(polynomials))
            , accumulator_challenge(std::move(accumulator_challenge))
            , instance_challenge(std::move(instance_challenge))
            , accumulator_evaluations(std::move(accumulator_evaluations))
            , instance_evaluations(std::move(instance_evaluations))
            , circuit_size(polynomials.get_polynomial_size())
        {}
    };
 
    class PartiallyEvaluatedMultivariates : public AllEntities<Polynomial> {
 
      public:
        PartiallyEvaluatedMultivariates() = default;
        PartiallyEvaluatedMultivariates(const ProverPolynomials& full_polynomials, size_t circuit_size)
        {
            for (auto [poly, full_poly] : zip_view(get_all(), full_polynomials.get_all())) {
                // After the initial sumcheck round, the new size is CEIL(size/2).
                size_t desired_size = full_poly.end_index() / 2 + full_poly.end_index() % 2;
                poly = Polynomial(desired_size, circuit_size / 2);
            }
        }
    };
 
    template <size_t LENGTH> using ProverUnivariates = AllEntities<bb::Univariate<FF, LENGTH>>;
 
    using ExtendedEdges = ProverUnivariates<MAX_PARTIAL_RELATION_LENGTH>;
 
    using WitnessCommitments = WitnessEntities<Commitment>;
 
    class CommitmentLabels : public AllEntities<std::string> {
      public:
        CommitmentLabels()
        {
            batched_unshifted_accumulator = "BATCHED_UNSHIFTED_ACCUMULATOR";
            batched_unshifted_instance = "BATCHED_UNSHIFTED_INSTANCE";
            eq_accumulator = "EQ_ACCUMULATOR";
            eq_instance = "EQ_INSTANCE";
            batched_shifted_accumulator = "BATCHED_SHIFTED_ACCUMULATOR";
            batched_shifted_instance = "BATCHED_SHIFTED_INSTANCE";
        };
    };
};
 
} // namespace bb