[backend]开始区分调用者保存寄存器与被调用者保存寄存器

2025-07-26 12:39:17 +08:00
parent 5bfa6d72a2
commit e57ac7709d
2 changed files with 9 additions and 2 deletions
--- a/src/RISCv64RegAlloc.cpp
+++ b/src/RISCv64RegAlloc.cpp
@@ -436,8 +436,12 @@ void RISCv64RegAlloc::rewriteFunction() {
                    if (reg_op->isVirtual()) {
                        unsigned vreg = reg_op->getVRegNum();
                        if (color_map.count(vreg)) {
-                            // 如果vreg被成功着色，替换为物理寄存器
-                            reg_op->setPReg(color_map.at(vreg));
+                            PhysicalReg preg = color_map.at(vreg);
+                            reg_op->setPReg(preg);
+                            // 检查这个物理寄存器是否是 s0-s11
+                            if (preg >= PhysicalReg::S0 && preg <= PhysicalReg::S11) {
+                                used_callee_saved_regs.insert(preg);
+                            }
                        } else if (spilled_vregs.count(vreg)) {
                            // 如果vreg被溢出，替换为专用的溢出物理寄存器t6
                            reg_op->setPReg(PhysicalReg::T6);