52 lines
1.4 KiB
C++
52 lines
1.4 KiB
C++
#ifndef TDL_INCLUDE_IR_CODEGEN_TUNE_H
|
|
#define TDL_INCLUDE_IR_CODEGEN_TUNE_H
|
|
|
|
#include <map>
|
|
#include <set>
|
|
#include <vector>
|
|
|
|
namespace tdl{
|
|
|
|
namespace ir{
|
|
class value;
|
|
class module;
|
|
class instruction;
|
|
class function;
|
|
}
|
|
|
|
namespace codegen{
|
|
|
|
class tune {
|
|
typedef std::pair<ir::value*, unsigned> node_t;
|
|
typedef std::map <node_t, std::set<node_t>> graph_t;
|
|
|
|
private:
|
|
void add_constraint(node_t x, node_t y);
|
|
void init_c_phi(ir::instruction *i);
|
|
void init_c_graph(ir::instruction *v);
|
|
void connected_components(node_t x, const std::vector<unsigned*> vals, std::set<node_t> &nodes, graph_t &graph);
|
|
void create_grids(std::vector<ir::instruction*> &grids, std::map<unsigned*, ir::instruction*> &references, ir::function *fn);
|
|
|
|
|
|
public:
|
|
std::vector<unsigned *> get_params(ir::module& mod);
|
|
std::map<std::string, unsigned *> get_params(ir::instruction* i);
|
|
unsigned *get_param(ir::value *value, const std::string &key) { return params_[value][key]; }
|
|
void copy(ir::value *dst, ir::value *src) { params_[dst] = params_[src]; }
|
|
bool check_constraints(ir::module &fn, std::map<ir::value *, std::vector<std::string>> &errors);
|
|
void run(ir::module &mod);
|
|
|
|
private:
|
|
std::map<ir::value*, std::map<std::string, unsigned*>> params_;
|
|
std::vector<unsigned*> pool_;
|
|
graph_t dependencies_;
|
|
std::set<node_t> nodes_;
|
|
std::map<node_t, unsigned> static_params_;
|
|
};
|
|
|
|
|
|
}
|
|
}
|
|
|
|
#endif
|