gh0s7/mysysy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[midend][backend]同步后端逻辑，修复编译错误

Browse Source
This commit is contained in:
Lixuanwang
2025-07-26 12:53:21 +08:00
parent 3df9b3bb06
commit 4b181261ce
2 changed files with 42 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
src/RISCv64AsmPrinter.cpp
View File

@@ -47,45 +47,45 @@ void RISCv64AsmPrinter::printPrologue() {
*OS << " addi s0, sp, " << aligned_stack_size << "\n"; *OS << " addi s0, sp, " << aligned_stack_size << "\n";
} }
// 为函数参数分配寄存器 // // 为函数参数分配寄存器
Function* F = MFunc->getFunc(); // Function* F = MFunc->getFunc();
if (F && F->getEntryBlock()) { // if (F && F->getEntryBlock()) {
int arg_idx = 0; // int arg_idx = 0;
RISCv64ISel* isel = MFunc->getISel(); // RISCv64ISel* isel = MFunc->getISel();
// 获取函数所有参数的类型列表 // // 获取函数所有参数的类型列表
auto param_types = F->getParamTypes(); // auto param_types = F->getParamTypes();
for (AllocaInst* alloca_for_param : F->getEntryBlock()->getArguments()) { // for (AllocaInst* alloca_for_param : F->getEntryBlock()->getArguments()) {
if (arg_idx >= 8) break; // if (arg_idx >= 8) break;
unsigned vreg = isel->getVReg(alloca_for_param); // unsigned vreg = isel->getVReg(alloca_for_param);
if (frame_info.alloca_offsets.count(vreg)) { // if (frame_info.alloca_offsets.count(vreg)) {
int offset = frame_info.alloca_offsets.at(vreg); // int offset = frame_info.alloca_offsets.at(vreg);
auto arg_reg = static_cast<PhysicalReg>(static_cast<int>(PhysicalReg::A0) + arg_idx); // auto arg_reg = static_cast<PhysicalReg>(static_cast<int>(PhysicalReg::A0) + arg_idx);
// 1. 获取当前参数的真实类型 // // 1. 获取当前参数的真实类型
// 注意：F->getParamTypes() 返回的是一个 range-based view，需要转换为vector或直接使用 // // 注意：F->getParamTypes() 返回的是一个 range-based view，需要转换为vector或直接使用
Type* current_param_type = nullptr; // Type* current_param_type = nullptr;
int temp_idx = 0; // int temp_idx = 0;
for(auto p_type : param_types) { // for(auto p_type : param_types) {
if (temp_idx == arg_idx) { // if (temp_idx == arg_idx) {
current_param_type = p_type; // current_param_type = p_type;
break; // break;
} // }
temp_idx++; // temp_idx++;
} // }
assert(current_param_type && "Could not find parameter type."); // assert(current_param_type && "Could not find parameter type.");
// 2. 根据类型决定使用 "sw" 还是 "sd" // // 2. 根据类型决定使用 "sw" 还是 "sd"
const char* store_op = current_param_type->isPointer() ? "sd" : "sw"; // const char* store_op = current_param_type->isPointer() ? "sd" : "sw";
// 3. 打印正确的存储指令 // // 3. 打印正确的存储指令
*OS << " " << store_op << " " << regToString(arg_reg) << ", " << offset << "(s0)\n"; // *OS << " " << store_op << " " << regToString(arg_reg) << ", " << offset << "(s0)\n";
} // }
arg_idx++; // arg_idx++;
} // }
} // }
} }
void RISCv64AsmPrinter::printEpilogue() { void RISCv64AsmPrinter::printEpilogue() {

12
src/RISCv64ISel.cpp
View File

@@ -51,22 +51,28 @@ std::unique_ptr<MachineFunction> RISCv64ISel::runOnFunction(Function* func) {
// 指令选择主流程 // 指令选择主流程
void RISCv64ISel::select() { void RISCv64ISel::select() {
// 遍历基本块，为它们创建对应的MachineBasicBlock
for (const auto& bb_ptr : F->getBasicBlocks()) { for (const auto& bb_ptr : F->getBasicBlocks()) {
auto mbb = std::make_unique<MachineBasicBlock>(bb_ptr->getName(), MFunc.get()); auto mbb = std::make_unique<MachineBasicBlock>(bb_ptr->getName(), MFunc.get());
bb_map[bb_ptr.get()] = mbb.get(); bb_map[bb_ptr.get()] = mbb.get();
MFunc->addBlock(std::move(mbb)); MFunc->addBlock(std::move(mbb));
} }
if (F->getEntryBlock()) { // 遍历Argument对象，为它们分配虚拟寄存器。
for (auto* arg_alloca : F->getEntryBlock()->getArguments()) { if (F) {
getVReg(arg_alloca); for (Argument* arg : F->getArguments()) {
// getVReg会为每个代表参数的Argument对象在vreg_map中创建一个条目。
// 这样，当后续的store指令使用这个参数时，就能找到对应的vreg。
getVReg(arg);
} }
} }
// 遍历基本块，进行指令选择
for (const auto& bb_ptr : F->getBasicBlocks()) { for (const auto& bb_ptr : F->getBasicBlocks()) {
selectBasicBlock(bb_ptr.get()); selectBasicBlock(bb_ptr.get());
} }
// 链接MachineBasicBlock的前驱和后继
for (const auto& bb_ptr : F->getBasicBlocks()) { for (const auto& bb_ptr : F->getBasicBlocks()) {
CurMBB = bb_map.at(bb_ptr.get()); CurMBB = bb_map.at(bb_ptr.get());
for (auto succ : bb_ptr->getSuccessors()) { for (auto succ : bb_ptr->getSuccessors()) {