finalize_mau_pred_deps_power.h

 
#ifndef BF_P4C_MAU_FINALIZE_MAU_PRED_DEPS_POWER_H_
#define BF_P4C_MAU_FINALIZE_MAU_PRED_DEPS_POWER_H_
 
#include <map>
#include <ostream>
 
#include "backends/tofino/bf-p4c/bf-p4c-options.h"
#include "backends/tofino/bf-p4c/ir/gress.h"
#include "backends/tofino/bf-p4c/mau/build_power_graph.h"
#include "backends/tofino/bf-p4c/mau/determine_power_usage.h"
#include "backends/tofino/bf-p4c/mau/jbay_next_table.h"
#include "backends/tofino/bf-p4c/mau/mau_power.h"
#include "backends/tofino/bf-p4c/mau/table_dependency_graph.h"
#include "backends/tofino/bf-p4c/phv/phv_fields.h"
#include "ir/pass_manager.h"
 
namespace MauPower {
 
class MauFeatures;

class FinalizeMauPredDepsPower : public PassManager {
 public:
    FinalizeMauPredDepsPower(const PhvInfo &phv, DependencyGraph &dep_graph,
                             const NextTable *next_table_properties, const BFN_Options &options);
    std::ostream &emit_stage_asm(std::ostream &out, gress_t g, int stage) const;
    bool requires_stage_asm(gress_t g, int stage) const;
 
 private:
    // inputs
    const PhvInfo &phv_;
    DependencyGraph &dep_graph_;
    const NextTable *next_table_properties_;
    const BFN_Options &options_;
 
    // derived
    MauFeatures *mau_features_;
    // flag indicating if table placement used more stages than available in the device.
    bool exceeds_stages_ = false;
    // map from match table UniqueId to tuple of memory accesses
    // A match table will also include any attached table's memory accesses used.
    std::map<UniqueId, PowerMemoryAccess> table_memory_access_ = {};
 
    // Per-gress graphs collected that will be walked to estimate power usage and compute MPR.
    BuildPowerGraph *graphs_ = nullptr;
    // Map from gress to MAU MPR settings.
    std::map<gress_t, MprSettings *> mpr_settings_ = {};
};
 
}  // end namespace MauPower
 
#endif /* BF_P4C_MAU_FINALIZE_MAU_PRED_DEPS_POWER_H_ */