reduction_or.h

 
#ifndef BF_P4C_MAU_REDUCTION_OR_H_
#define BF_P4C_MAU_REDUCTION_OR_H_
 
#include <map>
 
#include "backends/tofino/bf-p4c/mau/mau_visitor.h"
#include "lib/ordered_set.h"
 
using namespace P4;

struct ReductionOrInfo {
    using SaluReductionOrGroup = std::map<cstring, ordered_set<const IR::MAU::StatefulAlu *>>;
    using TblReductionOrGroup = std::map<cstring, ordered_set<const IR::MAU::Table *>>;
 
    SaluReductionOrGroup salu_reduction_or_group;
    TblReductionOrGroup tbl_reduction_or_group;
    bool is_reduction_or(const IR::MAU::Instruction *, const IR::MAU::Table *,
                         cstring &red_or_key) const;
 
    void clear() {
        salu_reduction_or_group.clear();
        tbl_reduction_or_group.clear();
    }
};
 
class GatherReductionOrReqs : public MauInspector {
    Visitor::profile_t init_apply(const IR::Node *node) override;
    bool preorder(const IR::MAU::StatefulAlu *) override;
    bool preorder(const IR::MAU::Action *) override { return false; }
 
    ReductionOrInfo &red_or_info;
 
 public:
    explicit GatherReductionOrReqs(ReductionOrInfo &red_or_info) : red_or_info(red_or_info) {}
};
 
#endif /* BF_P4C_MAU_REDUCTION_OR_H_ */