ZYM/MXCompiler
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

can out put colored program

Browse Source
This commit is contained in:
ZhuangYumin
2024-10-22 02:05:22 +00:00
parent 8216cb774c
commit bc742f4ee7
3 changed files with 303 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

167
src/opt/regalloc.cpp
View File

@ -136,13 +136,176 @@ void EnforcePhysicalRegs(CFGType &cfg) {
entry_node->corresponding_block->actions.push_front(new_def);
}
}
void TranslateColorResult(std::shared_ptr<FunctionDefItem> func, CFGType &cfg, ConfGraph &confgraph) {
using namespace opt;
std::unordered_map<std::string, size_t> spilled_var_to_id;
auto &var_to_id = cfg.var_to_id;
auto ApplyColoringResult = [&](std::string &var) {
auto confnode = confgraph.id_to_node[var_to_id[var]];
confnode = ConfGraphNode::FindFather(confnode);
if (!confnode->is_binded_with_physical_reg) {
if (spilled_var_to_id.find(var) != spilled_var_to_id.end()) {
var = "#" + std::to_string(spilled_var_to_id[var]);
} else {
spilled_var_to_id[var] = func->spilled_vars++;
var = "#" + std::to_string(spilled_var_to_id[var]);
}
} else {
var = "$reg." + std::to_string(confnode->color);
}
};
for (auto node : cfg.nodes) {
auto block = node->corresponding_block;
std::vector<size_t> cur_node_use;
std::vector<size_t> cur_node_def;
bool use_def_init = false;
func->spilled_vars = 0;
for (auto act : block->actions) {
if (auto br_act = std::dynamic_pointer_cast<BRAction>(act)) {
if (var_to_id.find(br_act->cond) != var_to_id.end()) {
ApplyColoringResult(br_act->cond);
}
} else if (auto ret_act = std::dynamic_pointer_cast<RETAction>(act)) {
if (!std::holds_alternative<LLVMVOIDType>(ret_act->type) && var_to_id.find(ret_act->value) != var_to_id.end()) {
ApplyColoringResult(ret_act->value);
}
} else if (auto bin_act = std::dynamic_pointer_cast<BinaryOperationAction>(act)) {
if (var_to_id.find(bin_act->operand1_full) != var_to_id.end()) {
ApplyColoringResult(bin_act->operand1_full);
}
if (var_to_id.find(bin_act->operand2_full) != var_to_id.end()) {
ApplyColoringResult(bin_act->operand2_full);
}
if (var_to_id.find(bin_act->result_full) != var_to_id.end()) {
ApplyColoringResult(bin_act->result_full);
}
} else if (auto load_act = std::dynamic_pointer_cast<LoadAction>(act)) {
if (var_to_id.find(load_act->ptr_full) != var_to_id.end()) {
ApplyColoringResult(load_act->ptr_full);
}
if (var_to_id.find(load_act->result_full) != var_to_id.end()) {
ApplyColoringResult(load_act->result_full);
}
} else if (auto store_act = std::dynamic_pointer_cast<StoreAction>(act)) {
if (var_to_id.find(store_act->value_full) != var_to_id.end()) {
ApplyColoringResult(store_act->value_full);
}
if (var_to_id.find(store_act->ptr_full) != var_to_id.end()) {
ApplyColoringResult(store_act->ptr_full);
}
} else if (auto get_act = std::dynamic_pointer_cast<GetElementPtrAction>(act)) {
std::unordered_set<std::string> used_vars;
if (var_to_id.find(get_act->ptr_full) != var_to_id.end()) {
ApplyColoringResult(get_act->ptr_full);
}
for (auto &idx : get_act->indices) {
if (var_to_id.find(idx) != var_to_id.end()) {
ApplyColoringResult(idx);
}
}
if (var_to_id.find(get_act->result_full) != var_to_id.end()) {
ApplyColoringResult(get_act->result_full);
}
} else if (auto icmp_act = std::dynamic_pointer_cast<ICMPAction>(act)) {
if (var_to_id.find(icmp_act->operand1_full) != var_to_id.end()) {
ApplyColoringResult(icmp_act->operand1_full);
}
if (var_to_id.find(icmp_act->operand2_full) != var_to_id.end()) {
ApplyColoringResult(icmp_act->operand2_full);
}
if (var_to_id.find(icmp_act->result_full) != var_to_id.end()) {
ApplyColoringResult(icmp_act->result_full);
}
} else if (auto call_act = std::dynamic_pointer_cast<CallItem>(act)) {
for (auto &arg : call_act->args_val_full) {
if (var_to_id.find(arg) != var_to_id.end()) {
ApplyColoringResult(arg);
}
}
if (!std::holds_alternative<LLVMVOIDType>(call_act->return_type) &&
var_to_id.find(call_act->result_full) != var_to_id.end()) {
ApplyColoringResult(call_act->result_full);
}
} else if (auto select_act = std::dynamic_pointer_cast<SelectItem>(act)) {
if (var_to_id.find(select_act->cond_full) != var_to_id.end()) {
ApplyColoringResult(select_act->cond_full);
}
if (var_to_id.find(select_act->true_val_full) != var_to_id.end()) {
ApplyColoringResult(select_act->true_val_full);
}
if (var_to_id.find(select_act->false_val_full) != var_to_id.end()) {
ApplyColoringResult(select_act->false_val_full);
}
if (var_to_id.find(select_act->result_full) != var_to_id.end()) {
ApplyColoringResult(select_act->result_full);
}
} else if (auto move_act = std::dynamic_pointer_cast<MoveInstruct>(act)) {
if (VRegCheck(move_act->src_full) && var_to_id.find(move_act->src_full) != var_to_id.end()) {
ApplyColoringResult(move_act->src_full);
}
if (var_to_id.find(move_act->dest_full) != var_to_id.end()) {
ApplyColoringResult(move_act->dest_full);
}
} else if (auto force_def_act = std::dynamic_pointer_cast<ForceDef>(act)) {
if (var_to_id.find(force_def_act->var_full) != var_to_id.end()) {
ApplyColoringResult(force_def_act->var_full);
}
} else if (auto force_use_act = std::dynamic_pointer_cast<ForceUse>(act)) {
if (var_to_id.find(force_use_act->var_full) != var_to_id.end()) {
ApplyColoringResult(force_use_act->var_full);
}
} else if (auto load_spilled_args_act = std::dynamic_pointer_cast<LoadSpilledArgs>(act)) {
if (var_to_id.find(load_spilled_args_act->var_full) != var_to_id.end()) {
ApplyColoringResult(load_spilled_args_act->var_full);
}
} else if (auto store_spilled_args_act = std::dynamic_pointer_cast<StoreSpilledArgs>(act)) {
if (var_to_id.find(store_spilled_args_act->var_full) != var_to_id.end()) {
ApplyColoringResult(store_spilled_args_act->var_full);
}
} else if (auto alloca_act = std::dynamic_pointer_cast<AllocaAction>(act)) {
ApplyColoringResult(alloca_act->name_full);
}
}
{
auto act = block->exit_action;
if (auto br_act = std::dynamic_pointer_cast<BRAction>(act)) {
if (var_to_id.find(br_act->cond) != var_to_id.end()) {
ApplyColoringResult(br_act->cond);
}
} else if (auto ret_act = std::dynamic_pointer_cast<RETAction>(act)) {
if (!std::holds_alternative<LLVMVOIDType>(ret_act->type) && var_to_id.find(ret_act->value) != var_to_id.end()) {
ApplyColoringResult(ret_act->value);
}
}
}
for (auto act_it = block->actions.begin(); act_it != block->actions.end(); ++act_it) {
bool need_remove = false;
if (auto move_act = std::dynamic_pointer_cast<MoveInstruct>(*act_it)) {
if (move_act->src_full == move_act->dest_full) {
need_remove = true;
}
} else if (auto force_def_act = std::dynamic_pointer_cast<ForceDef>(*act_it)) {
need_remove = true;
} else if (auto force_use_act = std::dynamic_pointer_cast<ForceUse>(*act_it)) {
need_remove = true;
}
if (need_remove) {
auto it_next = act_it;
++it_next;
block->actions.erase(act_it);
act_it = it_next;
--act_it;
}
}
}
}
void ConductRegAllocForFunction(std::shared_ptr<FunctionDefItem> func) {
std::cerr << "processing function " << func->func_name_raw << std::endl;
CFGType cfg;
ConfGraph confgraph;
cfg = BuildCFGForFunction(func);
EnforcePhysicalRegs(cfg);
func->RecursivePrint(std::cerr);
// func->RecursivePrint(std::cerr);
do {
cfg.init();
cfg = BuildCFGForFunction(func);
@ -150,6 +313,8 @@ void ConductRegAllocForFunction(std::shared_ptr<FunctionDefItem> func) {
confgraph.init();
confgraph = BuildConfGraph(cfg);
} while (ConductColoring(func, cfg, confgraph));
TranslateColorResult(func, cfg, confgraph);
func->RecursivePrint(std::cerr);
}
std::shared_ptr<ModuleItem> RegAlloc(std::shared_ptr<ModuleItem> src) {