mau_backtracker.h

 
#ifndef BF_P4C_PHV_MAU_BACKTRACKER_H_
#define BF_P4C_PHV_MAU_BACKTRACKER_H_
 
#include "backends/tofino/bf-p4c/mau/table_summary.h"
#include "ir/ir.h"
#include "lib/ordered_map.h"
 
class MauBacktracker : public Backtrack {
 private:
    State::state_t &state;
    static constexpr unsigned NUM_LOGICAL_TABLES_PER_STAGE = 16;
    int numInvoked = 0;
 
    ordered_map<cstring, ordered_set<int>> tables;
    ordered_map<cstring, ordered_set<int>> prevRoundTables;
 
    ordered_map<int, ordered_set<ordered_set<cstring>>> table_order;
 
    ordered_map<cstring, ordered_set<int>> internalTables;
 
    ordered_map<cstring, std::pair<cstring, cstring>> mergedGateways;
 
    // true if a valid table placement is stored in tables. NOTE: we need this variable instead
    // of just using !tables.empty() because for program with no table, an empty alloc is valid.
    bool has_table_placement = false;
 
    int maxStage = -1;
 
    bool metaInitDisable = false;
 
    bool ignorePackConflicts = false;
 
    bool firstRoundFit = false;
 
    TableSummary *table_summary;
 
    const IR::Node *apply_visitor(const IR::Node *root, const char *) override;
 
    bool backtrack(trigger &) override;
 
    ordered_set<int> inSameStage(const IR::MAU::Table *t1, const IR::MAU::Table *t2,
                                 const ordered_map<cstring, ordered_set<int>> &tableMap) const;
 
 public:
    ordered_set<int> inSameStage(const IR::MAU::Table *t1, const IR::MAU::Table *t2) const;
 
    std::string printTableAlloc() const;
 
    bool hasTablePlacement() const;
 
    ordered_set<int> stage(const IR::MAU::Table *t, bool internal = false) const;
 
    bool disableMetadataInitialization() const { return metaInitDisable; }
 
    bool didFirstRoundFit() const { return firstRoundFit; }
 
    bool happensBefore(const IR::MAU::Table *t1, const IR::MAU::Table *t2) const;
 
    int numStages() const;
 
    const TableSummary *get_table_summary() const { return table_summary; };
 
    const ordered_map<cstring, ordered_set<int>> &getInternalTablesInfo() const {
        return internalTables;
    }
 
    const ordered_map<cstring, std::pair<cstring, cstring>> &getMergedGatewaysInfo() const {
        return mergedGateways;
    }
 
    void clear();
 
    explicit MauBacktracker(State::state_t &state = *(new State::state_t),
                            TableSummary *table_summary = nullptr)
        : state(state), table_summary(table_summary) {}
};
 
#endif /* BF_P4C_PHV_MAU_BACKTRACKER_H_ */