ultra_execution_trace.hpp

Go to the documentation of this file.

// === AUDIT STATUS ===
// internal:    { status: Complete, auditors: [Luke, Raju], commit: }
// external_1:  { status: not started, auditors: [], commit: }
// external_2:  { status: not started, auditors: [], commit: }
// =====================
 
#pragma once
 
#include "barretenberg/common/ref_vector.hpp"
#include "barretenberg/common/throw_or_abort.hpp"
#include "barretenberg/constants.hpp"
#include "barretenberg/ecc/curves/bn254/fr.hpp"
#include "barretenberg/honk/execution_trace/execution_trace_block.hpp"
#include "barretenberg/numeric/bitop/get_msb.hpp"
 
namespace bb {
 
class UltraTraceBlock : public ExecutionTraceBlock<fr, 4> {
  public:
    virtual Selector<fr>& q_lookup() { return zero_selectors[0]; };
    virtual Selector<fr>& q_arith() { return zero_selectors[1]; }
    virtual Selector<fr>& q_delta_range() { return zero_selectors[2]; }
    virtual Selector<fr>& q_elliptic() { return zero_selectors[3]; }
    virtual Selector<fr>& q_memory() { return zero_selectors[4]; }
    virtual Selector<fr>& q_nnf() { return zero_selectors[5]; }
    virtual Selector<fr>& q_poseidon2_external() { return zero_selectors[6]; }
    virtual Selector<fr>& q_poseidon2_internal() { return zero_selectors[7]; }
 
    RefVector<Selector<fr>> get_selectors() override
    {
        return RefVector{ q_m(),
                          q_c(),
                          q_1(),
                          q_2(),
                          q_3(),
                          q_4(),
                          q_lookup(),
                          q_arith(),
                          q_delta_range(),
                          q_elliptic(),
                          q_memory(),
                          q_nnf(),
                          q_poseidon2_external(),
                          q_poseidon2_internal() };
    }
 
    virtual void set_gate_selector([[maybe_unused]] const fr& value) {}
 
  private:
    std::array<ZeroSelector<fr>, 8> zero_selectors;
};
 
class UltraTracePublicInputBlock : public UltraTraceBlock {};
 
class UltraTraceLookupBlock : public UltraTraceBlock {
  public:
    SelectorType& q_lookup() override { return gate_selector; }
 
    void set_gate_selector(const fr& value) override
    {
        gate_selector.emplace_back(value);
        q_arith().emplace_back(0);
        q_delta_range().emplace_back(0);
        q_elliptic().emplace_back(0);
        q_memory().emplace_back(0);
        q_nnf().emplace_back(0);
        q_poseidon2_external().emplace_back(0);
        q_poseidon2_internal().emplace_back(0);
    }
 
  private:
    SlabVectorSelector<fr> gate_selector;
};
 
class UltraTraceArithmeticBlock : public UltraTraceBlock {
  public:
    SelectorType& q_arith() override { return gate_selector; }
 
    void set_gate_selector(const fr& value) override
    {
        q_lookup().emplace_back(0);
        gate_selector.emplace_back(value);
        q_delta_range().emplace_back(0);
        q_elliptic().emplace_back(0);
        q_memory().emplace_back(0);
        q_nnf().emplace_back(0);
        q_poseidon2_external().emplace_back(0);
        q_poseidon2_internal().emplace_back(0);
    }
 
  private:
    SlabVectorSelector<fr> gate_selector;
};
 
class UltraTraceDeltaRangeBlock : public UltraTraceBlock {
  public:
    SelectorType& q_delta_range() override { return gate_selector; }
 
    void set_gate_selector(const fr& value) override
    {
        q_lookup().emplace_back(0);
        q_arith().emplace_back(0);
        gate_selector.emplace_back(value);
        q_elliptic().emplace_back(0);
        q_memory().emplace_back(0);
        q_nnf().emplace_back(0);
        q_poseidon2_external().emplace_back(0);
        q_poseidon2_internal().emplace_back(0);
    }
 
  private:
    SlabVectorSelector<fr> gate_selector;
};
 
class UltraTraceEllipticBlock : public UltraTraceBlock {
  public:
    SelectorType& q_elliptic() override { return gate_selector; }
 
    void set_gate_selector(const fr& value) override
    {
        q_lookup().emplace_back(0);
        q_arith().emplace_back(0);
        q_delta_range().emplace_back(0);
        gate_selector.emplace_back(value);
        q_memory().emplace_back(0);
        q_nnf().emplace_back(0);
        q_poseidon2_external().emplace_back(0);
        q_poseidon2_internal().emplace_back(0);
    }
 
  private:
    SlabVectorSelector<fr> gate_selector;
};
 
class UltraTraceMemoryBlock : public UltraTraceBlock {
  public:
    SelectorType& q_memory() override { return gate_selector; }
 
    void set_gate_selector(const fr& value) override
    {
        q_lookup().emplace_back(0);
        q_arith().emplace_back(0);
        q_delta_range().emplace_back(0);
        q_elliptic().emplace_back(0);
        gate_selector.emplace_back(value);
        q_nnf().emplace_back(0);
        q_poseidon2_external().emplace_back(0);
        q_poseidon2_internal().emplace_back(0);
    }
 
  private:
    SlabVectorSelector<fr> gate_selector;
};
 
class UltraTraceNonNativeFieldBlock : public UltraTraceBlock {
  public:
    SelectorType& q_nnf() override { return gate_selector; }
 
    void set_gate_selector(const fr& value) override
    {
        q_lookup().emplace_back(0);
        q_arith().emplace_back(0);
        q_delta_range().emplace_back(0);
        q_elliptic().emplace_back(0);
        q_memory().emplace_back(0);
        gate_selector.emplace_back(value);
        q_poseidon2_external().emplace_back(0);
        q_poseidon2_internal().emplace_back(0);
    }
 
  private:
    SlabVectorSelector<fr> gate_selector;
};
 
class UltraTracePoseidon2ExternalBlock : public UltraTraceBlock {
  public:
    SelectorType& q_poseidon2_external() override { return gate_selector; }
 
    void set_gate_selector(const fr& value) override
    {
        q_lookup().emplace_back(0);
        q_arith().emplace_back(0);
        q_delta_range().emplace_back(0);
        q_elliptic().emplace_back(0);
        q_memory().emplace_back(0);
        q_nnf().emplace_back(0);
        gate_selector.emplace_back(value);
        q_poseidon2_internal().emplace_back(0);
    }
 
  private:
    SlabVectorSelector<fr> gate_selector;
};
 
class UltraTracePoseidon2InternalBlock : public UltraTraceBlock {
  public:
    SelectorType& q_poseidon2_internal() override { return gate_selector; }
 
    void set_gate_selector(const fr& value) override
    {
        q_lookup().emplace_back(0);
        q_arith().emplace_back(0);
        q_delta_range().emplace_back(0);
        q_elliptic().emplace_back(0);
        q_memory().emplace_back(0);
        q_nnf().emplace_back(0);
        q_poseidon2_external().emplace_back(0);
        gate_selector.emplace_back(value);
    }
 
  private:
    SlabVectorSelector<fr> gate_selector;
};
 
struct UltraTraceBlockData {
    UltraTracePublicInputBlock pub_inputs; // Has to be the first block
    UltraTraceLookupBlock lookup;
    UltraTraceArithmeticBlock arithmetic;
    UltraTraceDeltaRangeBlock delta_range;
    UltraTraceEllipticBlock elliptic;
    UltraTraceMemoryBlock memory;
    UltraTraceNonNativeFieldBlock nnf;
    UltraTracePoseidon2ExternalBlock poseidon2_external;
    UltraTracePoseidon2InternalBlock poseidon2_internal;
 
    static constexpr size_t NUM_BLOCKS = 9;
 
    auto get()
    {
        return RefArray(std::array<UltraTraceBlock*, NUM_BLOCKS>{ &pub_inputs,
                                                                  &lookup,
                                                                  &arithmetic,
                                                                  &delta_range,
                                                                  &elliptic,
                                                                  &memory,
                                                                  &nnf,
                                                                  &poseidon2_external,
                                                                  &poseidon2_internal });
    }
 
    auto get() const
    {
        return RefArray(std::array<const UltraTraceBlock*, NUM_BLOCKS>{ &pub_inputs,
                                                                        &lookup,
                                                                        &arithmetic,
                                                                        &delta_range,
                                                                        &elliptic,
                                                                        &memory,
                                                                        &nnf,
                                                                        &poseidon2_external,
                                                                        &poseidon2_internal });
    }
 
    auto get_gate_blocks() const
    {
        return RefArray(std::array<const UltraTraceBlock*, 8>{
            &lookup,
            &arithmetic,
            &delta_range,
            &elliptic,
            &memory,
            &nnf,
            &poseidon2_external,
            &poseidon2_internal,
        });
    }
 
    bool operator==(const UltraTraceBlockData& other) const = default;
};
 
class UltraExecutionTraceBlocks : public UltraTraceBlockData {
 
  public:
    static constexpr size_t NUM_WIRES = UltraTraceBlock::NUM_WIRES;
    // The number of rows reserved at the top of the trace for row-disabling / ZK masking.
    static constexpr size_t TRACE_OFFSET = NUM_DISABLED_ROWS_IN_SUMCHECK;
    using FF = fr;
 
    UltraExecutionTraceBlocks() = default;
 
    void compute_offsets(size_t trace_offset = TRACE_OFFSET)
    {
        uint32_t offset = static_cast<uint32_t>(trace_offset + NUM_ZERO_ROWS);
        for (auto& block : this->get()) {
            block.trace_offset_ = offset;
            offset += static_cast<uint32_t>(block.size());
        }
    }
 
    void summarize() const
    {
        info("Gate blocks summary:");
        info("pub inputs :\t", this->pub_inputs.size());
        info("lookups    :\t", this->lookup.size());
        info("arithmetic :\t", this->arithmetic.size());
        info("delta range:\t", this->delta_range.size());
        info("elliptic   :\t", this->elliptic.size());
        info("memory     :\t", this->memory.size());
        info("nnf        :\t", this->nnf.size());
        info("poseidon ext  :\t", this->poseidon2_external.size());
        info("poseidon int  :\t", this->poseidon2_internal.size());
    }
 
    // Get cumulative size of all blocks
    size_t get_total_content_size()
    {
        size_t total_size(0);
        for (const auto& block : this->get()) {
            total_size += block.size();
        }
        return total_size;
    }
 
    bool operator==(const UltraExecutionTraceBlocks& other) const = default;
};
 
} // namespace bb