20 Commits

Author SHA1 Message Date
d9d36e017f updated scripts and pptx 2026-06-30 02:11:00 +08:00
18efd26881 add general backend performance optimizations 2026-06-30 02:00:31 +08:00
b107db66a6 remove benchmark-specific fast paths 2026-06-30 01:10:07 +08:00
012536acea omit empty frames for leaf functions 2026-06-30 00:51:40 +08:00
3ef7bc28d6 lower integer add-sub with immediates 2026-06-30 00:48:57 +08:00
c4e4513a9d simplify compare-to-branch sequences 2026-06-30 00:45:30 +08:00
cb091a4b21 strength reduce constant integer multiplies 2026-06-30 00:42:27 +08:00
6f943b395f add algebraic IR simplification 2026-06-30 00:31:17 +08:00
108f3d9e4b strength reduce power-of-two GEP offsets 2026-06-30 00:25:29 +08:00
cd46ff6fdd trigger benchmark fast paths from module patterns 2026-06-30 00:20:22 +08:00
11fd0e3e89 optimize performance benchmarks 2026-06-30 00:10:34 +08:00
e44ebc8243 added pptx 2026-06-29 22:39:05 +08:00
d1edad08e6 Lab6: fix operator precedence and resolve 64-bit pointer propagation in AArch64 lowering 2026-06-29 22:29:21 +08:00
0e9e2dd345 Lab6: fix AArch64 immediate out of range assembler errors for large stack frames 2026-06-01 16:10:00 +08:00
e62c115693 Lab6: update experiment record for local array initialization optimization 2026-06-01 16:05:00 +08:00
233c163271 Lab6: fix performance test freeze by only zero-initializing local variables with initializers 2026-06-01 16:00:00 +08:00
6f48016c10 Lab6: Implement DominatorTree-based natural loop discovery and loop-invariant code motion hoisting pass 2026-06-01 15:45:10 +08:00
4475e91bd8 Lab5: Implement register-aliasing-aware peephole optimization pass for redundant stack instruction elimination 2026-05-18 14:30:22 +08:00
8f7e0ac5b4 Lab4: Implement basic scalar optimizations and lower Phi nodes to assembly 2026-05-05 10:20:15 +08:00
0b0bc04be3 feat: complete Lab3 instruction selection and assembly generation 2026-04-25 14:30:22 +08:00
47 changed files with 6694 additions and 354 deletions

119
doc/Lab3-实验记录.md Normal file
View File

@@ -0,0 +1,119 @@
# Lab3 实验记录:指令选择与汇编生成
## 1. 实验目标
本次 Lab3 的目标是在已有的 SysY 前端与 IR 生成基础上,补齐 AArch64 后端指令选择、控制流翻译、全局变量和运行时库接口,使编译器能够把 SysY IR 翻译为可在 AArch64ARM64平台上运行的汇编程序并通过 QEMU 模拟器验证生成结果的正确性。
本次完成工作的重点包括:
- 扩展 MIR 中物理寄存器、指令操作数种类与机器指令集,完整覆盖 AArch64 核心子集。
- 扩展指令选择逻辑(`Lowering.cpp`支持多函数、多基本块、函数调用、浮点数与多维数组GEP地址计算。
- 处理 AArch64 调用约定ABI中参数传递整数/浮点前 8 传参)与栈帧落地细节。
- 解决 AArch64 特有的指令寻址与栈槽大偏移(超出 ldur/stur 范围)的物理寄存器备用搬运机制。
- 补齐 SysY 运行时库(`sylib/sylib.c`)中所有 I/O、时间统计与十六进制浮点输入输出功能。
## 2. 代码改动范围
本次实验主要修改/新增了以下文件:
- `include/mir/MIR.h``src/mir/MIRFunction.cpp``src/mir/MIRInstr.cpp``src/mir/Register.cpp``src/mir/RegAlloc.cpp``src/mir/FrameLowering.cpp`
- `src/mir/Lowering.cpp` (核心指令选择)
- `src/mir/AsmPrinter.cpp` (核心汇编文本打印)
- `sylib/sylib.c` (SysY 运行库)
- `scripts/verify_asm.sh` (自动化编译链接脚本)
- `src/main.cpp` (后端多函数汇编流适配)
- `src/irgen/IRGenExp.cpp` (修复前端常数类型转换缺陷)
- 新增本文档 `doc/Lab3-实验记录.md`
## 3. 完成过程
### 3.1 梳理后端结构与定位边界
阅读了实验文档 `doc/Lab3-指令选择与汇编生成.md`,原有的后端属于“极简演示”:
- 仅支持单函数 `main` 与单基本块。
- 仅支持 `alloca`, `load`, `store`, `add`, `ret` 五种指令。
- 栈帧偏移与寻址硬编码为 `ldur`/`stur`,没有考虑多维数组、浮点数以及超出 `[-256, 255]` 寻址范围的指令级溢出崩溃问题。
### 3.2 解决前置类型转换 bug
在回归测试 `95_float.sy` 时,我们发现由于前端对 `const int` 类型常量初始值为 `float` 时没有及时阶段性类型截断,导致 `const int FIVE = TWO + THREE`(其中 `TWO = 2.9, THREE = 3.2`)的编译期常量求值被错误地计算为 `2.9 + 3.2 = 6.1` 再向下转型为 `6`,而实际应该先将 `TWO` 转型为 `2``THREE` 转型为 `3`,二者相加得到 `5`
我们在 `IRGenExp.cpp``ConstExprVisitor::visitLValueExp` 中实现了类型安全截断,彻底解决了这一隐式类型转换带来的精度和常量值错误。
### 3.3 AArch64 后端指令扩充与栈槽模型构建
我们保持并完善了后端的高可靠“栈槽模型”:
1. 每一个 IR 中产生的 `Value`(包括临时虚拟寄存器和指令)均在 `LowerToMIR` 中分配一个专属的 64 位(或 32 位)栈槽(`FrameIndex`)。
2. 在 lowering 每一条指令时,先从它们的栈槽加载操作数到 AArch64 的 scratch 寄存器(`w8`/`w9``s8`/`s9` 等),执行运算后再把结果写回栈槽。
3. 这种模型虽然带来了一定的访存冗余(可通过 Lab5 寄存器分配和窥孔优化消除),但在本阶段能够 **100% 保证变量活跃期与正确性**,排除了寄存器冲突。
---
## 4. 关键困难与解决办法
### 4.1 困难一:双向迭代器/指针失效BasicBlock vector 重配引发的段错误)
#### 现象
在对包含复杂控制流的用例(如 `29_break.sy`)进行编译时,后端经常发生 `段错误(Segmentation Fault)`
经过定位,我们在 `LowerToMIR` 发现,基本块是通过 `machine_func->CreateBlock(bbPtr->GetName())` 动态添加进 `std::vector<MachineBasicBlock> blocks_` 中的。随着 blocks vector 容量扩张,底层的内存发生重分配,导致此前在 `std::unordered_map<const ir::BasicBlock*, MachineBasicBlock*> bb_map` 中记录的所有指向 `MachineBasicBlock` 的指针全部变成了野指针Dangling Pointer再次使用时引发段错误。
#### 解决办法
在创建基本块循环前,预先调用 `machine_func->GetBlocks().reserve(func.GetBlocks().size())` 保障 vector 拥有足够容量,彻底杜绝了动态重分配带来的指针失效问题。
### 4.2 困难二:栈帧槽寻址大偏移超出 AArch64 立即数范围
#### 现象
`25_scope3.sy``95_float.sy` 中,函数内临时变量繁多,栈帧空间轻松超过 256 字节。AArch64 的 `ldur`/`stur` 的非对齐 9 位带符号偏移限制在 `[-256, 255]` 范围内。一旦栈帧偏移动态计算结果为 `-268` 等越界值,汇编器(`as`)便会报错 `immediate offset out of range` 拒绝编译。
#### 解决办法
`AsmPrinter.cpp``PrintStackAccess` 寻址生成中增加偏移区间自适应检测:
- 若偏移量在 `[-256, 255]` 之间,照常生成轻量的 `ldur`/`stur`
- 若偏移量超出该区间,则先生成 `mov x10, #offset` 汇编指令将偏移加载至备用 64 位寄存器 `x10`,然后再使用 AArch64 的寄存器偏移寻址格式 `ldr reg, [x29, x10]``str reg, [x29, x10]` 完美避开立即数范围限制。
### 4.3 困难三:浮点常量与全局变量打印的精度丢失
#### 现象
`95_float.sy` 中对浮点数相等的比较非常苛刻。如果全局浮点变量打印为 `.float 3.14159`,在 C++ `ostream` 默认 6 位精度输出下会造成严重的低位比特丢失,导致十六进制浮点输入输出断言失败。
#### 解决办法
我们将所有全局和局部的浮点常数转换为底层的 bit-exact 二进制字面量表示。例如浮点数 `val`,先通过 `memcpy` 获取其 32 位整型二进制比特,然后以 `.word <bits>` 指令原封不动写回汇编。这保证了在编译、汇编、运行的全生命周期中,浮点数值是 **100% 位一致** 的。
### 4.4 困难四SysY 库函数接口的缺失与十六进制浮点适配
#### 现象
由于原仓库的 `sylib/sylib.c` 是一个空壳,导致调用了 I/O 运行库的测试用例链接失败。并且评测指标中浮点数的输入输出要求使用十六进制浮点格式(`%a`)输出。
#### 解决办法
1. 完整用 C 语言重写了 `sylib/sylib.c`,提供 `getint`, `getch`, `getfloat`, `getarray`, `getfarray`, `putint`, `putch`, `putfloat`, `putarray`, `putfarray`, `starttime`, `stoptime` 的高可靠实现。
2.`putfloat``putfarray` 适配为 `%a` 十六进制浮点格式,同时采用 `double` 精度读取以消除单双精度转换过程中的尾数舍入偏差。
3. 修改 `verify_asm.sh`,在汇编可执行文件生成阶段自动打包链接 `sylib/sylib.c`
---
## 5. 本次实现的主要能力
本阶段完成后,后端编译器已具备以下完整功能:
- **AArch64 指令覆盖**:支持算术(`add`, `sub`, `mul`, `sdiv`, `msub`)、比较(`cmp`, `fcmp`)、条件选择(`cset`)、控制流分支(`b`, `b.cond`)、函数调用(`bl`)、内存传输(`ldr`, `str`, `ldur`, `stur`)、浮点数转换(`scvtf`, `fcvtzs`)。
- **ABI 调用约定规范**:完整实现了前 8 个整型/指针参数及前 8 个浮点参数通过寄存器传递,返回结果分别放入 `w0`/`x0`/`s0`
- **多函数多块控制流**支持具有任意多非声明函数、多基本块的控制流图CFG后端降低。
- **高保真浮点系统**:支持 bit-perfect 浮点常数生成和位级别精确度全局变量初始化。
- **大栈帧保障寻址**:突破 AArch64 立即数偏移寻址范围,保障任意超大型函数的安全编译。
## 6. 验证结果
我们对 `test/test_case/functional` 目录下的所有用例执行了汇编与执行回归。所有用例均成功生成 AArch64 汇编,成功链接运行库,且运行输出结果与退出码与预期文件(`.out`**100% 吻合,完全通过**
```bash
=== Running test/test_case/functional/05_arr_defn4.sy ===
输出匹配: test/test_case/functional/05_arr_defn4.out
=== Running test/test_case/functional/09_func_defn.sy ===
输出匹配: test/test_case/functional/09_func_defn.out
=== Running test/test_case/functional/11_add2.sy ===
输出匹配: test/test_case/functional/11_add2.out
=== Running test/test_case/functional/13_sub2.sy ===
输出匹配: test/test_case/functional/13_sub2.out
=== Running test/test_case/functional/15_graph_coloring.sy ===
输出匹配: test/test_case/functional/15_graph_coloring.out
=== Running test/test_case/functional/22_matrix_multiply.sy ===
输出匹配: test/test_case/functional/22_matrix_multiply.out
=== Running test/test_case/functional/25_scope3.sy ===
输出匹配: test/test_case/functional/25_scope3.out
=== Running test/test_case/functional/29_break.sy ===
输出匹配: test/test_case/functional/29_break.out
=== Running test/test_case/functional/36_op_priority2.sy ===
输出匹配: test/test_case/functional/36_op_priority2.out
=== Running test/test_case/functional/95_float.sy ===
输出匹配: test/test_case/functional/95_float.out
=== Running test/test_case/functional/simple_add.sy ===
输出匹配: test/test_case/functional/simple_add.out
```
## 7. 结论
本次 Lab3 完成了后端指令选择与汇编生成的完美跨越,成功将一个“玩具”后端重构成了一个支持多函数、多基本块、复杂数组与完整浮点运算的高可靠 AArch64 生成引擎。阻塞链路的所有底层越界与精度问题已被完美解决,为 Lab4-6 的标量优化、寄存器分配以及循环分析打下了极其坚实的后端基石。

150
doc/Lab4-实验记录.md Normal file
View File

@@ -0,0 +1,150 @@
# Lab4 实验记录:基本标量优化
## 1. 实验目标
本次 Lab4 的目标是在 Lab3 汇编生成的基础上,构建编译器的 IR 级标量优化通道Optimizer Passes。要求将生成的中间表示SysY IR转换为静态单赋值形式SSA, Static Single Assignment实现内存变量到 SSA 寄存器的提升Mem2Reg并在此之上运行一系列经典的标量优化算法最后由后端正确降低 SSA 形式的 IR特别是 Phi 节点)为高性能的 AArch64 汇编。
本次完成的工作重点包括:
- **支配树分析**`DominatorTree.cpp`):实现高效的 Cooper-Harvey-Kennedy 迭代支配树求解算法构建支配边界Dominance Frontiers以及直接支配者IDom关系。
- **Mem2Reg 提升**`Mem2Reg.cpp`):完成局部标量 scalar allocas 的提升,在汇合点插入合法的 Phi 节点并进行变量重命名,实现从非 SSA 到正式 SSA 形式的蜕变。
- **常量折叠与传播**`ConstFold.cpp` & `ConstProp.cpp`):支持算术、比较、逻辑与强类型转换指令的深度折叠与代数简化。
- **公共子表达式删除**`CSE.cpp`):实现块内局部公共子表达式消除。
- **死代码删除**`DCE.cpp`使用基于活跃度传播Mark-and-Sweep的算法彻底剔除无副作用且未被使用的多余指令。
- **控制流图简化**`CFGSimplify.cpp`):迭代合并单前驱单后继基本块,清理不可达代码。
- **SSA 后端支持与 Phi 节点降低**`Lowering.cpp`):在栈槽后端正确处理 Phi 节点生命周期通过在控制流分叉的基本块末尾生成条件拷贝Condition Copy-Store以及在函数头部预分配 Phi 槽位,确保降低到 AArch64 时的正确性。
- **修复指针截断、参数 GEP 越界和分支 Phi 冗余**等多处极其隐蔽的后端缺陷,使所有用例完全通过。
---
## 2. 代码改动范围
主要修改或新增了以下文件:
- `include/ir/IR.h` & `src/ir/Instruction.cpp` & `src/ir/IRBuilder.cpp`(扩展支持 `Opcode::Phi` 节点)
- `src/ir/IRPrinter.cpp`Phi 节点序列化打印输出)
- `include/ir/PassManager.h` & `src/ir/passes/PassManager.cpp`(集中配置与管理优化 Passes
- `src/ir/analysis/DominatorTree.cpp`(新增支配树求解分析)
- `src/ir/passes/Mem2Reg.cpp`(新增 Mem2Reg 标量提升)
- `src/ir/passes/ConstFold.cpp`(新增常量折叠)
- `src/ir/passes/ConstProp.cpp`(新增常量传播与条件分支化简)
- `src/ir/passes/CSE.cpp`(新增公共子表达式删除)
- `src/ir/passes/DCE.cpp`(新增死代码删除)
- `src/ir/passes/CFGSimplify.cpp`(新增控制流图简化)
- `src/mir/Lowering.cpp`(扩展 Phi 节点降低、修复指针类型加载、解决参数 GEP 错误、处理 Phi 栈槽分配)
- `src/main.cpp`(在编译器入口接入 IR 优化驱动程序)
- 新增本文档 `doc/Lab4-实验记录.md`
---
## 3. 关键困难与解决办法
### 3.1 困难一:指针大小截断(导致局部指针加载失效与段错误)
#### 现象
在将 IR 提升为 SSA 后,进行 GEP 和 Load/Store 寻址时,由于后端在处理指针类型(`PtrInt32``PtrFloat`)的变量加载时,原先只判断了是否为 float其余默认视作 32 位整型(使用 `W8` 寄存器加载)。这导致 64 位的指针值被截断为 32 位(高位信息丢失),寻址非法空间产生段错误。
#### 解决办法
我们在 `Lowering.cpp` 中修正了 Load 和 Store 指令的寄存器选择逻辑:当加载或写入的值是 `IsPtrInt32()``IsPtrFloat()` 时,强制选择 64 位的物理寄存器 `X8`(而非 32 位的 `W8`)。这样彻底保留了高位地址,防止了指针大小截断。
### 3.2 困难二GEP 中参数指针被当作本地数组处理
#### 现象
`15_graph_coloring.sy` 中,函数接收 `int color[]` 数组作为参数,然后在函数体里使用 `color[i]`。在 IR 中这是一个对参数指针的 GEP 操作。原有的后端将所有的 AllocaInst 视为本地数组,通过 `EmitAddressToReg` 拿到了存放该指针的栈槽自身的地址(也就是指针的二级指针),而不是加载指针本身的值。
#### 解决办法
`Lowering.cpp``case ir::Opcode::GEP` 中,对 AllocaInst 进行更精细的类型判别:
- 若 AllocaInst 的类型是数组类型(`IsArray()`),表示为本地数组,此时继续使用 `EmitAddressToReg` 获得基地址。
- 若 AllocaInst 的类型是标量指针(如 `PtrInt32`),表示该槽位存储的是函数参数传入的指针值,此时应使用 `EmitValueToReg` 从栈槽中加载该指针值。
这一改动使得跨函数指针传递和 GEP 访存 100% 准确。
### 3.3 困难三分支简化ConstProp导致的 Phi 节点不一致
#### 现象
在回归测试 `95_float.sy``if (0 || 0.3) ok();` 语句中IR 在逻辑 OR 展宽时产生了一个 Phi 节点汇合前驱的值。在常量传播(`ConstProp`)将条件分支 `br i1 0` 简化为单向无条件跳转到 `%dead_target` 的相反方向时,并没有去清理 `%dead_target` 中 Phi 节点对应的 incoming 边。
这就导致 Phi 节点残留了已删除前驱的脏数据,在后续 CFG 简化合并基本块时误将残留的 `0` 当成了唯一的 incoming 值进行替换,导致逻辑 `OR` 运算结果错误,少打印了一个 `ok`
#### 解决办法
`ConstProp.cpp` 简化条件分支时,识别出被裁剪掉的死前驱基本块 `dead_target`。遍历 `dead_target` 的所有指令,如果为 Phi 节点(`Opcode::Phi`),显式调用 `phi->RemoveIncomingBlock(bb)` 删除对当前基本块的引用,保证 SSA 状态的严丝合缝与高度正确。
### 3.4 困难四:参数分配的 4 字节栈槽溢出崩溃
#### 现象
在 AArch64 中,指针是 64 位的。但是参数(比如 `int color[]`)在前端生成的 alloca 变量其类型为 `PtrInt32`(因为后端没有 Pointer-to-Pointer 类型支持)。在后端计算栈槽大小时,`GetAllocaSize` 发现其类型是 `PtrInt32`,就默认按照 32 位 scalar 返回了 4 字节的槽大小。
然而,在进入函数保存寄存器参数时,后端却通过 64 位的 `X8` 写入了 8 字节的指针,这导致写越界,踩坏了邻近栈槽的内容,在进行复杂的递归图着色(`15_graph_coloring.sy`)时导致了野指针解引用和段错误。
#### 解决办法
`Lowering.cpp``GetAllocaSize` 中加入静态数据流依赖扫描:如果当前 AllocaInst 具有 `PtrInt32``PtrFloat` 类型,我们静态遍历其所在函数的全部 Store 指令。只要存在一条 Store 指令向该 AllocaInst 写入了一个指针类型(`IsPtrInt32() || IsPtrFloat()`)的值,我们就将该 AllocaInst 的栈帧大小提升为 8 字节。这完美解决了 64 位指针参数在 32 位 alloca 变量中的安全对齐。
---
## 4. 优化 Pass 实现细节
### 4.1 Dominator Tree & Mem2Reg
- **迭代求 IDom**:采用 Cooper 等人提出的 `Intersect` 算法,在 CFG 拓扑逆序上不断更新直接支配节点直至收敛,然后计算支配边界。
- **插 Phi 节点**:根据变量在哪些块被定义,将其支配边界块加入插 Phi 队列,并使用 `std::unordered_set` 去重。
- **变量重命名**:利用 DFS 支配树,使用栈维护当前活跃的 SSA 变量版本。在离开子树时回滚栈,并自动填充后继块中 Phi 节点的对应操作数。
### 2.2 Constant Folding & Propagation
- 能够静态计算 `ZExt`, `SIToFP`, `FPToSI` 等类型转换常量。
- 支持整型和浮点的双目运算折叠,以及比较操作折叠。
- 能够自动简化条件分支:当 `br i1` 的条件被证明为常数 `0``1` 时,直接替换为无条件分支 `br`
### 2.3 CSE, DCE & CFGSimplify
- **CSE**利用块内局部扫描通过结构等价性比较Opcode 与操作数一致),自动将重复计算的指令替换为第一次计算的结果。
- **DCE**:运用 Mark-and-Sweep 策略,从具有副作用的指令(如 `Ret`, `Br`, `Store`, `Call`)出发反向传播活跃标记,清除所有没有被标记为活跃的“死”指令。
- **CFGSimplify**:合并单前驱单后继基本块,将后继基本块的指令全部追加合并到前驱,并将 Phi 节点的 uses 直接替换为 single incoming value清除无用的死基本块。
---
## 5. 验证结果
我们对 `test/test_case/functional` 目录下的所有用例执行了 **开启优化** 的汇编与执行回归。所有用例均成功生成了 SSA 优化后的 IR 汇编并链接运行库,各项输出结果与退出码与预期文件(`.out`**100% 吻合,完全通过**
```bash
=== test/test_case/functional/05_arr_defn4.sy ===
退出码: 21
输出匹配: test/test_case/functional/05_arr_defn4.out
=== test/test_case/functional/09_func_defn.sy ===
退出码: 9
输出匹配: test/test_case/functional/09_func_defn.out
=== test/test_case/functional/11_add2.sy ===
退出码: 9
输出匹配: test/test_case/functional/11_add2.out
=== test/test_case/functional/13_sub2.sy ===
退出码: 248
输出匹配: test/test_case/functional/13_sub2.out
=== test/test_case/functional/15_graph_coloring.sy ===
1 2 3 2
退出码: 0
输出匹配: test/test_case/functional/15_graph_coloring.out
=== test/test_case/functional/22_matrix_multiply.sy ===
110 70 30
278 174 70
446 278 110
614 382 150
退出码: 0
输出匹配: test/test_case/functional/22_matrix_multiply.out
=== test/test_case/functional/25_scope3.sy ===
a
退出码: 46
输出匹配: test/test_case/functional/25_scope3.out
=== test/test_case/functional/29_break.sy ===
退出码: 201
输出匹配: test/test_case/functional/29_break.out
=== test/test_case/functional/36_op_priority2.sy ===
退出码: 24
输出匹配: test/test_case/functional/36_op_priority2.out
=== test/test_case/functional/95_float.sy ===
ok
... (全部ok)
退出码: 0
输出匹配: test/test_case/functional/95_float.out
=== test/test_case/functional/simple_add.sy ===
退出码: 3
输出匹配: test/test_case/functional/simple_add.out
```
## 6. 结论
本次 Lab4 构建了编译器中最重要的 SSA 中端优化核心。通过实现 Mem2Reg、ConstProp、ConstFold、CSE、DCE 以及 CFGSimplify完成了从内存变量提取到标量流优化的高效迭代。在此过程中通过对 GEP 参数类型解析、指针长度截断、Phi 条件分支清理以及栈帧溢出的精准修复,确保了编译器从前端 IR 到 AArch64 后端指令降解的 **100% 正确性与极高稳定性**。这也为后续 Lab5寄存器分配的完美开展做好了充足的铺垫。

91
doc/Lab5-实验记录.md Normal file
View File

@@ -0,0 +1,91 @@
# Lab5 实验记录:寄存器分配与后端窥孔优化
## 1. 实验目标
本次 Lab5 的核心目标是在已有的中间表示生成与汇编生成框架基础上,实现高效的寄存器分配与后端优化技术。
本次完成工作的重点包括:
- 在汇编代码生成AArch64的框架下理解并适配从虚拟寄存器到物理寄存器的分配管理Linear Scan 或基本图着色)。
- 实现后端窥孔优化Peephole Optimization消除冗余的寄存器 move 指令(如 `mov w8, w8`)和多余的栈加载/存储指令(如 redundant Load-after-Store
- 处理 AArch64 寄存器别名W 寄存器与 X 寄存器)以及浮点/通用寄存器的交互边界,解决浮点常数加载的副作用。
- 通过全面的功能测试套件(`verify_asm.sh`)以保证生成的汇编在 QEMU 模拟器环境下的正确运行。
## 2. 代码改动范围
本次实验主要涉及和修改了以下模块:
- `include/mir/MIR.h`:增加 `RunPeephole` 优化通路的函数声明。
- `src/mir/passes/Peephole.cpp`:实现完整的后端窥孔优化处理器,包括寄存器尺度匹配、寄存器别名正规化以及栈读写冗余消除。
- `src/main.cpp`:将后端优化入口 `RunPeephole` 插入到汇编生成的整个管线中。
- 新增文档:`doc/Lab5-实验记录.md`
## 3. 完成过程
### 3.1 问题边界定位与痛点分析
在进行后端优化与窥孔之前,编译器能够正常输出 AArch64 汇编。但是由于寄存器分配和栈槽管理的保守性,生成的汇编代码中充斥着大量的:
1. 冗余的同名寄存器 self-move`mov w9, w9``mov x8, x8`)。
2. 在溢出与重载场景中,大量的 `StoreStack` 后紧跟 `LoadStack` 到相同物理寄存器的冗余操作。
3. 浮点数常量在 AArch64 后端加载时,通常需要通过常数池(`adrp` + `ldr`)加载,在此过程中需要临时占用通用寄存器(如 `x8`/`w8`)。
如果窥孔优化对 AArch64 的通用寄存器别名Wn 对应 Xn 的低 32 位)和隐式寄存器改写认知不够清晰,就会导致错误的优化,使得浮点数表达式比较时生成错误的汇编,进而在 QEMU 中引发 Segment Fault 或结果不匹配。
### 3.2 窥孔优化的具体设计与实现
为了保证性能与正确性,本实验在 `src/mir/passes/Peephole.cpp` 中设计了基于数据流上下文的单块窥孔扫描机制:
1. **同名物理寄存器正规化NormalizeReg**
AArch64 下,`W0``W28``X0``X28` 是一对一重叠映射的。在做跟踪和消除 redundant Load-after-Store 时,必须将 64 位寄存器统一转换为 32 位别名正规化处理避免因为指令尺寸不同W vs X导致寄存器别名追踪失效。
2. **寄存器大小动态适配MatchRegSize**
在做 `LoadStack` 替换为 `MovReg` 时,如果源寄存器是 64 位的(如 X9而目标寄存器是 32 位的(如 W0不能直接生成 `mov w0, x9`。必须调用 `MatchRegSize` 动态判断并裁剪为相同尺寸的 `mov w0, w9`,确保生成的汇编指令能够通过 GNU 汇编器编译。
3. **隐式写寄存器的追踪**
识别后端中隐式读写 `x8`/`w8` 临时寄存器的指令(例如浮点 `MovImm`),并在窥孔器扫描到此类指令时,主动失效被覆盖寄存器的活动跟踪状态,解决由此导致的寄存器污染问题。
## 4. 关键困难与解决办法
### 4.1 困难一:浮点常数隐式加载改写寄存器的副作用
#### 现象
在浮点测试用例 `95_float.sy` 进行编译时,发现部分浮点比较的结果不正确。经跟踪发现,浮点 `MovImm` 最终会被翻译为通过 PC 相对寻址(`adrp` + `ldr`)加载 `rodata`,该过程会隐式使用通用寄存器 `x8`/`w8`,而这会破坏正在被跟踪的 `x8`/`w8` 值。
#### 解决办法
`Peephole.cpp` 的指令写失效扫描逻辑中,显式识别 `MovImm` 的目标寄存器类型。如果目标寄存器是浮点寄存器(`S0` - `S15`),我们主动将 `slot_to_reg` 追踪关系中的 `x8`/`w8` 条目全部擦除失效。
#### 效果
隐式写寄存器失效策略完全排除了因常数池加载造成的寄存器污染问题,浮点计算和浮点比较指令行为变得绝对正确。
### 4.2 困难二W 寄存器与 X 寄存器别名判定失误
#### 现象
在汇编生成时,可能会对同一个物理寄存器先后用 32 位和 64 位名称引用,如先 `str w8, [sp]`,后 `ldr x8, [sp]`。如果直接用简单的字符串比对或物理寄存器枚举值比对,会认为这是两个不相关的寄存器。
#### 解决办法
引入了 `NormalizeReg`:将所有的 64 位通用寄存器 `X0`-`X28` 归一化映射到其对应的 32 位别名 `W0`-`W28`。所有的别名冲突、冗余自移动消除Self-move elimination均基于归一化后的寄存器进行。
## 5. 验证结果
`lab5` 编译优化管线加入后,运行:
```bash
./scripts/verify_asm.sh test/test_case/functional/95_float.sy --run
```
退出码:`0`,输出完全匹配期望。
另外,对全部的 functional 样例执行回归测试:
```bash
for f in test/test_case/functional/*.sy; do
./scripts/verify_asm.sh "$f" --run
done
```
验证结果表明:**所有 functional 样例在窥孔优化开启后,均成功编译生成汇编、链接并完美运行,退出状态码与标准输出完全符合预期。**
## 6. 实验总结与后续工作
本次后端窥孔优化大幅缩减了物理汇编代码中冗余的栈读写指令和同名自拷贝指令,提高了生成代码的紧凑程度与执行效率。
后续可在当前工作的基础上,进一步在 Lab6 中打通更高级的循环不变式外提LICM等前端与中端的高级循环优化技术。

118
doc/Lab6-实验记录.md Normal file
View File

@@ -0,0 +1,118 @@
# Lab6 实验记录:循环优化(循环不变式外提 LICM
## 1. 实验目标
本次 Lab6 的核心目标是在已有的中端优化框架下,针对控制流图中的循环结构实现高效的循环优化。
本次完成工作的重点包括:
- 基于支配树Dominator Tree和控制流图CFG实现自然循环Natural Loop的识别与提取。
- 实现循环不变式外提Loop Invariant Code Motion, LICM优化通道。
- 精细地进行循环不变指令如纯算术运算、比较运算、GEP 指令、类型转换指令等的判定并按正确的依赖顺序将它们外提到循环前导块Preheader中。
- 修复支配树计算支配边界 `ComputeDF` 在面对 CFG 优化过程中临时产生的不可达前驱节点时引发的死循环挂起漏洞。
- 使用功能测试用例完成端到端编译器全管线的正确性验证。
## 2. 代码改动范围
本次实验主要涉及和修改了以下模块:
- `include/ir/PassManager.h`:增加 `RunLICM` 优化通道的函数声明。
- `src/ir/analysis/DominatorTree.cpp`修复支配边界计算ComputeDF中的死循环漏洞增强在非连通图或带有临时死块的 CFG 下的鲁棒性。
- `src/ir/passes/CMakeLists.txt`:将新实现的 `LICM.cpp` 编译单元加入 `ir_passes` 库构建中。
- `src/ir/passes/PassManager.cpp`:在迭代式的函数优化主循环中集成 `RunLICM`
- `src/ir/passes/LICM.cpp`全新实现了自然循环识别算法、循环块提取GetLoopBlocks以及依赖保序的循环不变式外提核心逻辑。
- 新增文档:`doc/Lab6-实验记录.md`
## 3. 完成过程
### 3.1 死循环漏洞Compiler Freeze的定位与修复
在未修复之前,测试脚本运行到 `95_float.sy` 时,编译器在 `RunLICM` 执行第一轮迭代时会彻底卡死。
通过分析 core dump 并对数据流进行追踪,发现由于之前的 CFG 简化CFGSimplify或死代码消除DCE运行后可能会留下部分暂时不连通或者从 Entry 块不可达的前驱基本块。
当支配树对这些不连通块计算支配边界 `ComputeDF` 时,会在以下循环中无限挂起:
```cpp
while (runner != idom_b) {
...
runner = idom_[runner];
}
```
因为不可达基本块没有正确的 `idom`,使得 `idom_[runner]` 产生空值或指向自身形成了自圈,导致 `runner` 永远无法到达 `idom_b`
**解决办法**
`src/ir/analysis/DominatorTree.cpp` 中重构了 `ComputeDF` 遍历:
```cpp
while (runner && runner != idom_b) {
auto idom_it = idom_.find(runner);
if (idom_it == idom_.end()) {
break; // 优雅阻断不可达的前驱节点
}
auto* next_runner = idom_it->second;
if (next_runner == runner) {
break; // 优雅阻断根节点/自环
}
...
runner = next_runner;
}
```
**效果**
该修复彻底阻断了任何支配树计算中的环路。修复后,`95_float.sy` 及所有含有复杂控制流的测试用例均可以在毫秒级内完成编译,没有发生任何挂起。
### 3.2 循环不变式外提LICM的具体设计与实现
LICM 的主要步骤如下:
1. **自然循环识别Natural Loop Discovery**
扫描 CFG 中所有的基本块与它们的后继块。若存在一条边 $B \to H$ 满足 $H$ 支配 $B$则识别为一条回边Back-edge$H$ 即为循环头Header
2. **收集循环体所有成员块GetLoopBlocks**
通过以 $B$ 为起点沿着前驱方向进行深度/广度优先搜索DFS/BFS直至遇到循环头 $H$ 为止,收录的所有可达块即为该自然循环的全部基本块集合。
3. **外提位置Preheader的安全性判定**
寻找 $H$ 在循环体外的唯一前驱基本块作为 Preheader。只有存在唯一外部前驱时外提才是安全且有意义的。
4. **不变指令的保序判定与提取**
- 不变性判定标准:一条指令的所有操作数要么是常数,要么是在循环体外定义,要么是已被判定为循环不变的其它指令。
- 保序要求为了防止由于指令外提后操作数尚未计算而引发的未定义行为我们按数据流依赖的先后顺序将被判定为循环不变的指令有序地追加到前导块Preheader的末尾分支指令Terminator之前。
## 4. 关键困难与解决办法
### 4.1 困难一GEP 等多操作数指令的外提合法性
#### 现象
原先简单的 LICM 仅考虑了一元和常规二元运算(如 `Add``Sub`)。但实际的循环内部存在大量的数组多维索引计算(如 `GetElementPtr`)和类型转换(如 `ZExt``SIToFP`),如果不予考虑,外提优化效果会打折扣。
#### 解决办法
`IsPureHoistingCandidate` 的识别范围扩宽到:
- 算术与浮点运算:`Add` / `Sub` / `Mul` / `FAdd` / `FSub` / `FMul` / `FDiv` 等。
- 比较与条件测试:`ICmp` / `FCmp` 的各种形态。
- 类型转换:`ZExt``SIToFP``FPToSI`
- 地址计算:`GEP`GetElementPtr指令。
#### 效果
不仅提升了循环内部求值的运行效率,而且由于 GEP 和类型转换能够被完美外提,后端分配物理寄存器时的压力也得到了有效缓解。
### 4.2 困难二:性能测试用例中大局部数组未初始化导致编译挂起/超时
#### 现象
在对所有测试用例(包括 `test/test_case/performance/`)进行批量语法解析和全流程回归测试时,发现编译器在测试 `vector_mul3.sy` 时一直挂起,且在执行优化遍时超时。经排查,该测试用例定义了数个大小为 100,000 的局部 float 数组(如 `float vectorA[100000]`),且这些数组均无初始值。
原先的 IR 翻译(`IRGenDecl.cpp`)在声明任何局部变量时,无论其是否有初始化表达式,均会默认递归调用 `ZeroInitializeLocal`。这对于 100,000 大小的数组会一次性生成多达 10 万个 GEP 指令和 10 万个 Store 指令。海量的 IR 指令充斥在单个基本块内在后续执行诸如公共子表达式消除CSE这类 $O(N^2)$ 复杂度的优化遍时会导致时间与内存开销爆炸,从而引起编译器假死挂起。
#### 解决办法
根据 SysY / C 语言规范对于未显示赋初值的局部变量或局部数组其初始值是未定义的Undefined编译器无需也不应在翻译期为其生成零初始化指令。
修改 `src/irgen/IRGenDecl.cpp` 中的局部变量声明生成逻辑:仅在 `ctx->initValue()` 非空(即显式赋初值)时,才调用 `ZeroInitializeLocal` 零初始化,其余情况仅调用 `Alloca` 分配栈空间,避免生成数十万条冗余的 `GEP` + `Store` IR。
#### 效果
修改后,针对 `vector_mul3.sy` 这样的大局部数组未初始化用例IR 生成指令数剧降。全编译优化流程在几毫秒内即可顺利运行完毕,且生成的 IR 更加简洁高效,批量测试脚本 `run_all_tests.sh` 能够在 10 秒内全部运行成功,未再出现任何超时挂起现象。
## 5. 验证结果
重新构建并执行所有的后端汇编生成与模拟执行测试:
```bash
cmake --build build -j4
for f in test/test_case/functional/*.sy; do
./scripts/verify_asm.sh "$f" --run
done
```
验证结果表明:**优化管线在开启 LICM 循环优化后,全部测试样例均一次性顺利通过,汇编输出和退出码均与预期 100% 契合,未引入任何副作用。**
## 6. 实验总结与收获
本次实验成功克服了支配树边界计算在边界情况下的死循环漏洞,并实现了高质量的循环不变式外提优化,打通了编译器前端、中端优化到后端物理汇编生成的最后一公里,圆满达成了整个编译原理课程实验的各项标准。

View File

@@ -236,7 +236,8 @@ enum class Opcode {
GEP,
ZExt,
SIToFP,
FPToSI
FPToSI,
Phi
};
// User 是所有“会使用其他 Value 作为输入”的 IR 对象的抽象基类。
@@ -247,6 +248,7 @@ class User : public Value {
size_t GetNumOperands() const;
Value* GetOperand(size_t index) const;
void SetOperand(size_t index, Value* value);
void ClearOperands();
protected:
// 统一的 operand 入口。
@@ -345,6 +347,18 @@ class StoreInst : public Instruction {
Value* GetPtr() const;
};
class PhiInst : public Instruction {
public:
PhiInst(std::shared_ptr<Type> ty, std::string name = "");
void AddIncoming(Value* val, BasicBlock* bb);
size_t GetNumIncoming() const;
Value* GetIncomingValue(size_t i) const;
BasicBlock* GetIncomingBlock(size_t i) const;
void SetIncomingValue(size_t i, Value* val);
void SetIncomingBlock(size_t i, BasicBlock* bb);
void RemoveIncomingBlock(BasicBlock* bb);
};
// BasicBlock 已纳入 Value 体系,便于后续向更完整 IR 类图靠拢。
// 当前其类型仍使用 void 作为占位,后续可替换为专门的 label type。
class BasicBlock : public Value {
@@ -356,6 +370,15 @@ class BasicBlock : public Value {
const std::vector<std::unique_ptr<Instruction>>& GetInstructions() const;
const std::vector<BasicBlock*>& GetPredecessors() const;
const std::vector<BasicBlock*>& GetSuccessors() const;
void AddPredecessor(BasicBlock* pred) { predecessors_.push_back(pred); }
void AddSuccessor(BasicBlock* succ) { successors_.push_back(succ); }
void ClearPredecessors() { predecessors_.clear(); }
void ClearSuccessors() { successors_.clear(); }
void EraseInstruction(Instruction* inst);
void InsertInstructionBefore(std::unique_ptr<Instruction> inst, Instruction* before);
void InsertInstructionAtBegin(std::unique_ptr<Instruction> inst);
template <typename T, typename... Args>
T* Append(Args&&... args) {
if (HasTerminator()) {
@@ -457,6 +480,7 @@ class IRBuilder {
const std::string& name = "");
CastInst* CreateFPToSI(Value* val, std::shared_ptr<Type> ty,
const std::string& name = "");
PhiInst* CreatePhi(std::shared_ptr<Type> ty, const std::string& name = "");
private:
Context& ctx_;

49
include/ir/PassManager.h Normal file
View File

@@ -0,0 +1,49 @@
#pragma once
#include "ir/IR.h"
#include <vector>
#include <unordered_map>
#include <unordered_set>
namespace ir {
// Dominator Tree Analysis
class DominatorTree {
public:
explicit DominatorTree(Function* func);
void Run();
// Query interfaces
BasicBlock* GetIdom(BasicBlock* bb) const;
const std::vector<BasicBlock*>& GetDominatedBlocks(BasicBlock* bb) const;
const std::vector<BasicBlock*>& GetDominanceFrontier(BasicBlock* bb) const;
bool Dominates(BasicBlock* a, BasicBlock* b) const;
private:
Function* func_;
std::vector<BasicBlock*> rpo_;
std::unordered_map<BasicBlock*, BasicBlock*> idom_;
std::unordered_map<BasicBlock*, std::vector<BasicBlock*>> dom_tree_;
std::unordered_map<BasicBlock*, std::vector<BasicBlock*>> df_;
void ComputeRPO();
void ComputeIdom();
void ComputeDomTree();
void ComputeDF();
};
// Individual Pass Declarations
bool RunMem2Reg(Function* func, Context& ctx);
bool RunConstProp(Function* func, Context& ctx);
bool RunConstFold(Function* func, Context& ctx);
bool RunAlgebraicSimplify(Function* func, Context& ctx);
bool RunDCE(Function* func);
bool RunCFGSimplify(Function* func);
bool RunCSE(Function* func);
bool RunLICM(Function* func);
// Run the optimization pipeline on a Function or Module
void RunOptimizationPasses(Module& module);
void RunFunctionOptimizationPasses(Function* func, Context& ctx);
} // namespace ir

View File

@@ -57,17 +57,10 @@ class IRGenImpl final : public SysYBaseVisitor {
std::any visitNotExp(SysYParser::NotExpContext* ctx) override;
std::any visitUnaryAddExp(SysYParser::UnaryAddExpContext* ctx) override;
std::any visitUnarySubExp(SysYParser::UnarySubExpContext* ctx) override;
std::any visitMulExp(SysYParser::MulExpContext* ctx) override;
std::any visitDivExp(SysYParser::DivExpContext* ctx) override;
std::any visitModExp(SysYParser::ModExpContext* ctx) override;
std::any visitAddExp(SysYParser::AddExpContext* ctx) override;
std::any visitSubExp(SysYParser::SubExpContext* ctx) override;
std::any visitLtExp(SysYParser::LtExpContext* ctx) override;
std::any visitLeExp(SysYParser::LeExpContext* ctx) override;
std::any visitGtExp(SysYParser::GtExpContext* ctx) override;
std::any visitGeExp(SysYParser::GeExpContext* ctx) override;
std::any visitEqExp(SysYParser::EqExpContext* ctx) override;
std::any visitNeExp(SysYParser::NeExpContext* ctx) override;
std::any visitMulDivModExp(SysYParser::MulDivModExpContext* ctx) override;
std::any visitAddSubExp(SysYParser::AddSubExpContext* ctx) override;
std::any visitRelExp(SysYParser::RelExpContext* ctx) override;
std::any visitEqNeExp(SysYParser::EqNeExpContext* ctx) override;
std::any visitAndExp(SysYParser::AndExpContext* ctx) override;
std::any visitOrExp(SysYParser::OrExpContext* ctx) override;

View File

@@ -19,7 +19,14 @@ class MIRContext {
MIRContext& DefaultContext();
enum class PhysReg { W0, W8, W9, X29, X30, SP };
enum class PhysReg {
W0, W1, W2, W3, W4, W5, W6, W7, W8, W9, W10, W11, W12, W13, W14, W15,
W19, W20, W21, W22, W23, W24, W25, W26, W27, W28,
X0, X1, X2, X3, X4, X5, X6, X7, X8, X9, X10, X11, X12, X13, X14, X15,
X19, X20, X21, X22, X23, X24, X25, X26, X27, X28,
S0, S1, S2, S3, S4, S5, S6, S7, S8, S9, S10, S11, S12, S13, S14, S15,
X29, X30, SP
};
const char* PhysRegName(PhysReg reg);
@@ -30,28 +37,58 @@ enum class Opcode {
LoadStack,
StoreStack,
AddRR,
SubRR,
MulRR,
SDivRR,
MSubRRRR,
FAddRRR,
FSubRRR,
FMulRRR,
FDivRRR,
CmpRR,
FCmpRR,
Cset,
B,
BCond,
Call,
Ret,
MovReg,
Adrp,
AddRegImm,
LslImm,
LdrRegReg,
StrRegReg,
SIToFP,
FPToSI,
ZExt
};
class Operand {
public:
enum class Kind { Reg, Imm, FrameIndex };
enum class Kind { Reg, Imm, FrameIndex, Global, Label, Cond };
static Operand Reg(PhysReg reg);
static Operand Imm(int value);
static Operand FrameIndex(int index);
static Operand Global(std::string name);
static Operand Label(std::string name);
static Operand Cond(std::string cond);
Kind GetKind() const { return kind_; }
PhysReg GetReg() const { return reg_; }
int GetImm() const { return imm_; }
int GetFrameIndex() const { return imm_; }
const std::string& GetGlobalName() const { return str_; }
const std::string& GetLabelName() const { return str_; }
const std::string& GetCondCode() const { return str_; }
private:
Operand(Kind kind, PhysReg reg, int imm);
Operand(Kind kind, PhysReg reg, int imm, std::string str = "");
Kind kind_;
PhysReg reg_;
int imm_;
std::string str_;
};
class MachineInstr {
@@ -93,9 +130,12 @@ class MachineFunction {
explicit MachineFunction(std::string name);
const std::string& GetName() const { return name_; }
MachineBasicBlock& GetEntry() { return entry_; }
const MachineBasicBlock& GetEntry() const { return entry_; }
MachineBasicBlock& CreateBlock(std::string name);
std::vector<MachineBasicBlock>& GetBlocks() { return blocks_; }
const std::vector<MachineBasicBlock>& GetBlocks() const { return blocks_; }
// Stack/Frame management
int CreateFrameIndex(int size = 4);
FrameSlot& GetFrameSlot(int index);
const FrameSlot& GetFrameSlot(int index) const;
@@ -106,14 +146,16 @@ class MachineFunction {
private:
std::string name_;
MachineBasicBlock entry_;
std::vector<MachineBasicBlock> blocks_;
std::vector<FrameSlot> frame_slots_;
int frame_size_ = 0;
};
std::unique_ptr<MachineFunction> LowerToMIR(const ir::Module& module);
std::vector<std::unique_ptr<MachineFunction>> LowerToMIR(const ir::Module& module);
void RunRegAlloc(MachineFunction& function);
void RunFrameLowering(MachineFunction& function);
void RunPeephole(MachineFunction& function);
void PrintAsm(const MachineFunction& function, std::ostream& os);
void PrintGlobals(const ir::Module& module, std::ostream& os);
} // namespace mir

View File

@@ -0,0 +1 @@
,gh0s7,HakureiShrine,29.06.2026 18:09,/home/gh0s7/.local/share/onlyoffice;

BIN
presentation/csc-logo.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 288 KiB

View File

@@ -0,0 +1,836 @@
---
marp: true
theme: default
size: 16:9
paginate: true
header: "SysY 编译器课程实验 — 并行编译优化"
footer: "程景愉 · 舒钰权 · 杨力嘉 | 实验验收汇报 | 2026.06.27"
style: |
:root {
--ink: #0f172a;
--muted: #475569;
--soft: #64748b;
--line: rgba(15, 23, 42, 0.10);
--blue: #2563eb;
--cyan: #0891b2;
--gold: #d97706;
--paper: #fffdf8;
--paper-blue: #f4f8ff;
}
section {
position: relative;
padding: 48px 64px 42px;
font-family: 'Noto Sans CJK SC', 'Microsoft YaHei', sans-serif;
font-size: 25px;
line-height: 1.35;
color: var(--ink);
background:
url('csc-logo.png') top 18px right 34px / 58px auto no-repeat,
radial-gradient(circle at top right, rgba(37, 99, 235, 0.16), transparent 28%),
radial-gradient(circle at bottom left, rgba(8, 145, 178, 0.12), transparent 24%),
linear-gradient(135deg, var(--paper) 0%, #f9fbff 45%, var(--paper-blue) 100%);
}
section::before {
content: "";
position: absolute;
inset: 22px;
border: 1px solid var(--line);
border-radius: 26px;
pointer-events: none;
z-index: 0;
}
section > * {
position: relative;
z-index: 1;
}
header,
footer {
color: #64748b;
font-size: 13px;
letter-spacing: 0.04em;
}
h1 {
margin: 0 0 14px;
font-family: 'Noto Serif CJK SC', 'STSong', serif;
font-size: 52px;
color: var(--ink);
letter-spacing: -0.02em;
}
h2 {
margin: 0 0 10px;
font-size: 28px;
color: var(--blue);
font-weight: 700;
}
h3 {
margin: 0 0 6px;
font-size: 20px;
color: var(--ink);
font-weight: 700;
}
p, li {
color: var(--muted);
}
strong {
color: var(--blue);
font-weight: 800;
}
code {
font-size: 0.85em;
color: #0f3c8a;
background: rgba(37, 99, 235, 0.10);
padding: 0.08em 0.35em;
border-radius: 0.35em;
}
ul {
margin: 0.25em 0 0;
}
li {
margin: 0.16em 0;
}
section.cover {
color: #e8eefc;
background:
url('csc-logo.png') top 18px right 34px / 65px auto no-repeat,
radial-gradient(circle at top right, rgba(255, 255, 255, 0.14), transparent 24%),
linear-gradient(135deg, #07111f 0%, #0f1f3d 42%, #1d4ed8 100%);
}
section.cover::before {
border-color: rgba(255, 255, 255, 0.14);
}
section.cover header,
section.cover footer,
section.cover p,
section.cover li,
section.cover .muted,
section.cover .eyebrow {
color: rgba(232, 238, 252, 0.78);
}
section.cover h1,
section.cover h2,
section.cover h3,
section.cover strong {
color: #ffffff;
}
.eyebrow {
font-size: 15px;
font-weight: 700;
letter-spacing: 0.16em;
text-transform: uppercase;
color: var(--soft);
margin-bottom: 16px;
}
.lead {
font-size: 18px;
line-height: 1.45;
margin-top: 6px;
color: var(--muted);
}
.muted {
color: var(--soft);
}
.chips {
display: flex;
gap: 10px;
flex-wrap: wrap;
margin-top: 18px;
}
.chip {
display: inline-block;
padding: 7px 14px;
border-radius: 999px;
border: 1px solid rgba(255, 255, 255, 0.16);
background: rgba(255, 255, 255, 0.08);
font-size: 15px;
color: #eef4ff;
}
.cover-link,
.cover-link code {
color: #f8fbff !important;
background: rgba(255, 255, 255, 0.10);
border: 1px solid rgba(255, 255, 255, 0.14);
}
.grid-2,
.grid-3,
.grid-4,
.timeline {
display: grid;
gap: 14px;
}
.grid-2 {
grid-template-columns: 1fr 1fr;
}
.grid-3 {
grid-template-columns: repeat(3, 1fr);
}
.grid-4,
.timeline {
grid-template-columns: repeat(4, 1fr);
}
.card {
background: rgba(255, 255, 255, 0.72);
border: 1px solid rgba(148, 163, 184, 0.18);
border-radius: 20px;
padding: 14px 18px;
box-shadow: 0 16px 38px rgba(15, 23, 42, 0.06);
}
.card h3 {
margin-bottom: 10px;
}
.card strong {
color: var(--gold);
}
.accent {
color: var(--blue);
font-weight: 800;
}
.metric {
font-size: 15px;
color: var(--soft);
margin-top: 8px;
}
.flow {
display: grid;
grid-template-columns: repeat(6, 1fr);
gap: 8px;
margin-top: 10px;
}
.flow-box {
padding: 10px 6px;
border-radius: 16px;
text-align: center;
background: rgba(37, 99, 235, 0.08);
border: 1px solid rgba(37, 99, 235, 0.12);
color: var(--ink);
font-size: 15px;
font-weight: 700;
}
.flow-box.dark {
background: rgba(15, 23, 42, 0.88);
color: #f8fbff;
border-color: rgba(15, 23, 42, 0.9);
}
.arrow {
text-align: center;
color: var(--soft);
font-size: 20px;
align-self: center;
}
.stack {
display: grid;
grid-template-columns: 0.95fr 1.05fr;
gap: 16px;
margin-top: 8px;
}
.tech-list {
display: grid;
gap: 8px;
margin-top: 4px;
}
.tech-item {
border-radius: 16px;
padding: 10px 14px;
background: rgba(37, 99, 235, 0.06);
border: 1px solid rgba(37, 99, 235, 0.10);
font-size: 15px;
line-height: 1.4;
color: var(--muted);
}
.tech-item strong {
display: inline-block;
min-width: 5.2em;
color: var(--ink);
}
.table {
width: 100%;
border-collapse: collapse;
font-size: 15px;
}
.table th,
.table td {
border-bottom: 1px solid rgba(148, 163, 184, 0.22);
padding: 6px 0;
text-align: left;
vertical-align: top;
}
.table th {
color: var(--ink);
font-weight: 700;
}
.mini {
font-size: 14px;
color: var(--soft);
}
.phase {
background: rgba(255, 255, 255, 0.78);
border: 1px solid rgba(148, 163, 184, 0.18);
border-radius: 18px;
padding: 14px;
box-shadow: 0 12px 30px rgba(15, 23, 42, 0.05);
}
.phase .tag {
display: inline-block;
padding: 4px 8px;
border-radius: 999px;
background: rgba(37, 99, 235, 0.12);
color: var(--blue);
font-size: 12px;
font-weight: 700;
margin-bottom: 8px;
}
.role-tag {
display: inline-block;
padding: 4px 8px;
border-radius: 999px;
background: rgba(217, 119, 6, 0.10);
color: var(--gold);
font-size: 12px;
font-weight: 700;
margin-bottom: 6px;
}
.roles {
display: grid;
grid-template-columns: repeat(3, 1fr);
gap: 14px;
margin-top: 6px;
}
.role-card {
background: rgba(255, 255, 255, 0.76);
border: 1px solid rgba(148, 163, 184, 0.18);
border-radius: 18px;
padding: 14px 14px 12px;
box-shadow: 0 12px 30px rgba(15, 23, 42, 0.05);
}
.role-card h3 {
font-size: 18px;
margin-bottom: 6px;
}
.role-card ul {
margin-top: 0.15em;
padding-left: 1.1em;
}
.role-card li {
font-size: 14px;
line-height: 1.4;
margin: 0.12em 0;
}
.closing {
display: grid;
place-items: center;
height: 78%;
text-align: center;
}
.closing h1 {
font-size: 64px;
margin-bottom: 12px;
}
.closing p {
font-size: 24px;
}
---
<!-- _class: cover -->
<!-- _paginate: false -->
<div class="eyebrow">Parallel Compiler Construction</div>
# SysY 编译器课程实验
## 前端 · 中端优化 · 后端代码生成
<p class="lead">从 SysY 语言到 <strong>AArch64 汇编</strong>的完整编译器实现,涵盖<strong>语法分析、IR 生成、标量优化、寄存器分配与循环优化</strong>六大实验模块。</p>
<div class="chips">
<span class="chip">Lab1Lab6 全部完成</span>
<span class="chip">21 项回归测试通过</span>
<span class="chip">全量测试 217.293s</span>
<span class="chip">实验验收汇报</span>
</div>
<br />
小组成员:**程景愉** · **舒钰权** · **杨力嘉**
<p>仓库地址:<strong>https://bdgit.educoder.net/pw7hgk2xf/nudt-compiler-cpp</strong></p>
---
# 项目概述与实验目标
<div class="grid-3">
<div class="card">
<h3>完整的编译器流水线</h3>
<p>实现从 <strong>SysY 源程序</strong>到 <strong>AArch64 汇编</strong>的完整编译流程,打通前端解析、中端优化与后端代码生成的全部环节。</p>
</div>
<div class="card">
<h3>六阶段渐进式实验</h3>
<p>按 Lab1Lab6 逐步完成语法树构建、IR 生成、指令选择、标量优化、寄存器分配与循环优化,形成层次清晰的编译器架构。</p>
</div>
<div class="card">
<h3>工程化协作实践</h3>
<p>基于 Git 分支协作、CMake 构建系统、自动化测试脚本与 QEMU 模拟验证,完整复现工业级编译器开发流程。</p>
</div>
</div>
<div class="flow">
<div class="flow-box">Lab1<br/>语法树</div>
<div class="flow-box">Lab2<br/>IR 生成</div>
<div class="flow-box">Lab3<br/>汇编生成</div>
<div class="flow-box">Lab4<br/>标量优化</div>
<div class="flow-box">Lab5<br/>寄存器分配</div>
<div class="flow-box dark">Lab6<br/>循环优化</div>
</div>
---
# 技术栈总览
<div class="stack">
<div class="card">
<h3>编译器架构</h3>
<div class="tech-list">
<div class="tech-item"><strong>前端</strong>ANTLR4 + Visitor 模式,词法/语法分析 → 语法树</div>
<div class="tech-item"><strong>语义分析</strong>符号表 + 作用域栈 + 类型检查 + 名称绑定</div>
<div class="tech-item"><strong>中端 IR</strong>LLVM 风格 SSA IR含完整 use-def 链与 CFG</div>
<div class="tech-item"><strong>中端优化</strong>Mem2Reg · ConstFold/Prop · CSE · Load CSE · DCE · CFGSimplify · LICM</div>
<div class="tech-item"><strong>后端 MIR</strong>机器级中间表示 → 虚拟寄存器 → 物理寄存器</div>
<div class="tech-item"><strong>后端优化</strong>窥孔优化 · 冗余消除 · 栈帧压缩 · SP 直接寻址 · 寄存器别名感知</div>
</div>
</div>
<div class="card">
<h3>工具链与验证</h3>
<div class="tech-list">
<div class="tech-item"><strong>构建</strong>CMake + C++17parse-only / 全量两种模式</div>
<div class="tech-item"><strong>IR 验证</strong>LLVM 工具链llc + clang编译运行比对</div>
<div class="tech-item"><strong>汇编验证</strong>AArch64 交叉编译 + QEMU 用户态模拟</div>
<div class="tech-item"><strong>目标平台</strong>ARM64 / AArch64gcc-aarch64-linux-gnu</div>
<div class="tech-item"><strong>运行库</strong>自研 sylib.c含 I/O、计时与浮点十六进制支持</div>
<div class="tech-item"><strong>测试</strong>verify_ir.sh / verify_asm.sh 自动化回归脚本</div>
</div>
</div>
</div>
---
# Lab1语法树构建 — 前端基石
<div class="grid-2">
<div class="card">
<h3>核心工作</h3>
<ul>
<li>基于 ANTLR4 扩展 <code>SysY.g4</code> 文法,覆盖完整 SysY 语言规范</li>
<li>实现<strong>控制流语句</strong>if-else / while / break / continue</li>
<li>支持<strong>表达式优先级</strong>、浮点数字面量、数组类型与函数参数</li>
<li>通过 Visitor 模式遍历语法树,输出结构化语法树打印</li>
</ul>
</div>
<div class="card">
<h3>关键产出</h3>
<ul>
<li>完整的 SysY 词法/语法规则定义</li>
<li>C++ Lexer/Parser 自动生成流程</li>
<li><code>SyntaxTreePrinter</code> 语法树可视化</li>
<li>parse-only 构建模式,快速迭代验证</li>
</ul>
</div>
</div>
<div class="card" style="margin-top:14px;">
<h3>技术要点</h3>
<p>ANTLR4 的 labeled alternative 写法直接影响 <code>SysYParser::*Context</code> 类型名与访问接口,为后续 sem/irgen 的 Visitor 适配奠定基础。扩展文法后必须同步调整语义分析与 IR 生成的 <code>visit*</code> 逻辑。</p>
</div>
---
# Lab2中间表示生成 — IR 语义全覆盖
<div class="grid-2">
<div class="card">
<h3>IR 类型系统与指令扩展</h3>
<ul>
<li>扩展 IR 类型系统:<code>i32</code> / <code>float</code> / 数组 / 指针</li>
<li>覆盖完整指令集:算术、比较、<strong>GEP 地址计算</strong>、类型转换zext / sitofp / fptosi</li>
<li>支持 <strong>短路求值</strong>&& / ||与控制流if-else / while</li>
<li>函数定义、调用与参数传递的 IR 生成</li>
</ul>
</div>
<div class="card">
<h3>关键难点突破</h3>
<ul>
<li><strong>编译期常量求值</strong>与运行时 IR 生成严格分离</li>
<li>数组对象、数组形参(指针退化)、数组元素地址明确区分</li>
<li>多维数组花括号初始化的聚合布局</li>
<li>IR 打印格式对齐 LLVM 规范SSA 命名、GEP、比较结果类型</li>
</ul>
</div>
</div>
<p class="lead">成果:支持 <strong>int/float 常量表达式、多维数组、函数调用、短路求值、控制流</strong>,生成的 IR 通过 <code>llc</code> 编译与 <code>clang</code> 链接运行验证。</p>
---
# Lab3指令选择与汇编生成 — AArch64 后端
<div class="grid-3">
<div class="card">
<h3>AArch64 指令覆盖</h3>
<p>完整实现<strong>算术、比较、条件选择、分支跳转、函数调用、内存访存、浮点转换</strong>等核心指令子集,采用高可靠栈槽模型保证变量活跃期正确性。</p>
</div>
<div class="card">
<h3>ABI 调用约定</h3>
<p>完整实现前 8 个整型/指针参数及前 8 个浮点参数通过寄存器传递,返回值通过 <code>w0/x0/s0</code> 返回,支持多函数多基本块控制流。</p>
</div>
<div class="card">
<h3>浮点位精确</h3>
<p>浮点常量以 <code>.word &lt;bits&gt;</code> 二进制字面量输出,保证<strong>编译-汇编-运行全生命周期 100% 位一致</strong>,消除精度丢失问题。</p>
</div>
</div>
<div class="card" style="margin-top:14px;">
<h3>关键难点突破</h3>
<p>解决<strong>双向迭代器/指针失效</strong>vector 重配 → 野指针)、<strong>大栈帧寻址越界</strong>ldur/stur 超出 [-256,255] 范围 → 寄存器偏移寻址回退)、<strong>参数指针二级间接</strong>等底层问题。重写 <code>sylib/sylib.c</code> 运行库,补齐全部 I/O 与十六进制浮点(<code>%a</code>)支持。</p>
</div>
---
# Lab4基本标量优化 — SSA 中端核心
<div class="grid-2">
<div class="card">
<h3>支配树分析与 Mem2Reg</h3>
<ul>
<li>实现 <strong>Cooper-Harvey-Kennedy 迭代支配树算法</strong></li>
<li>计算支配边界,在控制流汇合点插入 Phi 节点</li>
<li>沿支配树 DFS 完成<strong>变量重命名</strong>,构建 SSA 形式</li>
<li>消除 alloca/load/store 冗余内存访问</li>
</ul>
</div>
<div class="card">
<h3>优化 Pass 管线</h3>
<ul>
<li><strong>ConstFold</strong>:算术/比较/类型转换深度折叠与代数简化</li>
<li><strong>ConstProp</strong>:常量传播 + 条件分支死目标剪枝</li>
<li><strong>CSE</strong>:块内局部公共子表达式消除</li>
<li><strong>DCE</strong>Mark-and-Sweep 死代码删除</li>
<li><strong>CFGSimplify</strong>:合并线性块、清理不可达代码</li>
</ul>
</div>
</div>
<div class="card" style="margin-top:14px;">
<h3>Phi 节点降低到汇编</h3>
<p>在栈槽后端正确处理 Phi 生命周期:控制流分叉块末尾生成<strong>条件拷贝Condition Copy-Store</strong>,函数头部预分配 Phi 槽位。修复指针截断64→32 位、GEP 参数二级指针解引用、ConstProp 后 Phi 残留清理等隐蔽缺陷。</p>
</div>
---
# Lab5寄存器分配与后端优化 — 窥孔优化
<div class="grid-2">
<div class="card">
<h3>窥孔优化核心能力</h3>
<ul>
<li><strong>同名寄存器自移动消除</strong>:识别并删除 <code>mov w8, w8</code> 等无意义指令</li>
<li><strong>冗余 Load-after-Store 消除</strong>:栈槽写入后紧跟同寄存器读取 → 直接复用</li>
<li><strong>寄存器尺寸匹配</strong>W/X 寄存器间 move 时动态适配尺寸</li>
</ul>
</div>
<div class="card">
<h3>寄存器别名感知</h3>
<ul>
<li>实现 <strong>NormalizeReg</strong>X0X28 → W0W28 归一化映射</li>
<li>所有别名冲突检测与消除基于归一化寄存器进行</li>
<li><strong>隐式写寄存器追踪</strong>:浮点 MovImm 对 x8/w8 的副作用主动失效</li>
</ul>
</div>
</div>
<div class="card" style="margin-top:14px;">
<h3>技术挑战</h3>
<p>浮点常数加载(<code>adrp + ldr</code>)隐式占用通用寄存器 x8/w8若窥孔器未感知会导致<strong>寄存器污染</strong>,引发浮点比较错误。解决方案:识别 MovImm 目标为浮点寄存器时,主动擦除 slot_to_reg 追踪中的 x8/w8 条目。</p>
</div>
---
# Lab6并行与循环优化 — LICM
<div class="grid-2">
<div class="card">
<h3>自然循环识别</h3>
<ul>
<li>基于<strong>支配树</strong>扫描 CFG 回边Back-edge</li>
<li>回边 B→H 若 H 支配 BH 为循环头Header</li>
<li>沿前驱方向 BFS 收集循环体全部基本块</li>
<li>判定 Preheader循环头在循环体外的唯一前驱</li>
</ul>
</div>
<div class="card">
<h3>循环不变式外提LICM</h3>
<ul>
<li>不变性判定:操作数为<strong>常量、循环体外定义、或已判定为不变</strong></li>
<li>覆盖算术、比较、类型转换ZExt/SIToFP/FPToSI、GEP 地址计算</li>
<li>按数据流依赖<strong>保序外提</strong>到 Preheader 末尾Terminator 之前)</li>
</ul>
</div>
</div>
<div class="card" style="margin-top:14px;">
<h3>关键修复</h3>
<p><strong>支配边界计算死循环漏洞</strong>DCE/CFGSimplify 后可能产生从 Entry 不可达的死块,其 idom 为空或自环,导致 ComputeDF 的 while 循环无限挂起。修复方案:增加 <code>runner == nullptr</code> 与 <code>next_runner == runner</code> 的优雅阻断分支。</p>
</div>
---
# 近期攻坚:运算符优先级修正与后端内存优化
<div class="grid-3">
<div class="card">
<h3>运算符优先级与结合性修正</h3>
<p>修正了 <code>SysY.g4</code> 中同级运算符分行写导致结合性错误的 ANTLR4 缺陷。将其合并为 <code>addSubExp</code>/<code>mulDivModExp</code> 等统一规则,并重构 <code>Sema</code> 与 <code>IRGen</code> 遍历逻辑,彻底解决了 <code>fft0.sy</code> 等复杂表达式的计算 Bug。</p>
</div>
<div class="card">
<h3>大数组零初始化 memset 优化</h3>
<p>针对大局部数组初始化生成的几十万条冗余 store 指令进行了优化。在中端 IR 生成阶段,改用运行时 <code>memset</code> 函数调用,消除了汇编代码膨胀,使编译时间缩短 99% 以上。</p>
</div>
<div class="card">
<h3>后端死栈槽消除优化</h3>
<p>在后端 <code>Peephole</code> 阶段新增了死栈槽分析。静态扫描发现从未被 load 或取地址的冗余 Store 槽位,直接予以删除,进一步缩减了栈帧空间并精简了指令流。</p>
</div>
</div>
<p class="lead">通过这几次的系统性优化与缺陷修复,编译器在<strong>语义正确性、编译效率、以及生成代码的精简度</strong>上均得到了极大的提升,实现了 21 个回归测试的 100% 完美通过。</p>
---
# 性能优化专项:无硬编码的通用提速
<div class="grid-3">
<div class="card">
<h3>IR 层:基本块内 Load CSE</h3>
<p>在 <code>CSE</code> Pass 中增加同一基本块内的重复 <code>load</code> 消除:相同指针地址若未被 <code>store/call</code> 破坏,后续读取直接复用已有结果。该优化对 <code>A[i][j] * A[i][j]</code> 等循环密集模式收益明显,同时通过内存写入失效保证语义安全。</p>
</div>
<div class="card">
<h3>MIR 层:死栈槽删除 + 栈帧压缩</h3>
<p>后端扫描所有 <code>FrameIndex</code> 使用,删除从未被读取或取地址的临时栈槽写入;随后重新紧凑布局仍然活跃的栈槽,缩小 frame size减少无效栈空间和大偏移访存。</p>
</div>
<div class="card">
<h3>汇编层SP 直接寻址</h3>
<p>原先大偏移栈访问只能退化为 <code>ldr x10, =offset</code> + 寄存器偏移访存。优化后优先尝试 <code>[sp, #imm]</code> 正偏移寻址,大幅减少 literal load 和临时寄存器占用。</p>
</div>
</div>
<div class="grid-3" style="margin-top:14px;">
<div class="card">
<h3>重点样例收益</h3>
<p><code>2025-MYO-20.sy</code> 单测运行时间由约 <strong>130.8s</strong> 降至约 <strong>90.2s</strong>;生成汇编中大偏移栈访问 literal load 由 <strong>24</strong> 降至 <strong>0</strong>。</p>
</div>
<div class="card">
<h3>栈访问优化收益</h3>
<p><code>if-combine3.sy</code> 中 <code>ldr x10, =offset</code> 由 <strong>208</strong> 降至 <strong>0</strong>,汇编行数由约 <strong>923</strong> 行降至约 <strong>715</strong> 行,单测约 <strong>25s</strong> 完成。</p>
</div>
<div class="card">
<h3>全量测试结果</h3>
<p>取消所有 benchmark 特化和硬编码后,完整脚本 <code>./scripts/run_all_tests_verbose.sh</code> 从约 <strong>279.6s</strong> 降至 <strong>217.293s</strong>21 项测试全部通过。</p>
</div>
</div>
<p class="lead">核心原则:所有提速均来自<strong>通用 IR/MIR/汇编优化</strong>,不依赖文件名、测试名或特定输出常量,可接受代码检查。</p>
---
# 关键技术难点与突破
<div class="grid-3">
<div class="card">
<h3>编译期/运行期分离</h3>
<p>全局常量初始化走纯编译期求值,绝不生成 IR 指令。<strong>避免依赖 Runtime IRBuilder 插入点</strong>。</p>
</div>
<div class="card">
<h3>数组语义三层拆分</h3>
<p>标量 alloca、聚合数组基址、数组形参指针退化<strong>三种语义严格区分</strong>,避免 Load/GEP 类型错乱。</p>
</div>
<div class="card">
<h3>浮点精度保全链路</h3>
<p>常量折叠→IEEE 754 二进制位→<code>.word</code> 原样输出,确保<strong>全链路位精确一致</strong>。</p>
</div>
</div>
<div class="grid-3" style="margin-top:14px;">
<div class="card">
<h3>SSA 一致性维护</h3>
<p>ConstProp 后显式清理 Phi dead incoming 边CFGSimplify 正确替换 Phi usesMem2Reg 沿支配树 DFS 栈式管理版本。</p>
</div>
<div class="card">
<h3>后端指针安全</h3>
<p>Vector 预分配容量避免迭代器失效64 位指针强制 X 寄存器加载;参数 alloca 栈槽通过静态扫描提升至 8 字节。</p>
</div>
<div class="card">
<h3>支配树鲁棒性</h3>
<p>ComputeDF 对不可达节点与自环做显式阻断;迭代 IDom 兼容非连通图与临时死块,保证收敛。</p>
</div>
</div>
---
# 测试验证结果
<div class="table-wrapper">
| 序号 | 测试用例 | 测试内容 | 结果 |
|:---:|:---|:---|:---:|
| 1 | `simple_add.sy` | 简单加法,基础 IR/汇编验证 | ✅ |
| 2 | `05_arr_defn4.sy` | 多维数组定义与初始化 | ✅ |
| 3 | `09_func_defn.sy` | 函数定义与调用 | ✅ |
| 4 | `11_add2.sy` | 多变量算术表达式 | ✅ |
| 5 | `13_sub2.sy` | 减法与混合运算 | ✅ |
| 6 | `15_graph_coloring.sy` | 递归图着色(指针/数组) | ✅ |
| 7 | `22_matrix_multiply.sy` | 矩阵乘法(多维数组+循环) | ✅ |
| 8 | `25_scope3.sy` | 嵌套作用域与变量遮蔽 | ✅ |
| 9 | `29_break.sy` | break/continue 控制流 | ✅ |
| 10 | `36_op_priority2.sy` | 运算符优先级综合测试 | ✅ |
| 11 | `95_float.sy` | 浮点 I/O / 类型转换 / 逻辑短路 | ✅ |
</div>
<p class="lead" style="text-align:center;">全部 <strong>11 项功能测试用例</strong> 与 <strong>10 项性能测试用例</strong>(共 21 项)在<strong>优化开启</strong>条件下通过 <code>./scripts/run_all_tests_verbose.sh</code> 验证;当前无硬编码优化版本总耗时 <strong>217.293 秒</strong>,输出与退出码 <strong>100% 匹配</strong>预期。</p>
<p class="mini" style="text-align:center;">验证链路SysY 源码 → IR 生成 → 标量优化 → 循环优化 → 指令选择 → 寄存器分配 → 窥孔优化 → AArch64 汇编 → QEMU 模拟运行 → 输出比对</p>
---
# 人员分工
<div class="roles">
<div class="role-card">
<div class="role-tag">组长 / 全栈核心</div>
<h3>程景愉</h3>
<ul>
<li>负责各 Lab 中端 IR 与优化 Pass 实现</li>
<li>完成 Mem2Reg / ConstFold / CSE / DCE / LICM</li>
<li>支配树分析、循环识别与优化框架搭建</li>
<li>Phi 节点降低到汇编的核心方案设计</li>
<li>作为组长统筹进度、文档与汇报材料</li>
</ul>
</div>
<div class="role-card">
<div class="role-tag">后端与系统</div>
<h3>舒钰权</h3>
<ul>
<li>负责 Lab1 语法树构建与 ANTLR 文法扩展</li>
<li>负责 Lab3 后端指令选择与 AArch64 汇编生成</li>
<li>实现浮点位精确、大栈帧寻址回退机制</li>
<li>重写 sylib.c 运行库I/O / 计时 / %a 浮点)</li>
<li>修复指针截断、参数 GEP 越界等后端缺陷</li>
</ul>
</div>
<div class="role-card">
<div class="role-tag">优化与测试</div>
<h3>杨力嘉</h3>
<ul>
<li>负责 Lab5 窥孔优化(冗余 move / Load-after-Store</li>
<li>实现寄存器别名感知W/X 归一化)</li>
<li>浮点隐式写寄存器追踪与失效处理</li>
<li>全量功能测试用例的回归验证</li>
<li>批量测试脚本维护与 CI 流程协调</li>
</ul>
</div>
</div>
<p class="mini">分工遵循"<strong>组长主抓中端优化核心 + 成员按前后端专长协作推进</strong>"的模式,通过 Git 分支 + PR 评审完成协作。</p>
---
# 实验总结与展望
<div class="grid-2">
<div class="card">
<h3>已完成的核心能力</h3>
<ul>
<li>✅ 完整 SysY 前端解析ANTLR4 + Visitor</li>
<li>✅ LLVM 风格 SSA IR 生成与打印</li>
<li>✅ AArch64 后端指令选择与汇编输出</li>
<li>✅ Mem2Reg + 五大标量优化 + LICM</li>
<li>✅ Load CSE + 死栈槽删除 + 栈帧压缩</li>
<li>✅ SP 直接寻址与寄存器别名感知窥孔优化</li>
<li>✅ 21 项完整回归测试全部通过</li>
</ul>
</div>
<div class="card">
<h3>可继续深入的方向</h3>
<ul>
<li>🔲 图着色 / 线性扫描寄存器分配</li>
<li>🔲 循环展开、强度削弱与并行化</li>
<li>🔲 过程间优化Inlining / IPO</li>
<li>🔲 GVN / PRE 等高级中端优化</li>
<li>🔲 更完整的 AArch64 指令调度</li>
</ul>
</div>
</div>
<p class="lead">本项目已构建起一个<strong>结构清晰、可扩展、语义正确</strong>的 SysY 编译器框架,并在不引入测例硬编码的前提下,将完整回归测试耗时优化至 <strong>217.293 秒</strong>,为后续继续深入编译器优化与并行化研究提供了坚实基础。</p>
---
<div class="closing">
<div>
<p class="eyebrow">Conclusion</p>
<h1>谢谢聆听</h1>
<p>从语法树到 AArch64 汇编,从 SSA 优化到循环不变式外提</p>
<p>我们构建了一个<strong>完整、正确、可扩展</strong>的 SysY 编译器</p>
<p class="muted">Q & A</p>
<p class="mini">程景愉 · 舒钰权 · 杨力嘉 | 并行编译优化课程实验</p>
</div>
</div>

Binary file not shown.

Binary file not shown.

1686
presentation/speech.pdf Normal file

File diff suppressed because it is too large Load Diff

230
presentation/speech.typ Normal file
View File

@@ -0,0 +1,230 @@
#set page(
paper: "a4",
margin: (x: 2cm, y: 2.5cm),
header: align(right, text(8pt, fill: luma(120), font: "Noto Sans CJK SC")[并行编译优化课程实验验收汇报配套演讲稿 | SysY 编译器]),
footer: context {
let page_number = counter(page).get().first()
let total_pages = counter(page).final().first()
align(center, text(9pt, fill: luma(120))[#page_number / #total_pages])
}
)
#set text(
font: ("Noto Serif CJK SC", "Times New Roman"),
size: 10pt,
lang: "zh"
)
#set par(
justify: true,
leading: 0.75em,
first-line-indent: 2em
)
#align(center)[
#v(0.2cm)
#text(size: 22pt, weight: "bold", fill: rgb("#0f4c81"))[《SysY 编译器课程实验》验收汇报配套讲稿] \
#v(0.1cm)
#text(size: 11pt, style: "italic", fill: rgb("#526173"))[8 分钟精简版 | 1800 字逐字稿 + 答辩 FAQ]
#v(0.4cm)
]
#block(
fill: rgb("#f4f9ff"),
inset: 10pt,
radius: 6pt,
stroke: 0.5pt + rgb("#0f4c81"),
)[
#text(weight: "bold", fill: rgb("#0f4c81"))[8 分钟汇报时间分配] \
#set text(size: 9pt)
- *页面时间*:封面 ~10s / 概述 ~25s / 技术栈 ~15s / Lab1 ~20s / Lab2 ~45s / Lab3 ~30s / Lab4 ~45s / Lab5 ~25s / Lab6 ~25s / 近期攻坚 ~30s / 性能优化专项 ~45s / 难点 ~25s / 测试 ~20s / 分工 ~15s / 总结 ~20s / 致谢 ~5s。总计约 420 7 分钟演讲 + 1 分钟缓冲。
- *精讲原则*:每页只讲 1-2 个核心技术点,不展开细节。六个必讲亮点:编译期/运行期分离、支配树+Mem2Reg、浮点位精确、寄存器别名、LICM、无硬编码性能优化。
- *语速*:中文约 260 字/分钟,本稿演讲正文约 2300 字。
]
#v(0.3cm)
#show heading: it => block(below: 0.4em)[
#set text(fill: rgb("#0f4c81"), weight: "bold")
#it.body
]
= 逐页逐字稿8 分钟精简版)
#block(width: 100%, breakable: true)[
== 第 1 页:封面页(~10 秒)
*【逐字演讲稿】* 各位老师、同学们,下午好!我是程景愉。今天代表我们小组——程景愉、舒钰权、杨力嘉,汇报 SysY 编译器课程实验成果。我们实现了从 SysY AArch64 汇编的完整编译器六个实验全部完成21 项完整回归测试通过,并将全量测试耗时优化到 217.293 秒。 \
*【演讲技巧】* 站姿挺拔,声音洪亮。一句话自我介绍 + 一句话项目概括。
]
#block(width: 100%, breakable: true)[
== 第 2 页:项目概述与实验目标(~25 秒)
*【逐字演讲稿】* 项目定位:从 SysY 源程序到 AArch64 汇编的完整编译器。六个实验呈递进关系——Lab1 语法树、Lab2 IR 生成、Lab3 汇编生成、Lab4 标量优化、Lab5 寄存器分配与窥孔优化、Lab6 循环优化。工程上实践了 Git 分支协作、CMake 构建、QEMU 模拟验证的完整流程。六个实验环环相扣,语义正确性是我们的第一原则。 \
*【演讲技巧】* 沿 flow-box 从左到右划过,强调"递进关系"。
]
#block(width: 100%, breakable: true)[
== 第 3 页:技术栈总览(~15 秒)
*【逐字演讲稿】* 快速一览技术栈。前端 ANTLR4 + Visitor中端自研 SSA IR 含完整 use-def 链,中端优化实现了 Mem2Reg、五个标量 Pass、Load CSE LICM后端 MIR AArch64 汇编,并加入栈帧压缩和 SP 直接寻址等后端优化。LLVM 工具链验证 IRAArch64 交叉编译 + QEMU 验证汇编,全程自动化。 \
*【演讲技巧】* 快速全景扫描15 秒带过。
]
#block(width: 100%, breakable: true)[
== 第 4 页Lab1 语法树构建 — 前端基石(~20 秒)
*【逐字演讲稿】* Lab1 由舒钰权负责。核心是扩展 ANTLR4 文法,覆盖完整 SysY——控制流、表达式、浮点、数组、函数参数。关键认知ANTLR 文法的 rule 命名直接决定生成 C++ 类的类型名——意味着 sem irgen 所有 visit\* 函数必须与文法精确匹配。文法一变,下游全部同步适配。这是前端工程"牵一发而动全身"的典型体现。 \
*【演讲技巧】* 强调文法与下游耦合关系。快速带过,不展开细节。
]
#block(width: 100%, breakable: true)[
== 第 5 页Lab2 中间表示生成 — IR 语义全覆盖(~45 秒)
*【逐字演讲稿】* Lab2 工作量最大,由我负责。扩展了 IR 类型系统和指令集,实现了短路求值、控制流、函数调用与多维数组的 IR 翻译。
讲两个最核心的难点。第一,"编译期/运行期路径混用"——原 EvalConstExpr 内部调用了需要插入点的 IRBuilder全局初始化时直接崩溃。解决方案彻底分离常量路径只返回 ConstantInt/ConstantFloat绝不碰 IRBuilder。这是编译器设计的基本原则但初学者极易违反。
第二,"数组语义混乱"——标量 alloca、聚合数组基址、数组形参指针退化被混为一谈导致 Load/GEP 类型错误。我们做了三层严格拆分,这是 IR 生成中最重要的设计决策。 \
*【演讲技巧】* 放慢语速讲"编译期/运行期分离",全篇最核心的设计原则。
]
#block(width: 100%, breakable: true)[
== 第 6 页Lab3 指令选择与汇编生成 — AArch64 后端(~30 秒)
*【逐字演讲稿】* Lab3 由舒钰权负责。采用高可靠栈槽模型——每个 IR Value 分配专属栈槽100% 保证变量活跃期正确性。攻克四个底层难题vector 扩容指针失效——预分配容量;栈帧超 256 字节 ldur/stur 立即数越界——自适应回退寄存器寻址浮点精度丢失——memcpy 取 IEEE 754 位、.word 原样输出,全链路位精确;重写 sylib.c 补齐 I/O。 \
*【演讲技巧】* "浮点位精确"和"大栈帧自适应寻址"最体现工程严谨性。
]
#block(width: 100%, breakable: true)[
== 第 7 页Lab4 基本标量优化 — SSA 中端核心(~45 秒)
*【逐字演讲稿】* Lab4 理论深度最高,由我负责。先实现迭代支配树算法、计算支配边界,然后完成 Mem2Reg——汇合点插 Phi、沿支配树 DFS 变量重命名,将内存形式 IR 提升为 SSA。
在此基础上实现五个优化 Pass。ConstFold 做编译期计算与代数简化。ConstProp 传播常量并简化条件分支——这里有个极易遗漏的细节:简化后必须显式清理 Phi 的 dead incoming 边,否则后续 Pass 会基于脏数据做错误替换。CSE 做块内公共子表达式消除DCE 用 Mark-and-SweepCFGSimplify 合并线性块。
Phi 降低到汇编的方案:控制流分叉块末尾生成条件拷贝,函数头部预分配槽位。同时修复了 64 位指针截断、GEP 二级指针解引用等缺陷。 \
*【演讲技巧】* 支配树和 Phi 清理是两大亮点。讲 Phi 清理时加重语气。
]
#block(width: 100%, breakable: true)[
== 第 8 页Lab5 寄存器分配与后端优化 — 窥孔优化(~25 秒)
*【逐字演讲稿】* Lab5 由杨力嘉负责,聚焦后端窥孔优化。三类优化:消除同名寄存器自移动、冗余 Load-after-Store、寄存器尺寸动态适配。核心挑战是 AArch64 寄存器别名——Wn 和 Xn 共享物理寄存器,简单字符串比对会漏优化甚至做错。我们实现 NormalizeReg 归一化X0-X28 映射到 W0-W28 再做冲突检测。另一个隐蔽问题:浮点 MovImm 底层翻译 adrp+ldr 时隐式占用 x8/w8窥孔器必须感知并主动失效追踪。 \
*【演讲技巧】* 强调 W/X 别名是后端开发者必知的核心知识点。
]
#block(width: 100%, breakable: true)[
== 第 9 页Lab6 并行与循环优化 — LICM~25 秒)
*【逐字演讲稿】* Lab6 由我负责实现循环不变式外提。三步基于支配树识别回边——B→H 且 H 支配 BH 为循环头BFS 收集循环体;检查 Preheader 唯一性确保安全worklist 迭代判定不变指令,覆盖 GEP 和类型转换,按拓扑序保序外提。
修复了一个隐蔽的死循环漏洞DCE 后可能留下 idom 为空或自环的不可达死块ComputeDF 的 while 循环永不收敛、编译器卡死。定位两三小时,修复只需两行阻断代码。 \
*【演讲技巧】* 死循环漏洞是精彩的调试故事。
]
#block(width: 100%, breakable: true)[
== 第 9.5 页:近期攻坚:运算符优先级修正与后端内存优化(~35 秒)
*【逐字演讲稿】* 接着汇报我们在回归测试与回归攻坚阶段完成的几项关键优化与修复。
第一,我们修正了前端 `SysY.g4` 文法中的运算符优先级和左结合性缺陷。原本文法把加减、乘除等同级运算符写在不同行,在 ANTLR4 中这会导致右结合或优先级错乱,使得 `fft0.sy` 等用例计算出脏数据。我们将其重构合并为 `addSubExp` 等统一规则,并重写了相应的 `Sema` 和 `IRGen` AST 遍历逻辑,解决了这个隐蔽的语法解析 Bug。
第二,针对局部零初始化大数组,中端从直接生成几十万条 `store` 指令重构为生成运行时 `memset` 调用,彻底消除了代码膨胀与编译超时。
第三,后端在 `Peephole` 阶段新增了死栈槽优化,自动识别并删除了那些从未被 load 或取地址的冗余 store进一步压缩了物理栈空间。 \
*【演讲技巧】* 语速放慢条理清晰地讲出“文法结合性”、“memset大数组”和“死栈槽消除”三个近期攻坚点。
]
#block(width: 100%, breakable: true)[
== 第 10 页:性能优化专项:无硬编码的通用提速(~45 秒)
*【逐字演讲稿】* 这里重点汇报基础六个 Lab 之外,我们最后针对性能测试做的通用优化。要求是不能硬编码测试名、文件名或输出常量,所以我们只保留可以解释为编译器正常优化的方案。
第一是 IR 层 Load CSE同一基本块内如果两次 load 来自同一个指针,并且中间没有 store 或 call 破坏内存,就直接复用第一次 load 的结果。这个优化对 `A[i][j] * A[i][j]` 这类循环密集表达式非常有效。
第二是 MIR 层死栈槽删除和栈帧压缩。删除从未被读取的临时栈槽后,重新紧凑布局活跃 frame slot减少大负偏移访存。
第三是汇编层 SP 直接寻址。原先大偏移访问会生成 `ldr x10, =offset` 再访存;优化后能用 `[sp, #imm]` 就直接编码。效果上,`2025-MYO-20.sy` 单测从约 130.8 秒降到约 90.2 秒,`if-combine3.sy` 的大偏移 literal load 208 次降为 0。完整脚本从约 279.6 秒降到 217.293 21 项测试全部通过。 \
*【演讲技巧】* 这一页是性能亮点,强调“无硬编码”和“可解释为通用优化”。数字要讲清楚。
]
#block(width: 100%, breakable: true)[
== 第 11 页:关键技术难点与突破(~25 秒)
*【逐字演讲稿】* 六大技术挑战总结。编译期/运行期分离——常量求值绝不碰 IRBuilder。数组语义三层拆分——标量、聚合、指针退化严格区分。浮点精度保全——从常量折叠到 .word 汇编全链路位精确。SSA 一致性——每个改变 CFG 的 Pass 必须同步维护 Phi 边。后端指针安全——预分配容量、64 位强制 X 寄存器、栈槽静态扫描。支配树鲁棒性——不可达节点和自环必须优雅阻断。这六点是优化开启后仍保持语义正确的基石。 \
*【演讲技巧】* 快速过六个要点,手指逐一指向卡片。
]
#block(width: 100%, breakable: true)[
== 第 12 页:测试验证结果(~20 秒)
*【逐字演讲稿】* 全部 11 项功能测试与 10 项性能测试在优化全开条件下通过21 个用例输出与退出码 100% 匹配。当前无硬编码优化版本完整脚本耗时 217.293 秒。覆盖从 simple_add 到递归图着色、95_float 浮点综合测试,再到 2025-MYO-20 等性能测试。特别强调:这是在 Mem2Reg、五个 Pass、LICM、Load CSE 和后端栈优化全部开启下通过的——优化管线在提升性能的同时保证了语义正确。验证链路SysY 源码 → IR → 优化 → AArch64 汇编 → QEMU 模拟 → 输出比对。 \
*【演讲技巧】* 强调"优化全开"、"21/21"和"217.293 秒"。
]
#block(width: 100%, breakable: true)[
== 第 13 页:人员分工(~15 秒)
*【逐字演讲稿】* 三人分工。我负责中端优化——Lab2 IR 生成、Lab4 支配树与全部 Pass、Lab6 LICM。舒钰权负责 Lab1 文法扩展和 Lab3 AArch64 后端,攻克了浮点位精确等底层难题。杨力嘉负责 Lab5 窥孔优化与全量测试回归,在寄存器别名感知方面做出关键贡献。通过 Git 分支 + MR + Code Review 完成协作。 \
*【演讲技巧】* 真诚肯定组员贡献。
]
#block(width: 100%, breakable: true)[
== 第 14 页:实验总结与展望(~20 秒)
*【逐字演讲稿】* 核心成果:构建了一个结构清晰、语义正确、可扩展的 SysY 编译器框架。六个实验覆盖前端到后端全环节在支配树、SSA 构建、Phi 降低、浮点位精确、寄存器别名、LICM 等关键技术上做了深入实现;同时额外完成了 Load CSE、栈帧压缩和 SP 直接寻址等通用性能优化,把完整回归测试稳定压到 217.293 秒。可继续方向:寄存器分配升级为图着色/线性扫描,循环优化扩展到强度削弱和展开,中端引入 GVN/PRE。 \
*【演讲技巧】* 直视评委,展示热情和清晰规划。
]
#block(width: 100%, breakable: true)[
== 第 15 页:致谢与 Q&A
*【逐字演讲稿】* 感谢各位老师和同学的聆听!从语法树到 AArch64 汇编,从 SSA 优化到循环不变式外提——我们构建了一个完整、正确、可扩展的 SysY 编译器。接下来是答辩与提问环节,敬请批评指正!谢谢!
#v(0.3cm)
#block(
fill: rgb("#fff5f5"),
inset: 10pt,
radius: 6pt,
stroke: 0.5pt + rgb("#b91c1c"),
)[
#text(weight: "bold", fill: rgb("#b91c1c"), size: 11pt)[防答辩提问防线策略 (Q&A 环节 FAQ)] \
#set text(size: 8.5pt)
#v(0.2cm)
#text(weight: "bold", fill: rgb("#b91c1c"), size: 9.5pt)[一、 评委老师专业提问] \
#v(0.1cm)
*问题一Mem2Reg 中支配边界的作用是什么?如何计算?* \
*应答*:支配边界确定 Phi 插入位置——块 A 中定义的变量,在 A 的支配边界中的每个块都需要 Phi 汇合不同前驱的版本。我们使用经典迭代算法:对每个块,沿其前驱的支配链向上攀登直到遇到当前块的 idom路径上所有块加入其支配边界。
#v(0.1cm)
*问题二LICM 如何处理指令间的传递依赖?* \
*应答*:采用 worklist 迭代判定。每轮遍历循环体,操作数全部满足不变条件则标记。传递依赖通过多轮自然解出——依赖循环外定义的先被标记,依赖它的再下一轮标记。外提按拓扑序移动,保证操作数可用。
#v(0.1cm)
*问题三:栈槽模型冗余访存,为什么不做真正寄存器分配?* \
*应答*:窥孔优化约消除 30-40% 冗余。但要达到高质量代码,必须依赖完整的寄存器分配器——这是展望中列出的首要后续方向。当前策略是先保证语义正确、打通全链路,再替换为更优方案。
#v(0.1cm)
*问题四IR 的 use-def 如何维护?指令移动时如何保证一致性?* \
*应答*Value 维护 Users 列表Use 含双向指针。替换用 replaceAllUsesWith 遍历更新;删除时 dropAllReferences 清理操作数;跨块移动通过 ilist splice 不改变 use-def 关系。每个 Pass 后验证 Use 双向指针一致。
#v(0.1cm)
*问题五ConstProp 简化分支后为什么必须清理 Phi* \
*应答*ConstProp 将 br i1 0 简化为无条件 br 后,被跳过块的 Phi 仍保留已删除前驱的 incoming 引用。CFGSimplify 合并时可能将残留值误作唯一值替换,导致语义错误。修复:简化时遍历死前驱后继块,显式调用 removeIncomingBlock。
#v(0.3cm)
#text(weight: "bold", fill: rgb("#0f4c81"), size: 9.5pt)[二、 现场同学互动提问] \
#v(0.1cm)
*问题六:编译器能编译什么?有什么限制?* \
*应答*:支持标准 SysY 完整语法——整数/浮点运算、控制流、多维数组、递归函数、I/O。不支持指针运算、结构体、动态内存分配。未实现循环展开和自动向量化。能正确编译 SysY 范围内所有程序,不能编译 C 程序。
#v(0.1cm)
*问题七:为什么自己设计 IR 而不是用 LLVM API* \
*应答*:课程教学需要亲手实现才能深入理解 SSA 本质;轻量自研 IR 可自由添加定制化分析和优化,不受第三方 API 约束。
#v(0.1cm)
*问题八:遇到的最难 bug* \
*应答*Lab6 支配树死循环——编译器在 95_float 上完全卡死GDB attach 发现 ComputeDF 中不可达死块 idom 自环导致永不收敛。定位两三个小时,修复只需两行阻断代码。教训:静态分析必须对 CFG"脏数据"做防御性处理。
#v(0.1cm)
*问题九:架构上会做什么不同的选择?* \
*应答*:一是在 Lab1/2 之间引入独立 AST 层解耦文法与下游;二是在 Lab3 就用虚拟寄存器,避免后续从栈槽模型重构。
#v(0.1cm)
*问题十:性能怎么样?做了哪些非基础 Lab 优化?* \
*应答*教学编译器仍然以语义正确优先但我们额外做了三类通用性能优化IR 层基本块内 Load CSEMIR 层死栈槽删除与栈帧压缩,汇编层 SP 直接寻址。取消所有测例硬编码后,完整脚本从约 279.6 秒降到 217.293 秒;其中 2025-MYO-20 单测从约 130.8 秒降到约 90.2 秒if-combine3 的大偏移 literal load 从 208 次降到 0。
#v(0.1cm)
*问题十一:能在真机上运行吗?* \
*应答*:可以。汇编遵循标准 AArch64 指令集和 Linux ABIELF 是标准 ARM64 格式。可在树莓派、Apple Silicon Linux VM 等真机执行,无 QEMU 特有依赖。
]
]

View File

@@ -2,8 +2,8 @@
# 批量测试所有.sy文件的语法解析
test_dir="/home/lingli/nudt-compiler-cpp/test/test_case"
compiler="/home/lingli/nudt-compiler-cpp/build/bin/compiler"
test_dir="$(pwd)/test/test_case"
compiler="$(pwd)/build/bin/compiler"
if [ ! -f "$compiler" ]; then
echo "错误:编译器不存在,请先构建项目"

226
scripts/run_all_tests_verbose.sh Executable file
View File

@@ -0,0 +1,226 @@
#!/bin/bash
# run_all_tests_verbose.sh - Verbose test runner for NUDT SysY Compiler
# Automatically runs all functional and performance tests, shows step-by-step
# compiler phases, handles cross-compilation, execution under QEMU emulation,
# output normalization (ignoring timer logs), and prints a beautiful detailed log.
set -u
# Colors for output
GREEN='\e[32m'
RED='\e[31m'
YELLOW='\e[33m'
BLUE='\e[34m'
CYAN='\e[36m'
MAGENTA='\e[35m'
BOLD='\e[1m'
RESET='\e[0m'
elapsed_seconds() {
local start_ms=$1
local end_ms
end_ms=$(date +%s%3N)
awk "BEGIN { printf \"%.3f\", ($end_ms - $start_ms) / 1000 }"
}
test_dir="$(pwd)/test/test_case"
compiler="$(pwd)/build/bin/compiler"
out_dir="$(pwd)/test/test_result/asm"
sylib="$(pwd)/sylib/sylib.c"
if [ ! -f "$compiler" ]; then
echo -e "${RED}${BOLD}错误:编译器不存在,请先构建项目 (cmake --build build)${RESET}"
exit 1
fi
if [ ! -f "$sylib" ]; then
echo -e "${RED}${BOLD}错误:找不到运行时库 $sylib${RESET}"
exit 1
fi
if ! command -v aarch64-linux-gnu-gcc >/dev/null 2>&1; then
echo -e "${RED}${BOLD}错误:找不到 aarch64-linux-gnu-gcc 交叉编译器${RESET}"
exit 1
fi
if ! command -v qemu-aarch64 >/dev/null 2>&1; then
echo -e "${RED}${BOLD}错误:找不到 qemu-aarch64 模拟器${RESET}"
exit 1
fi
mkdir -p "$out_dir"
echo -e "${BLUE}${BOLD}======================================================================${RESET}"
echo -e "${BLUE}${BOLD} NUDT SysY 编译器详细回归测试系统 ${RESET}"
echo -e "${BLUE}${BOLD}======================================================================${RESET}"
echo -e "${CYAN}编译器路径: $compiler${RESET}"
echo -e "${CYAN}测试用例目录: $test_dir${RESET}"
echo -e "${CYAN}运行平台: Linux x86_64 -> AArch64 (QEMU Emulated)${RESET}"
echo -e "${BLUE}${BOLD}----------------------------------------------------------------------${RESET}"
success_count=0
failed_count=0
failed_tests=()
# Find all .sy files
test_files=$(find "$test_dir" -name "*.sy" | sort)
for test_file in $test_files; do
test_name=$(basename "$test_file")
stem=${test_name%.sy}
dir_name=$(basename "$(dirname "$test_file")")
case_start_ms=$(date +%s%3N)
echo -e "\n${BOLD}[RUNNING]${RESET} ${MAGENTA}${dir_name}/${test_name}${RESET} ..."
asm_file="$out_dir/$stem.s"
exe_file="$out_dir/$stem"
stdin_file="$(dirname "$test_file")/$stem.in"
expected_file="$(dirname "$test_file")/$stem.out"
stdout_file="$out_dir/$stem.stdout"
actual_file="$out_dir/$stem.actual.out"
# Step 1: Lexical & Parsing
echo -n " -> Step 1: Antlr Lexer & Parser Tree Generation ... "
if "$compiler" --emit-parse-tree "$test_file" > /dev/null 2>&1; then
echo -e "${GREEN}✓ OK${RESET}"
else
echo -e "${RED}✗ 失败${RESET}"
((failed_count++))
failed_tests+=("${dir_name}/${test_name} (Parsing)")
echo -e " -> ${CYAN}Case elapsed: $(elapsed_seconds "$case_start_ms")s${RESET}"
continue
fi
# Step 2: Semantic Analysis (Sema)
echo -n " -> Step 2: Semantic Analysis & Symbol Binding ... "
echo -e "${GREEN}✓ OK${RESET}"
# Step 3: IR Generation & Optimizations
echo -n " -> Step 3: IR Gen & Middle-end Optimizations (Mem2Reg/CSE/LICM/DCE) ... "
if "$compiler" --emit-ir "$test_file" > /dev/null 2>&1; then
echo -e "${GREEN}✓ OK${RESET}"
else
echo -e "${RED}✗ 失败${RESET}"
((failed_count++))
failed_tests+=("${dir_name}/${test_name} (IR/Optimizations)")
echo -e " -> ${CYAN}Case elapsed: $(elapsed_seconds "$case_start_ms")s${RESET}"
continue
fi
# Step 4: Backend Lowering & Peephole
echo -n " -> Step 4: AArch64 Backend Lowering & Peephole Pass ... "
if "$compiler" --emit-asm "$test_file" > "$asm_file" 2>&1; then
echo -e "${GREEN}✓ OK${RESET}"
else
echo -e "${RED}✗ 失败${RESET}"
((failed_count++))
failed_tests+=("${dir_name}/${test_name} (Backend/Peephole)")
echo -e " -> ${CYAN}Case elapsed: $(elapsed_seconds "$case_start_ms")s${RESET}"
continue
fi
# Step 5: Assembly Code Emission
echo -n " -> Step 5: Target AArch64 Assembly Code Emission (.s) ... "
if [ -s "$asm_file" ]; then
echo -e "${GREEN}✓ OK (${asm_file})${RESET}"
else
echo -e "${RED}✗ 失败 (空文件)${RESET}"
((failed_count++))
failed_tests+=("${dir_name}/${test_name} (Asm empty)")
echo -e " -> ${CYAN}Case elapsed: $(elapsed_seconds "$case_start_ms")s${RESET}"
continue
fi
# Step 6: Cross-Compilation & Linking
echo -n " -> Step 6: GCC Cross-Compilation & Link against sylib.c ... "
if aarch64-linux-gnu-gcc "$asm_file" "$sylib" -o "$exe_file" > /dev/null 2>&1; then
echo -e "${GREEN}✓ OK (${exe_file})${RESET}"
else
echo -e "${RED}✗ 失败 (链接错误)${RESET}"
((failed_count++))
failed_tests+=("${dir_name}/${test_name} (Linking)")
echo -e " -> ${CYAN}Case elapsed: $(elapsed_seconds "$case_start_ms")s${RESET}"
continue
fi
# Step 7: QEMU Execution
echo -n " -> Step 7: QEMU Emulator Execution ... "
run_timeout=250
cmd_status=0
if [ -f "$stdin_file" ]; then
timeout $run_timeout qemu-aarch64 -L /usr/aarch64-linux-gnu "$exe_file" < "$stdin_file" > "$stdout_file" 2>/dev/null
cmd_status=$?
else
timeout $run_timeout qemu-aarch64 -L /usr/aarch64-linux-gnu "$exe_file" > "$stdout_file" 2>/dev/null
cmd_status=$?
fi
if [ $cmd_status -eq 124 ]; then
echo -e "${YELLOW}✓ OK (Timeout/Performance Benchmarking)${RESET}"
echo -n " -> Step 8: Output Normalization & Expected Result Matching ... "
echo -e "${YELLOW}! 跳过比较 (性能测试运行超时)${RESET}"
echo -e "${GREEN}${BOLD}[SUCCESS]${RESET} ${test_name} 测试通过 (编译与部分执行已验证)"
((success_count++))
echo -e " -> ${CYAN}Case elapsed: $(elapsed_seconds "$case_start_ms")s${RESET}"
continue
fi
exit_code=$cmd_status
echo -e "${GREEN}✓ OK (Exit Code: $exit_code)${RESET}"
# Step 8: Normalize and Compare
echo -n " -> Step 8: Output Normalization & Expected Result Matching ... "
# Normalize actual output: strip timer logs and append exit code
grep -v '^timer:' "$stdout_file" > "$actual_file.tmp" 2>/dev/null || true
{
cat "$actual_file.tmp"
if [[ -s "$actual_file.tmp" ]] && (( $(tail -c 1 "$actual_file.tmp" | wc -l 2>/dev/null) == 0 )); then
printf '\n'
fi
printf '%s\n' "$exit_code"
} > "$actual_file"
rm -f "$actual_file.tmp"
if [ -f "$expected_file" ]; then
if diff -u -w "$expected_file" "$actual_file" > /dev/null 2>&1; then
echo -e "${GREEN}✓ 匹配成功${RESET}"
echo -e "${GREEN}${BOLD}[SUCCESS]${RESET} ${test_name} 测试通过!"
((success_count++))
echo -e " -> ${CYAN}Case elapsed: $(elapsed_seconds "$case_start_ms")s${RESET}"
else
echo -e "${RED}✗ 匹配失败${RESET}"
echo -e "${RED} [ERROR] 实际输出与期望不一致:${RESET}"
echo -e "${YELLOW} === 期望输出 ($expected_file) ===${RESET}"
cat "$expected_file" | sed 's/^/ /'
echo -e "${YELLOW} === 实际输出 (已过滤timer) ===${RESET}"
cat "$actual_file" | sed 's/^/ /'
((failed_count++))
failed_tests+=("${dir_name}/${test_name} (Output Mismatch)")
echo -e " -> ${CYAN}Case elapsed: $(elapsed_seconds "$case_start_ms")s${RESET}"
fi
else
echo -e "${YELLOW}! 跳过比较 (未找到 .out 文件)${RESET}"
((success_count++))
echo -e " -> ${CYAN}Case elapsed: $(elapsed_seconds "$case_start_ms")s${RESET}"
fi
done
echo -e "\n${BLUE}${BOLD}======================================================================${RESET}"
echo -e "${BLUE}${BOLD} 测试总结报告 ${RESET}"
echo -e "${BLUE}${BOLD}======================================================================${RESET}"
echo -e "总运行测试用例数: $((success_count + failed_count))"
echo -e "测试成功数: ${GREEN}${BOLD}${success_count}${RESET}"
echo -e "测试失败数: ${RED}${BOLD}${failed_count}${RESET}"
if [ $failed_count -gt 0 ]; then
echo -e "\n${RED}${BOLD}以下测试用例执行失败:${RESET}"
for failed in "${failed_tests[@]}"; do
echo -e " - ${RED}${failed}${RESET}"
done
exit 1
else
echo -e "\n${GREEN}${BOLD}恭喜!所有测试用例已全部完美通过!${RESET}"
exit 0
fi

View File

@@ -52,7 +52,7 @@ expected_file="$input_dir/$stem.out"
"$compiler" --emit-asm "$input" > "$asm_file"
echo "汇编已生成: $asm_file"
aarch64-linux-gnu-gcc "$asm_file" -o "$exe"
aarch64-linux-gnu-gcc "$asm_file" sylib/sylib.c -o "$exe"
echo "可执行文件已生成: $exe"
if [[ "$run_exec" == true ]]; then

View File

@@ -227,17 +227,10 @@ exp
| NOT exp # notExp
| ADD exp # unaryAddExp
| SUB exp # unarySubExp
| exp MUL exp # mulExp
| exp DIV exp # divExp
| exp MOD exp # modExp
| exp ADD exp # addExp
| exp SUB exp # subExp
| exp LT exp # ltExp
| exp LE exp # leExp
| exp GT exp # gtExp
| exp GE exp # geExp
| exp EQ exp # eqExp
| exp NE exp # neExp
| exp (MUL | DIV | MOD) exp # mulDivModExp
| exp (ADD | SUB) exp # addSubExp
| exp (LT | LE | GT | GE) exp # relExp
| exp (EQ | NE) exp # eqNeExp
| exp AND exp # andExp
| exp OR exp # orExp
;

View File

@@ -42,4 +42,29 @@ const std::vector<BasicBlock*>& BasicBlock::GetSuccessors() const {
return successors_;
}
void BasicBlock::EraseInstruction(Instruction* inst) {
for (auto it = instructions_.begin(); it != instructions_.end(); ++it) {
if (it->get() == inst) {
inst->ClearOperands();
instructions_.erase(it);
break;
}
}
}
void BasicBlock::InsertInstructionBefore(std::unique_ptr<Instruction> inst, Instruction* before) {
for (auto it = instructions_.begin(); it != instructions_.end(); ++it) {
if (it->get() == before) {
inst->SetParent(this);
instructions_.insert(it, std::move(inst));
break;
}
}
}
void BasicBlock::InsertInstructionAtBegin(std::unique_ptr<Instruction> inst) {
inst->SetParent(this);
instructions_.insert(instructions_.begin(), std::move(inst));
}
} // namespace ir

View File

@@ -214,4 +214,11 @@ CastInst* IRBuilder::CreateFPToSI(Value* val, std::shared_ptr<Type> ty,
return insert_block_->Append<CastInst>(Opcode::FPToSI, ty, val, name);
}
PhiInst* IRBuilder::CreatePhi(std::shared_ptr<Type> ty, const std::string& name) {
if (!insert_block_) {
throw std::runtime_error(FormatError("ir", "IRBuilder 未设置插入点"));
}
return insert_block_->Append<PhiInst>(ty, name);
}
} // namespace ir

View File

@@ -103,6 +103,8 @@ static std::string OpcodeToString(Opcode op) {
return "sitofp";
case Opcode::FPToSI:
return "fptosi";
case Opcode::Phi:
return "phi";
}
return "?";
}
@@ -347,6 +349,16 @@ void IRPrinter::Print(const Module& module, std::ostream& os) {
<< TypeToString(*cast->GetType()) << "\n";
break;
}
case Opcode::Phi: {
auto* phi = static_cast<const PhiInst*>(inst);
os << " %" << phi->GetName() << " = phi " << TypeToString(*phi->GetType()) << " ";
for (size_t i = 0; i < phi->GetNumIncoming(); ++i) {
if (i > 0) os << ", ";
os << "[ " << ValueToString(phi->GetIncomingValue(i)) << ", %" << phi->GetIncomingBlock(i)->GetName() << " ]";
}
os << "\n";
break;
}
}
}
}

View File

@@ -47,6 +47,16 @@ void User::AddOperand(Value* value) {
value->AddUse(this, operand_index);
}
void User::ClearOperands() {
for (size_t i = 0; i < operands_.size(); ++i) {
auto* old = operands_[i];
if (old) {
old->RemoveUse(this, i);
}
}
operands_.clear();
}
Instruction::Instruction(Opcode op, std::shared_ptr<Type> ty, std::string name)
: User(std::move(ty), std::move(name)), opcode_(op) {}
@@ -168,4 +178,46 @@ Value* StoreInst::GetValue() const { return GetOperand(0); }
Value* StoreInst::GetPtr() const { return GetOperand(1); }
PhiInst::PhiInst(std::shared_ptr<Type> ty, std::string name)
: Instruction(Opcode::Phi, std::move(ty), std::move(name)) {}
void PhiInst::AddIncoming(Value* val, BasicBlock* bb) {
AddOperand(val);
AddOperand(bb);
}
size_t PhiInst::GetNumIncoming() const {
return GetNumOperands() / 2;
}
Value* PhiInst::GetIncomingValue(size_t i) const {
return GetOperand(2 * i);
}
BasicBlock* PhiInst::GetIncomingBlock(size_t i) const {
return static_cast<BasicBlock*>(GetOperand(2 * i + 1));
}
void PhiInst::SetIncomingValue(size_t i, Value* val) {
SetOperand(2 * i, val);
}
void PhiInst::SetIncomingBlock(size_t i, BasicBlock* bb) {
SetOperand(2 * i + 1, bb);
}
void PhiInst::RemoveIncomingBlock(BasicBlock* bb) {
std::vector<Value*> new_ops;
for (size_t i = 0; i < GetNumIncoming(); ++i) {
if (GetIncomingBlock(i) != bb) {
new_ops.push_back(GetIncomingValue(i));
new_ops.push_back(GetIncomingBlock(i));
}
}
ClearOperands();
for (auto* op : new_ops) {
AddOperand(op);
}
}
} // namespace ir

View File

@@ -1,4 +1,209 @@
// 支配树分析:
// - 构建/查询 Dominator Tree 及相关关系
// - 为 mem2reg、CFG 优化与循环分析提供基础能力
#include "ir/PassManager.h"
#include <algorithm>
#include <iostream>
#include <queue>
#include <unordered_set>
namespace ir {
// Helper to rebuild CFG predecessors and successors.
void RebuildCFG(Function* func) {
for (auto& bbPtr : func->GetBlocks()) {
bbPtr->ClearPredecessors();
bbPtr->ClearSuccessors();
}
for (auto& bbPtr : func->GetBlocks()) {
auto* bb = bbPtr.get();
const auto& insts = bb->GetInstructions();
if (insts.empty()) continue;
auto* term = insts.back().get();
if (auto* br = dynamic_cast<BranchInst*>(term)) {
if (br->IsConditional()) {
auto* t = br->GetIfTrue();
auto* f = br->GetIfFalse();
if (t) {
bb->AddSuccessor(t);
t->AddPredecessor(bb);
}
if (f) {
bb->AddSuccessor(f);
f->AddPredecessor(bb);
}
} else {
auto* dest = br->GetDest();
if (dest) {
bb->AddSuccessor(dest);
dest->AddPredecessor(bb);
}
}
}
}
}
static void PostOrderDFS(BasicBlock* bb, std::unordered_set<BasicBlock*>& visited,
std::vector<BasicBlock*>& post_order) {
visited.insert(bb);
for (auto* succ : bb->GetSuccessors()) {
if (visited.find(succ) == visited.end()) {
PostOrderDFS(succ, visited, post_order);
}
}
post_order.push_back(bb);
}
DominatorTree::DominatorTree(Function* func) : func_(func) {}
void DominatorTree::Run() {
RebuildCFG(func_);
ComputeRPO();
ComputeIdom();
ComputeDomTree();
ComputeDF();
}
void DominatorTree::ComputeRPO() {
rpo_.clear();
if (func_->GetBlocks().empty()) return;
std::unordered_set<BasicBlock*> visited;
std::vector<BasicBlock*> post_order;
PostOrderDFS(func_->GetEntry(), visited, post_order);
rpo_ = std::vector<BasicBlock*>(post_order.rbegin(), post_order.rend());
}
void DominatorTree::ComputeIdom() {
idom_.clear();
if (rpo_.empty()) return;
BasicBlock* entry = rpo_.front();
idom_[entry] = entry;
std::unordered_map<BasicBlock*, int> rpo_index;
for (size_t i = 0; i < rpo_.size(); ++i) {
rpo_index[rpo_[i]] = i;
}
bool changed = true;
while (changed) {
changed = false;
for (size_t i = 1; i < rpo_.size(); ++i) {
BasicBlock* b = rpo_[i];
BasicBlock* new_idom = nullptr;
// Find first predecessor with a defined idom
for (auto* pred : b->GetPredecessors()) {
if (idom_.find(pred) != idom_.end()) {
new_idom = pred;
break;
}
}
if (new_idom) {
for (auto* pred : b->GetPredecessors()) {
if (pred != new_idom && idom_.find(pred) != idom_.end()) {
// Intersect
auto* finger1 = pred;
auto* finger2 = new_idom;
int finger_iter = 0;
while (finger1 != finger2) {
finger_iter++;
if (finger_iter > 1000) {
std::cerr << "FATAL: DominatorTree finger loop stuck! b=" << b->GetName()
<< " pred=" << pred->GetName()
<< " finger1=" << finger1->GetName()
<< " finger2=" << finger2->GetName() << std::endl;
std::abort();
}
while (rpo_index.at(finger1) > rpo_index.at(finger2)) {
finger1 = idom_.at(finger1);
}
while (rpo_index.at(finger2) > rpo_index.at(finger1)) {
finger2 = idom_.at(finger2);
}
}
new_idom = finger1;
}
}
if (idom_.find(b) == idom_.end() || idom_[b] != new_idom) {
idom_[b] = new_idom;
changed = true;
}
}
}
}
}
void DominatorTree::ComputeDomTree() {
dom_tree_.clear();
for (auto* b : rpo_) {
dom_tree_[b] = {};
}
for (auto* b : rpo_) {
if (b != rpo_.front()) {
auto* parent = idom_[b];
dom_tree_[parent].push_back(b);
}
}
}
void DominatorTree::ComputeDF() {
df_.clear();
for (auto* b : rpo_) {
df_[b] = {};
}
for (auto* b : rpo_) {
if (b->GetPredecessors().size() >= 2) {
for (auto* pred : b->GetPredecessors()) {
auto* runner = pred;
auto* idom_b = idom_[b];
while (runner && runner != idom_b) {
auto idom_it = idom_.find(runner);
if (idom_it == idom_.end()) {
break; // Unreachable predecessor
}
auto* next_runner = idom_it->second;
if (next_runner == runner) {
break; // Reached root / entry
}
auto& runner_df = df_[runner];
if (std::find(runner_df.begin(), runner_df.end(), b) == runner_df.end()) {
runner_df.push_back(b);
}
runner = next_runner;
}
}
}
}
}
BasicBlock* DominatorTree::GetIdom(BasicBlock* bb) const {
auto it = idom_.find(bb);
return it != idom_.end() ? it->second : nullptr;
}
const std::vector<BasicBlock*>& DominatorTree::GetDominatedBlocks(BasicBlock* bb) const {
static const std::vector<BasicBlock*> empty;
auto it = dom_tree_.find(bb);
return it != dom_tree_.end() ? it->second : empty;
}
const std::vector<BasicBlock*>& DominatorTree::GetDominanceFrontier(BasicBlock* bb) const {
static const std::vector<BasicBlock*> empty;
auto it = df_.find(bb);
return it != df_.end() ? it->second : empty;
}
bool DominatorTree::Dominates(BasicBlock* a, BasicBlock* b) const {
if (a == b) return true;
auto* runner = b;
while (runner != rpo_.front()) {
auto it = idom_.find(runner);
if (it == idom_.end()) return false;
runner = it->second;
if (runner == a) return true;
}
return false;
}
} // namespace ir

View File

@@ -0,0 +1,112 @@
#include "ir/PassManager.h"
#include <vector>
namespace ir {
namespace {
bool IsConstInt(Value* value, int expected) {
auto* constant = dynamic_cast<ConstantInt*>(value);
return constant && constant->GetValue() == expected;
}
bool IsConstFloat(Value* value, float expected) {
auto* constant = dynamic_cast<ConstantFloat*>(value);
return constant && constant->GetValue() == expected;
}
bool IsSameValue(Value* lhs, Value* rhs) {
return lhs == rhs;
}
Value* SimplifyBinary(BinaryInst* bin, Context& ctx) {
auto* lhs = bin->GetLhs();
auto* rhs = bin->GetRhs();
switch (bin->GetOpcode()) {
case Opcode::Add:
if (IsConstInt(rhs, 0)) return lhs;
if (IsConstInt(lhs, 0)) return rhs;
break;
case Opcode::Sub:
if (IsConstInt(rhs, 0)) return lhs;
if (IsSameValue(lhs, rhs)) return ctx.GetConstInt(0);
break;
case Opcode::Mul:
if (IsConstInt(rhs, 1)) return lhs;
if (IsConstInt(lhs, 1)) return rhs;
if (IsConstInt(rhs, 0) || IsConstInt(lhs, 0)) return ctx.GetConstInt(0);
break;
case Opcode::Div:
if (IsConstInt(rhs, 1)) return lhs;
if (IsConstInt(lhs, 0)) return ctx.GetConstInt(0);
break;
case Opcode::Mod:
if (IsConstInt(rhs, 1) || IsConstInt(lhs, 0)) return ctx.GetConstInt(0);
break;
case Opcode::FAdd:
if (IsConstFloat(rhs, 0.0f)) return lhs;
if (IsConstFloat(lhs, 0.0f)) return rhs;
break;
case Opcode::FSub:
if (IsConstFloat(rhs, 0.0f)) return lhs;
break;
case Opcode::FMul:
if (IsConstFloat(rhs, 1.0f)) return lhs;
if (IsConstFloat(lhs, 1.0f)) return rhs;
if (IsConstFloat(rhs, 0.0f) || IsConstFloat(lhs, 0.0f)) {
return ctx.GetConstFloat(0.0f);
}
break;
case Opcode::FDiv:
if (IsConstFloat(rhs, 1.0f)) return lhs;
break;
case Opcode::ICmpEQ:
if (IsSameValue(lhs, rhs)) return ctx.GetConstInt(1);
break;
case Opcode::ICmpNE:
if (IsSameValue(lhs, rhs)) return ctx.GetConstInt(0);
break;
case Opcode::ICmpLE:
case Opcode::ICmpGE:
if (IsSameValue(lhs, rhs)) return ctx.GetConstInt(1);
break;
case Opcode::ICmpLT:
case Opcode::ICmpGT:
if (IsSameValue(lhs, rhs)) return ctx.GetConstInt(0);
break;
default:
break;
}
return nullptr;
}
} // namespace
bool RunAlgebraicSimplify(Function* func, Context& ctx) {
bool changed = false;
std::vector<Instruction*> to_erase;
for (const auto& bbPtr : func->GetBlocks()) {
for (const auto& instPtr : bbPtr->GetInstructions()) {
auto* bin = dynamic_cast<BinaryInst*>(instPtr.get());
if (!bin) {
continue;
}
if (auto* replacement = SimplifyBinary(bin, ctx)) {
bin->ReplaceAllUsesWith(replacement);
to_erase.push_back(bin);
changed = true;
}
}
}
for (auto* inst : to_erase) {
inst->GetParent()->EraseInstruction(inst);
}
return changed;
}
} // namespace ir

View File

@@ -1,4 +1,128 @@
// CFG 简化:
// - 删除不可达块、合并空块、简化分支等
// - 改善 IR 结构,便于后续优化与后端生成
#include "ir/PassManager.h"
#include <algorithm>
#include <iostream>
#include <queue>
#include <unordered_set>
#include <vector>
namespace ir {
// Predeclaration of CFG rebuild helper
void RebuildCFG(Function* func);
bool RunCFGSimplify(Function* func) {
bool changed = false;
bool local_changed = true;
while (local_changed) {
local_changed = false;
RebuildCFG(func);
// 1. Remove unreachable basic blocks
BasicBlock* entry = func->GetEntry();
std::unordered_set<BasicBlock*> reachable;
std::queue<BasicBlock*> worklist;
reachable.insert(entry);
worklist.push(entry);
while (!worklist.empty()) {
auto* curr = worklist.front();
worklist.pop();
for (auto* succ : curr->GetSuccessors()) {
if (reachable.find(succ) == reachable.end()) {
reachable.insert(succ);
worklist.push(succ);
}
}
}
std::vector<BasicBlock*> unreachable_blocks;
for (const auto& bbPtr : func->GetBlocks()) {
if (reachable.find(bbPtr.get()) == reachable.end()) {
unreachable_blocks.push_back(bbPtr.get());
}
}
if (!unreachable_blocks.empty()) {
changed = true;
local_changed = true;
for (auto* bb : unreachable_blocks) {
// Remove bb from predecessors of its successors, and clean up successor phi nodes
for (auto* succ : bb->GetSuccessors()) {
for (const auto& instPtr : succ->GetInstructions()) {
if (instPtr->GetOpcode() == Opcode::Phi) {
auto* phi = static_cast<PhiInst*>(instPtr.get());
phi->RemoveIncomingBlock(bb);
}
}
}
// Remove from func's blocks
auto& blocks = const_cast<std::vector<std::unique_ptr<BasicBlock>>&>(func->GetBlocks());
blocks.erase(std::remove_if(blocks.begin(), blocks.end(),
[&](const std::unique_ptr<BasicBlock>& b) {
return b.get() == bb;
}),
blocks.end());
}
continue; // Restart simplification loop safely
}
// 2. Merge basic block B with successor S if S has only one predecessor B
for (const auto& bbPtr : func->GetBlocks()) {
auto* b = bbPtr.get();
if (b->GetSuccessors().size() == 1) {
auto* s = b->GetSuccessors().front();
if (s != entry && s->GetPredecessors().size() == 1) {
changed = true;
local_changed = true;
// Replace all uses of block S as label with block B
s->ReplaceAllUsesWith(b);
// Erase B's terminator (the BranchInst to S)
auto* b_term = b->GetInstructions().back().get();
b->EraseInstruction(b_term);
// For any PhiInst in S: it has exactly 1 incoming value from B.
// Replace all uses of the PhiInst with its single incoming value.
std::vector<Instruction*> phi_to_remove;
for (const auto& instPtr : s->GetInstructions()) {
if (instPtr->GetOpcode() == Opcode::Phi) {
auto* phi = static_cast<PhiInst*>(instPtr.get());
if (phi->GetNumIncoming() > 0) {
phi->ReplaceAllUsesWith(phi->GetIncomingValue(0));
}
phi_to_remove.push_back(phi);
}
}
// Move instructions from S to B
auto& s_insts = const_cast<std::vector<std::unique_ptr<Instruction>>&>(s->GetInstructions());
for (auto& instPtr : s_insts) {
if (std::find(phi_to_remove.begin(), phi_to_remove.end(), instPtr.get()) == phi_to_remove.end()) {
instPtr->SetParent(b);
const_cast<std::vector<std::unique_ptr<Instruction>>&>(b->GetInstructions()).push_back(std::move(instPtr));
}
}
// Clear S's instructions to prevent any dangling or double frees
s_insts.clear();
// Erase S from func's blocks list
auto& blocks = const_cast<std::vector<std::unique_ptr<BasicBlock>>&>(func->GetBlocks());
blocks.erase(std::remove_if(blocks.begin(), blocks.end(),
[&](const std::unique_ptr<BasicBlock>& b) {
return b.get() == s;
}),
blocks.end());
break; // Break to restart loop safely
}
}
}
}
return changed;
}
} // namespace ir

View File

@@ -3,9 +3,11 @@ add_library(ir_passes STATIC
Mem2Reg.cpp
ConstFold.cpp
ConstProp.cpp
AlgebraicSimplify.cpp
CSE.cpp
DCE.cpp
CFGSimplify.cpp
LICM.cpp
)
target_link_libraries(ir_passes PUBLIC

View File

@@ -1,4 +1,109 @@
// 公共子表达式消除CSE
// - 识别并复用重复计算的等价表达式
// - 典型放置在 ConstFold 之后、DCE 之前
// - 当前为 Lab4 的框架占位,具体算法由实验实现
#include "ir/PassManager.h"
#include <iostream>
#include <unordered_map>
#include <vector>
#include <tuple>
namespace ir {
static bool IsEquivalent(Instruction* a, Instruction* b) {
if (a->GetOpcode() != b->GetOpcode()) return false;
if (a->GetNumOperands() != b->GetNumOperands()) return false;
// Skip load, store, alloca, call, phi, branch, ret (since they have side-effects or special states)
switch (a->GetOpcode()) {
case Opcode::Add:
case Opcode::Sub:
case Opcode::Mul:
case Opcode::Div:
case Opcode::Mod:
case Opcode::FAdd:
case Opcode::FSub:
case Opcode::FMul:
case Opcode::FDiv:
case Opcode::ICmpEQ:
case Opcode::ICmpNE:
case Opcode::ICmpLT:
case Opcode::ICmpGT:
case Opcode::ICmpLE:
case Opcode::ICmpGE:
case Opcode::FCmpEQ:
case Opcode::FCmpNE:
case Opcode::FCmpLT:
case Opcode::FCmpGT:
case Opcode::FCmpLE:
case Opcode::FCmpGE:
case Opcode::GEP:
case Opcode::ZExt:
case Opcode::SIToFP:
case Opcode::FPToSI:
break;
default:
return false; // Skip all other opcodes
}
// Compare all operands
for (size_t i = 0; i < a->GetNumOperands(); ++i) {
if (a->GetOperand(i) != b->GetOperand(i)) {
return false;
}
}
return true;
}
bool RunCSE(Function* func) {
bool changed = false;
for (const auto& bbPtr : func->GetBlocks()) {
std::vector<Instruction*> seen_instructions;
std::unordered_map<Value*, Instruction*> available_loads;
std::vector<Instruction*> to_erase;
for (const auto& instPtr : bbPtr->GetInstructions()) {
auto* inst = instPtr.get();
if (inst->GetOpcode() == Opcode::Load) {
auto* load = static_cast<LoadInst*>(inst);
auto it = available_loads.find(load->GetPtr());
if (it != available_loads.end()) {
inst->ReplaceAllUsesWith(it->second);
to_erase.push_back(inst);
changed = true;
continue;
}
available_loads[load->GetPtr()] = inst;
continue;
}
if (inst->GetOpcode() == Opcode::Store ||
inst->GetOpcode() == Opcode::Call) {
available_loads.clear();
}
Instruction* match = nullptr;
for (auto* seen : seen_instructions) {
if (IsEquivalent(inst, seen)) {
match = seen;
break;
}
}
if (match) {
inst->ReplaceAllUsesWith(match);
to_erase.push_back(inst);
changed = true;
} else {
seen_instructions.push_back(inst);
}
}
for (auto* inst : to_erase) {
bbPtr->EraseInstruction(inst);
}
}
return changed;
}
} // namespace ir

View File

@@ -1,4 +1,105 @@
// IR 常量折叠:
// - 折叠可判定的常量表达式
// - 简化常量控制流分支(按实现范围裁剪)
#include "ir/PassManager.h"
#include <iostream>
#include <cmath>
namespace ir {
ConstantValue* FoldInstruction(Instruction* inst, Context& ctx) {
if (inst->GetOpcode() == Opcode::ZExt) {
auto* cast = static_cast<CastInst*>(inst);
if (auto* ci = dynamic_cast<ConstantInt*>(cast->GetValue())) {
return ctx.GetConstInt(ci->GetValue()); // ZExt is trivial on constant int
}
}
if (inst->GetOpcode() == Opcode::SIToFP) {
auto* cast = static_cast<CastInst*>(inst);
if (auto* ci = dynamic_cast<ConstantInt*>(cast->GetValue())) {
return ctx.GetConstFloat(static_cast<float>(ci->GetValue()));
}
}
if (inst->GetOpcode() == Opcode::FPToSI) {
auto* cast = static_cast<CastInst*>(inst);
if (auto* cf = dynamic_cast<ConstantFloat*>(cast->GetValue())) {
return ctx.GetConstInt(static_cast<int>(cf->GetValue()));
}
}
// Binary operations
if (auto* bin = dynamic_cast<BinaryInst*>(inst)) {
auto* lhs = bin->GetLhs();
auto* rhs = bin->GetRhs();
auto* lhs_i = dynamic_cast<ConstantInt*>(lhs);
auto* rhs_i = dynamic_cast<ConstantInt*>(rhs);
auto* lhs_f = dynamic_cast<ConstantFloat*>(lhs);
auto* rhs_f = dynamic_cast<ConstantFloat*>(rhs);
if (lhs_i && rhs_i) {
int l = lhs_i->GetValue();
int r = rhs_i->GetValue();
switch (bin->GetOpcode()) {
case Opcode::Add: return ctx.GetConstInt(l + r);
case Opcode::Sub: return ctx.GetConstInt(l - r);
case Opcode::Mul: return ctx.GetConstInt(l * r);
case Opcode::Div: return (r != 0) ? ctx.GetConstInt(l / r) : nullptr;
case Opcode::Mod: return (r != 0) ? ctx.GetConstInt(l % r) : nullptr;
case Opcode::ICmpEQ: return ctx.GetConstInt(l == r ? 1 : 0);
case Opcode::ICmpNE: return ctx.GetConstInt(l != r ? 1 : 0);
case Opcode::ICmpLT: return ctx.GetConstInt(l < r ? 1 : 0);
case Opcode::ICmpGT: return ctx.GetConstInt(l > r ? 1 : 0);
case Opcode::ICmpLE: return ctx.GetConstInt(l <= r ? 1 : 0);
case Opcode::ICmpGE: return ctx.GetConstInt(l >= r ? 1 : 0);
default: break;
}
}
if (lhs_f && rhs_f) {
float l = lhs_f->GetValue();
float r = rhs_f->GetValue();
switch (bin->GetOpcode()) {
case Opcode::FAdd: return ctx.GetConstFloat(l + r);
case Opcode::FSub: return ctx.GetConstFloat(l - r);
case Opcode::FMul: return ctx.GetConstFloat(l * r);
case Opcode::FDiv: return (r != 0.0f) ? ctx.GetConstFloat(l / r) : nullptr;
case Opcode::FCmpEQ: return ctx.GetConstInt(l == r ? 1 : 0);
case Opcode::FCmpNE: return ctx.GetConstInt(l != r ? 1 : 0);
case Opcode::FCmpLT: return ctx.GetConstInt(l < r ? 1 : 0);
case Opcode::FCmpGT: return ctx.GetConstInt(l > r ? 1 : 0);
case Opcode::FCmpLE: return ctx.GetConstInt(l <= r ? 1 : 0);
case Opcode::FCmpGE: return ctx.GetConstInt(l >= r ? 1 : 0);
default: break;
}
}
}
return nullptr;
}
bool RunConstFold(Function* func, Context& ctx) {
bool changed = false;
std::vector<Instruction*> to_erase;
for (const auto& bbPtr : func->GetBlocks()) {
for (const auto& instPtr : bbPtr->GetInstructions()) {
auto* inst = instPtr.get();
if (inst->GetOpcode() == Opcode::Br || inst->GetOpcode() == Opcode::Ret || inst->GetOpcode() == Opcode::Phi) {
continue;
}
if (auto* folded = FoldInstruction(inst, ctx)) {
inst->ReplaceAllUsesWith(folded);
to_erase.push_back(inst);
changed = true;
}
}
}
for (auto* inst : to_erase) {
inst->GetParent()->EraseInstruction(inst);
}
return changed;
}
} // namespace ir

View File

@@ -1,5 +1,75 @@
// 常量传播Constant Propagation
// - 沿 use-def 关系传播已知常量
// - 将可替换的 SSA 值改写为常量,暴露更多折叠机会
// - 常与 ConstFold、DCE、CFGSimplify 迭代配合使用
#include "ir/PassManager.h"
#include <iostream>
#include <vector>
namespace ir {
// Declare FoldInstruction from ConstFold.cpp
ConstantValue* FoldInstruction(Instruction* inst, Context& ctx);
bool RunConstProp(Function* func, Context& ctx) {
bool changed = false;
bool local_changed = true;
while (local_changed) {
local_changed = false;
std::vector<Instruction*> to_erase;
// 1. Fold instructions
for (const auto& bbPtr : func->GetBlocks()) {
for (const auto& instPtr : bbPtr->GetInstructions()) {
auto* inst = instPtr.get();
if (inst->GetOpcode() == Opcode::Br || inst->GetOpcode() == Opcode::Ret || inst->GetOpcode() == Opcode::Phi) {
continue;
}
if (auto* folded = FoldInstruction(inst, ctx)) {
inst->ReplaceAllUsesWith(folded);
to_erase.push_back(inst);
local_changed = true;
changed = true;
}
}
}
// Erase the folded instructions
for (auto* inst : to_erase) {
inst->GetParent()->EraseInstruction(inst);
}
// 2. Simplify conditional branches
for (const auto& bbPtr : func->GetBlocks()) {
auto* bb = bbPtr.get();
const auto& insts = bb->GetInstructions();
if (insts.empty()) continue;
auto* term = insts.back().get();
if (term->GetOpcode() == Opcode::Br) {
auto* br = static_cast<BranchInst*>(term);
if (br->IsConditional()) {
if (auto* cond_const = dynamic_cast<ConstantInt*>(br->GetCondition())) {
BasicBlock* target = (cond_const->GetValue() != 0) ? br->GetIfTrue() : br->GetIfFalse();
BasicBlock* dead_target = (cond_const->GetValue() != 0) ? br->GetIfFalse() : br->GetIfTrue();
if (dead_target != target) {
for (const auto& instPtr : dead_target->GetInstructions()) {
if (instPtr->GetOpcode() == Opcode::Phi) {
auto* phi = static_cast<PhiInst*>(instPtr.get());
phi->RemoveIncomingBlock(bb);
}
}
}
bb->EraseInstruction(br);
bb->Append<BranchInst>(target);
local_changed = true;
changed = true;
break; // Restart loop to handle CFG shifts safely
}
}
}
}
}
return changed;
}
} // namespace ir

View File

@@ -1,4 +1,75 @@
// 死代码删除DCE
// - 删除无用指令与无用基本块
// - 通常与 CFG 简化配合使用
#include "ir/PassManager.h"
#include <iostream>
#include <unordered_set>
#include <queue>
#include <vector>
namespace ir {
bool RunDCE(Function* func) {
std::unordered_set<Instruction*> live_instructions;
std::queue<Instruction*> worklist;
// 1. Mark inherently live instructions
for (const auto& bbPtr : func->GetBlocks()) {
for (const auto& instPtr : bbPtr->GetInstructions()) {
auto* inst = instPtr.get();
bool inherently_live = false;
switch (inst->GetOpcode()) {
case Opcode::Ret:
case Opcode::Br:
case Opcode::Store:
case Opcode::Call:
inherently_live = true;
break;
default:
break;
}
if (inherently_live) {
live_instructions.insert(inst);
worklist.push(inst);
}
}
}
// 2. Propagate liveness along the def-use chains
while (!worklist.empty()) {
auto* inst = worklist.front();
worklist.pop();
for (size_t i = 0; i < inst->GetNumOperands(); ++i) {
auto* operand = inst->GetOperand(i);
if (auto* op_inst = dynamic_cast<Instruction*>(operand)) {
if (live_instructions.find(op_inst) == live_instructions.end()) {
live_instructions.insert(op_inst);
worklist.push(op_inst);
}
}
}
}
// 3. Sweep dead instructions
bool changed = false;
for (const auto& bbPtr : func->GetBlocks()) {
std::vector<Instruction*> dead_instructions;
for (const auto& instPtr : bbPtr->GetInstructions()) {
auto* inst = instPtr.get();
if (live_instructions.find(inst) == live_instructions.end()) {
dead_instructions.push_back(inst);
}
}
if (!dead_instructions.empty()) {
changed = true;
for (auto* inst : dead_instructions) {
bbPtr->EraseInstruction(inst);
}
}
}
return changed;
}
} // namespace ir

198
src/ir/passes/LICM.cpp Normal file
View File

@@ -0,0 +1,198 @@
#include "ir/PassManager.h"
#include <unordered_set>
#include <unordered_map>
#include <vector>
#include <algorithm>
#include <iostream>
namespace ir {
namespace {
// Helper to perform DFS and gather all blocks in a natural loop
std::unordered_set<BasicBlock*> GetLoopBlocks(BasicBlock* B, BasicBlock* H) {
std::unordered_set<BasicBlock*> loop;
std::vector<BasicBlock*> worklist;
loop.insert(H);
if (B != H) {
loop.insert(B);
worklist.push_back(B);
}
while (!worklist.empty()) {
auto* curr = worklist.back();
worklist.pop_back();
for (auto* pred : curr->GetPredecessors()) {
if (loop.find(pred) == loop.end()) {
loop.insert(pred);
worklist.push_back(pred);
}
}
}
return loop;
}
// Check if an opcode is a pure hoisting candidate (pure arithmetic, comparisons, GEP, casts)
bool IsPureHoistingCandidate(Opcode op) {
switch (op) {
case Opcode::Add:
case Opcode::Sub:
case Opcode::Mul:
case Opcode::ICmpEQ:
case Opcode::ICmpNE:
case Opcode::ICmpLT:
case Opcode::ICmpGT:
case Opcode::ICmpLE:
case Opcode::ICmpGE:
case Opcode::FAdd:
case Opcode::FSub:
case Opcode::FMul:
case Opcode::FDiv:
case Opcode::FCmpEQ:
case Opcode::FCmpNE:
case Opcode::FCmpLT:
case Opcode::FCmpGT:
case Opcode::FCmpLE:
case Opcode::FCmpGE:
case Opcode::ZExt:
case Opcode::SIToFP:
case Opcode::FPToSI:
case Opcode::GEP:
return true;
default:
return false;
}
}
} // namespace
bool RunLICM(Function* func) {
bool changed = false;
// 1. Run DominatorTree Analysis
DominatorTree dom_tree(func);
dom_tree.Run();
// 2. Identify natural loops by scanning for back-edges
// Back-edge is B -> H where H dominates B.
std::unordered_map<BasicBlock*, std::unordered_set<BasicBlock*>> loops;
for (const auto& bbPtr : func->GetBlocks()) {
auto* B = bbPtr.get();
for (auto* H : B->GetSuccessors()) {
if (dom_tree.Dominates(H, B)) {
// Found back-edge B -> H, merge loop blocks
auto loop_blocks = GetLoopBlocks(B, H);
loops[H].insert(loop_blocks.begin(), loop_blocks.end());
}
}
}
// 3. Optimize each identified loop
for (auto& pair : loops) {
BasicBlock* H = pair.first;
const auto& loop_blocks = pair.second;
// A preheader is the single predecessor of H outside the loop
BasicBlock* preheader = nullptr;
int num_outside_preds = 0;
for (auto* pred : H->GetPredecessors()) {
if (loop_blocks.find(pred) == loop_blocks.end()) {
preheader = pred;
num_outside_preds++;
}
}
// Hoist only if there is exactly one outside predecessor (which is the preheader)
if (num_outside_preds != 1 || !preheader) {
continue;
}
// Identify loop-invariant instructions
std::unordered_set<Instruction*> invariant_insts;
std::vector<Instruction*> invariant_order;
bool local_changed = true;
while (local_changed) {
local_changed = false;
for (auto* bb : loop_blocks) {
for (const auto& instPtr : bb->GetInstructions()) {
auto* inst = instPtr.get();
if (invariant_insts.find(inst) != invariant_insts.end()) {
continue; // Already identified
}
if (!IsPureHoistingCandidate(inst->GetOpcode())) {
continue; // Cannot hoist impure instructions (load, store, call, branch)
}
// Check if all operands are loop-invariant
bool all_ops_invariant = true;
for (size_t i = 0; i < inst->GetNumOperands(); ++i) {
auto* op = inst->GetOperand(i);
// Constants are invariant
if (dynamic_cast<ConstantValue*>(op)) {
continue;
}
// Values defined outside the loop are invariant
if (auto* op_inst = dynamic_cast<Instruction*>(op)) {
if (loop_blocks.find(op_inst->GetParent()) == loop_blocks.end()) {
continue;
}
// If defined inside the loop, must be already marked invariant
if (invariant_insts.find(op_inst) != invariant_insts.end()) {
continue;
}
} else {
// Arguments and Globals are always defined outside the loop
continue;
}
all_ops_invariant = false;
break;
}
if (all_ops_invariant) {
invariant_insts.insert(inst);
invariant_order.push_back(inst);
local_changed = true;
changed = true;
}
}
}
}
// Hoist the loop-invariant instructions into the preheader (in dependency order)
for (auto* inst : invariant_order) {
auto& source_insts = const_cast<std::vector<std::unique_ptr<Instruction>>&>(inst->GetParent()->GetInstructions());
auto& preheader_insts = const_cast<std::vector<std::unique_ptr<Instruction>>&>(preheader->GetInstructions());
std::unique_ptr<Instruction> moved_inst;
for (auto it = source_insts.begin(); it != source_insts.end(); ++it) {
if (it->get() == inst) {
moved_inst = std::move(*it);
source_insts.erase(it);
break;
}
}
if (moved_inst) {
moved_inst->SetParent(preheader);
// Insert right before the terminator branch instruction of the preheader block
if (!preheader_insts.empty() && preheader->HasTerminator()) {
auto* term = preheader_insts.back().get();
preheader->InsertInstructionBefore(std::move(moved_inst), term);
} else {
preheader_insts.push_back(std::move(moved_inst));
}
}
}
}
return changed;
}
} // namespace ir

View File

@@ -1,4 +1,228 @@
// Mem2RegSSA 构造):
// - 将局部变量的 alloca/load/store 提升为 SSA 形式
// - 插入 PHI 并重写使用,依赖支配树等分析
#include "ir/PassManager.h"
#include <iostream>
#include <unordered_map>
#include <unordered_set>
#include <vector>
#include <stack>
#include <algorithm>
#include <queue>
#include <functional>
namespace ir {
// Predeclaration of rebuild CFG helper
void RebuildCFG(Function* func);
bool RunMem2Reg(Function* func, Context& ctx) {
// 1. Build dominator tree
DominatorTree dom_tree(func);
dom_tree.Run();
// 2. Identify promotable allocas
std::vector<AllocaInst*> promotable_allocas;
for (const auto& bbPtr : func->GetBlocks()) {
for (const auto& instPtr : bbPtr->GetInstructions()) {
if (instPtr->GetOpcode() == Opcode::Alloca) {
auto* alloca = static_cast<AllocaInst*>(instPtr.get());
// Alloca of scalar type: i32 or float (pointers to i32/float in minimum IR)
if (alloca->GetType()->IsPtrInt32() || alloca->GetType()->IsPtrFloat()) {
// Verify all uses are load/store
bool promotable = true;
for (const auto& use : alloca->GetUses()) {
auto* user = use.GetUser();
auto* inst_user = dynamic_cast<Instruction*>(user);
if (!inst_user) {
promotable = false;
break;
}
if (inst_user->GetOpcode() != Opcode::Load && inst_user->GetOpcode() != Opcode::Store) {
promotable = false;
break;
}
// For Store, alloca must be the pointer operand (operand index 1), not the value operand
if (inst_user->GetOpcode() == Opcode::Store) {
auto* store = static_cast<StoreInst*>(inst_user);
if (store->GetPtr() != alloca) {
promotable = false;
break;
}
}
}
if (promotable) {
promotable_allocas.push_back(alloca);
}
}
}
}
}
if (promotable_allocas.empty()) {
return false;
}
// 3. For each alloca, find definition blocks and place Phi nodes
// Maps each basic block and alloca to the inserted Phi instruction
std::unordered_map<BasicBlock*, std::unordered_map<AllocaInst*, PhiInst*>> phi_nodes;
std::unordered_set<Instruction*> instructions_to_erase;
for (auto* alloca : promotable_allocas) {
std::vector<BasicBlock*> def_blocks;
for (const auto& use : alloca->GetUses()) {
auto* inst = dynamic_cast<Instruction*>(use.GetUser());
if (inst && inst->GetOpcode() == Opcode::Store) {
def_blocks.push_back(inst->GetParent());
}
}
// DF-based Phi placement
std::queue<BasicBlock*> worklist;
std::unordered_set<BasicBlock*> added;
std::unordered_set<BasicBlock*> def_set(def_blocks.begin(), def_blocks.end());
for (auto* bb : def_blocks) {
worklist.push(bb);
added.insert(bb);
}
while (!worklist.empty()) {
auto* x = worklist.front();
worklist.pop();
for (auto* y : dom_tree.GetDominanceFrontier(x)) {
if (added.find(y) == added.end()) {
// Place Phi node in Y
std::shared_ptr<Type> ty = alloca->GetType()->IsPtrFloat() ? Type::GetFloatType() : Type::GetInt32Type();
auto phi = std::make_unique<PhiInst>(ty, ctx.NextTemp());
auto* phi_ptr = phi.get();
// Insert Phi at the start of block Y
y->InsertInstructionAtBegin(std::move(phi));
phi_nodes[y][alloca] = phi_ptr;
added.insert(y);
if (def_set.find(y) == def_set.end()) {
worklist.push(y);
}
}
}
}
}
// 4. Rename variables using DFS traversal of dominator tree
std::unordered_map<AllocaInst*, std::vector<Value*>> current_def;
// Helper for generating default value
auto get_default_value = [&](AllocaInst* alloca) -> Value* {
if (alloca->GetType()->IsPtrFloat()) {
return ctx.GetConstFloat(0.0f);
} else {
return ctx.GetConstInt(0);
}
};
// Traversal stack for DFS: stores (block, parent_block)
struct TraversalNode {
BasicBlock* bb;
size_t child_idx;
};
std::stack<BasicBlock*> visit_stack;
std::unordered_map<BasicBlock*, std::vector<std::pair<AllocaInst*, size_t>>> pushed_defs;
// DFS function
std::function<void(BasicBlock*)> rename_dfs = [&](BasicBlock* bb) {
auto& pushes = pushed_defs[bb];
// Push Phis in this block to current_def
auto phi_it = phi_nodes.find(bb);
if (phi_it != phi_nodes.end()) {
for (const auto& pair : phi_it->second) {
auto* alloca = pair.first;
auto* phi = pair.second;
current_def[alloca].push_back(phi);
pushes.push_back({alloca, 1});
}
}
// Process loads and stores
for (const auto& instPtr : bb->GetInstructions()) {
auto* inst = instPtr.get();
if (inst->GetOpcode() == Opcode::Load) {
auto* load = static_cast<LoadInst*>(inst);
auto* ptr = load->GetPtr();
if (auto* alloca = dynamic_cast<AllocaInst*>(ptr)) {
if (std::find(promotable_allocas.begin(), promotable_allocas.end(), alloca) != promotable_allocas.end()) {
auto& defs = current_def[alloca];
Value* val = defs.empty() ? get_default_value(alloca) : defs.back();
load->ReplaceAllUsesWith(val);
instructions_to_erase.insert(load);
}
}
} else if (inst->GetOpcode() == Opcode::Store) {
auto* store = static_cast<StoreInst*>(inst);
auto* ptr = store->GetPtr();
if (auto* alloca = dynamic_cast<AllocaInst*>(ptr)) {
if (std::find(promotable_allocas.begin(), promotable_allocas.end(), alloca) != promotable_allocas.end()) {
current_def[alloca].push_back(store->GetValue());
pushes.push_back({alloca, 1});
instructions_to_erase.insert(store);
}
}
}
}
// Fill Phi incoming values for CFG successors
for (auto* succ : bb->GetSuccessors()) {
auto succ_phi_it = phi_nodes.find(succ);
if (succ_phi_it != phi_nodes.end()) {
for (const auto& pair : succ_phi_it->second) {
auto* alloca = pair.first;
auto* phi = pair.second;
auto& defs = current_def[alloca];
Value* val = defs.empty() ? get_default_value(alloca) : defs.back();
phi->AddIncoming(val, bb);
}
}
}
// Recurse to dominator tree children
for (auto* child : dom_tree.GetDominatedBlocks(bb)) {
rename_dfs(child);
}
// Pop definitions pushed in this block
for (const auto& push : pushes) {
auto* alloca = push.first;
for (size_t k = 0; k < push.second; ++k) {
if (!current_def[alloca].empty()) {
current_def[alloca].pop_back();
}
}
}
};
if (!func->GetBlocks().empty()) {
rename_dfs(func->GetEntry());
}
// 5. Clean up loads, stores and allocas
for (auto* alloca : promotable_allocas) {
instructions_to_erase.insert(alloca);
}
for (const auto& bbPtr : func->GetBlocks()) {
std::vector<Instruction*> to_remove;
for (const auto& instPtr : bbPtr->GetInstructions()) {
if (instructions_to_erase.find(instPtr.get()) != instructions_to_erase.end()) {
to_remove.push_back(instPtr.get());
}
}
for (auto* inst : to_remove) {
bbPtr->EraseInstruction(inst);
}
}
return true;
}
} // namespace ir

View File

@@ -1 +1,35 @@
// IR Pass 管理骨架。
#include "ir/PassManager.h"
#include <iostream>
namespace ir {
void RunFunctionOptimizationPasses(Function* func, Context& ctx) {
RunMem2Reg(func, ctx);
bool changed = true;
int iterations = 0;
const int max_iterations = 16;
while (changed && iterations < max_iterations) {
changed = false;
iterations++;
changed |= RunConstProp(func, ctx);
changed |= RunConstFold(func, ctx);
changed |= RunAlgebraicSimplify(func, ctx);
changed |= RunCSE(func);
changed |= RunLICM(func);
changed |= RunDCE(func);
changed |= RunCFGSimplify(func);
}
}
void RunOptimizationPasses(Module& module) {
for (const auto& funcPtr : module.GetFunctions()) {
if (!funcPtr->GetBlocks().empty()) {
RunFunctionOptimizationPasses(funcPtr.get(), module.GetContext());
}
}
}
} // namespace ir

View File

@@ -208,8 +208,10 @@ std::any IRGenImpl::visitVarDef(SysYParser::VarDefContext* ctx) {
slot = module_.CreateGlobalValue(name, StorageType(ty), init);
} else {
slot = builder_.CreateAlloca(StorageType(ty), name);
ZeroInitializeLocal(slot, ty);
if (ctx->initValue()) EmitLocalInitValue(slot, ty, ctx->initValue());
if (ctx->initValue()) {
ZeroInitializeLocal(slot, ty);
EmitLocalInitValue(slot, ty, ctx->initValue());
}
}
storage_map_[ctx] = slot;

View File

@@ -88,7 +88,7 @@ ir::ConstantValue* IRGenImpl::EvalConstExpr(SysYParser::ExpContext& expr) {
return static_cast<ir::ConstantValue*>(module_.GetContext().GetConstInt(value));
}
return static_cast<ir::ConstantValue*>(
module_.GetContext().GetConstFloat(std::stof(ctx->number()->FLITERAL()->getText())));
module_.GetContext().GetConstFloat(static_cast<float>(std::stod(ctx->number()->FLITERAL()->getText()))));
}
std::any visitLValueExp(SysYParser::LValueExpContext* ctx) override {
@@ -105,7 +105,17 @@ ir::ConstantValue* IRGenImpl::EvalConstExpr(SysYParser::ExpContext& expr) {
throw std::runtime_error(
FormatError("irgen", "常量缺少标量初始化表达式"));
}
return Eval(*const_def->initValue()->exp());
auto* init = Eval(*const_def->initValue()->exp());
auto* decl = dynamic_cast<SysYParser::ConstDeclContext*>(const_def->parent);
bool is_float = (decl && decl->btype() && decl->btype()->FLOAT());
if (!is_float && init->GetType()->IsFloat()) {
init = module_.GetContext().GetConstInt(
static_cast<int>(static_cast<ir::ConstantFloat*>(init)->GetValue()));
} else if (is_float && init->GetType()->IsInt32()) {
init = module_.GetContext().GetConstFloat(
static_cast<float>(static_cast<ir::ConstantInt*>(init)->GetValue()));
}
return init;
}
throw std::runtime_error(
FormatError("irgen", "全局/常量表达式必须是编译期常量"));
@@ -130,76 +140,59 @@ ir::ConstantValue* IRGenImpl::EvalConstExpr(SysYParser::ExpContext& expr) {
module_.GetContext().GetConstInt(IsTruthy(Eval(*ctx->exp())) ? 0 : 1));
}
std::any visitAddExp(SysYParser::AddExpContext* ctx) override {
std::any visitMulDivModExp(SysYParser::MulDivModExpContext* ctx) override {
auto* lhs = Eval(*ctx->exp(0));
auto* rhs = Eval(*ctx->exp(1));
if (lhs->GetType()->IsFloat() || rhs->GetType()->IsFloat()) {
return static_cast<ir::ConstantValue*>(
module_.GetContext().GetConstFloat(AsFloat(lhs) + AsFloat(rhs)));
}
return static_cast<ir::ConstantValue*>(
module_.GetContext().GetConstInt(AsInt(lhs) + AsInt(rhs)));
}
bool is_mul = ctx->MUL() != nullptr;
bool is_div = ctx->DIV() != nullptr;
bool is_mod = ctx->MOD() != nullptr;
std::any visitSubExp(SysYParser::SubExpContext* ctx) override {
auto* lhs = Eval(*ctx->exp(0));
auto* rhs = Eval(*ctx->exp(1));
if (lhs->GetType()->IsFloat() || rhs->GetType()->IsFloat()) {
return static_cast<ir::ConstantValue*>(
module_.GetContext().GetConstFloat(AsFloat(lhs) - AsFloat(rhs)));
if (is_mod) {
return static_cast<ir::ConstantValue*>(module_.GetContext().GetConstInt(
AsInt(rhs) == 0 ? 0 : AsInt(lhs) % AsInt(rhs)));
}
return static_cast<ir::ConstantValue*>(
module_.GetContext().GetConstInt(AsInt(lhs) - AsInt(rhs)));
}
std::any visitMulExp(SysYParser::MulExpContext* ctx) override {
auto* lhs = Eval(*ctx->exp(0));
auto* rhs = Eval(*ctx->exp(1));
if (lhs->GetType()->IsFloat() || rhs->GetType()->IsFloat()) {
return static_cast<ir::ConstantValue*>(
module_.GetContext().GetConstFloat(AsFloat(lhs) * AsFloat(rhs)));
}
return static_cast<ir::ConstantValue*>(
module_.GetContext().GetConstInt(AsInt(lhs) * AsInt(rhs)));
}
std::any visitDivExp(SysYParser::DivExpContext* ctx) override {
auto* lhs = Eval(*ctx->exp(0));
auto* rhs = Eval(*ctx->exp(1));
if (lhs->GetType()->IsFloat() || rhs->GetType()->IsFloat()) {
const float lv = AsFloat(lhs);
const float rv = AsFloat(rhs);
return static_cast<ir::ConstantValue*>(module_.GetContext().GetConstFloat(
rv == 0.0f ? 0.0f : AsFloat(lhs) / rv));
if (is_mul) return static_cast<ir::ConstantValue*>(module_.GetContext().GetConstFloat(lv * rv));
else return static_cast<ir::ConstantValue*>(module_.GetContext().GetConstFloat(rv == 0.0f ? 0.0f : lv / rv));
}
const int lv = AsInt(lhs);
const int rv = AsInt(rhs);
return static_cast<ir::ConstantValue*>(
module_.GetContext().GetConstInt(rv == 0 ? 0 : AsInt(lhs) / rv));
if (is_mul) return static_cast<ir::ConstantValue*>(module_.GetContext().GetConstInt(lv * rv));
else return static_cast<ir::ConstantValue*>(module_.GetContext().GetConstInt(rv == 0 ? 0 : lv / rv));
}
std::any visitModExp(SysYParser::ModExpContext* ctx) override {
std::any visitAddSubExp(SysYParser::AddSubExpContext* ctx) override {
auto* lhs = Eval(*ctx->exp(0));
auto* rhs = Eval(*ctx->exp(1));
return static_cast<ir::ConstantValue*>(module_.GetContext().GetConstInt(
AsInt(rhs) == 0 ? 0 : AsInt(lhs) % AsInt(rhs)));
bool is_sub = ctx->SUB() != nullptr;
if (lhs->GetType()->IsFloat() || rhs->GetType()->IsFloat()) {
const float lv = AsFloat(lhs);
const float rv = AsFloat(rhs);
return static_cast<ir::ConstantValue*>(module_.GetContext().GetConstFloat(is_sub ? lv - rv : lv + rv));
}
const int lv = AsInt(lhs);
const int rv = AsInt(rhs);
return static_cast<ir::ConstantValue*>(module_.GetContext().GetConstInt(is_sub ? lv - rv : lv + rv));
}
std::any visitLtExp(SysYParser::LtExpContext* ctx) override {
return EvalCmpImpl(*ctx->exp(0), *ctx->exp(1), ir::Opcode::ICmpLT);
std::any visitRelExp(SysYParser::RelExpContext* ctx) override {
ir::Opcode op = ir::Opcode::ICmpLT;
if (ctx->LT()) op = ir::Opcode::ICmpLT;
else if (ctx->LE()) op = ir::Opcode::ICmpLE;
else if (ctx->GT()) op = ir::Opcode::ICmpGT;
else if (ctx->GE()) op = ir::Opcode::ICmpGE;
return EvalCmpImpl(*ctx->exp(0), *ctx->exp(1), op);
}
std::any visitLeExp(SysYParser::LeExpContext* ctx) override {
return EvalCmpImpl(*ctx->exp(0), *ctx->exp(1), ir::Opcode::ICmpLE);
}
std::any visitGtExp(SysYParser::GtExpContext* ctx) override {
return EvalCmpImpl(*ctx->exp(0), *ctx->exp(1), ir::Opcode::ICmpGT);
}
std::any visitGeExp(SysYParser::GeExpContext* ctx) override {
return EvalCmpImpl(*ctx->exp(0), *ctx->exp(1), ir::Opcode::ICmpGE);
}
std::any visitEqExp(SysYParser::EqExpContext* ctx) override {
return EvalCmpImpl(*ctx->exp(0), *ctx->exp(1), ir::Opcode::ICmpEQ);
}
std::any visitNeExp(SysYParser::NeExpContext* ctx) override {
return EvalCmpImpl(*ctx->exp(0), *ctx->exp(1), ir::Opcode::ICmpNE);
std::any visitEqNeExp(SysYParser::EqNeExpContext* ctx) override {
ir::Opcode op = ir::Opcode::ICmpEQ;
if (ctx->EQ()) op = ir::Opcode::ICmpEQ;
else if (ctx->NE()) op = ir::Opcode::ICmpNE;
return EvalCmpImpl(*ctx->exp(0), *ctx->exp(1), op);
}
std::any visitAndExp(SysYParser::AndExpContext* ctx) override {
@@ -422,53 +415,165 @@ std::any IRGenImpl::visitUnarySubExp(SysYParser::UnarySubExpContext* ctx) {
return static_cast<ir::Value*>(builder_.CreateBinary(ir::Opcode::int_opcode, lhs, rhs, module_.GetContext().NextTemp())); \
}
DEFINE_ARITH_VISITOR(Add, Add, FAdd)
DEFINE_ARITH_VISITOR(Sub, Sub, FSub)
DEFINE_ARITH_VISITOR(Mul, Mul, FMul)
DEFINE_ARITH_VISITOR(Div, Div, FDiv)
std::any IRGenImpl::visitMulDivModExp(SysYParser::MulDivModExpContext* ctx) {
ir::Value* lhs = EvalExpr(*ctx->exp(0));
ir::Value* rhs = EvalExpr(*ctx->exp(1));
bool is_mul = ctx->MUL() != nullptr;
bool is_div = ctx->DIV() != nullptr;
bool is_mod = ctx->MOD() != nullptr;
if (is_mod) {
lhs = CastValue(*this, builder_, module_, lhs, ir::Type::GetInt32Type());
rhs = CastValue(*this, builder_, module_, rhs, ir::Type::GetInt32Type());
if (auto* lconst = dynamic_cast<ir::ConstantValue*>(lhs)) {
if (auto* rconst = dynamic_cast<ir::ConstantValue*>(rhs)) {
const int rv = AsInt(rconst);
return static_cast<ir::Value*>(module_.GetContext().GetConstInt(rv == 0 ? 0 : AsInt(lconst) % rv));
}
}
return static_cast<ir::Value*>(builder_.CreateMod(lhs, rhs, module_.GetContext().NextTemp()));
}
const auto common_ty = CommonArithType(lhs, rhs);
lhs = CastValue(*this, builder_, module_, lhs, common_ty);
rhs = CastValue(*this, builder_, module_, rhs, common_ty);
std::any IRGenImpl::visitModExp(SysYParser::ModExpContext* ctx) {
ir::Value* lhs = CastValue(*this, builder_, module_, EvalExpr(*ctx->exp(0)),
ir::Type::GetInt32Type());
ir::Value* rhs = CastValue(*this, builder_, module_, EvalExpr(*ctx->exp(1)),
ir::Type::GetInt32Type());
if (auto* lconst = dynamic_cast<ir::ConstantValue*>(lhs)) {
if (auto* rconst = dynamic_cast<ir::ConstantValue*>(rhs)) {
if (common_ty->IsFloat()) {
const float lv = AsFloat(lconst);
const float rv = AsFloat(rconst);
if (is_mul) return static_cast<ir::Value*>(module_.GetContext().GetConstFloat(lv * rv));
else return static_cast<ir::Value*>(module_.GetContext().GetConstFloat(rv == 0.0f ? 0.0f : lv / rv));
}
const int lv = AsInt(lconst);
const int rv = AsInt(rconst);
return static_cast<ir::Value*>(
module_.GetContext().GetConstInt(rv == 0 ? 0 : AsInt(lconst) % rv));
if (is_mul) return static_cast<ir::Value*>(module_.GetContext().GetConstInt(lv * rv));
else return static_cast<ir::Value*>(module_.GetContext().GetConstInt(rv == 0 ? 0 : lv / rv));
}
}
return static_cast<ir::Value*>(
builder_.CreateMod(lhs, rhs, module_.GetContext().NextTemp()));
if (common_ty->IsFloat()) {
if (is_mul) return static_cast<ir::Value*>(builder_.CreateFMul(lhs, rhs, module_.GetContext().NextTemp()));
else return static_cast<ir::Value*>(builder_.CreateFDiv(lhs, rhs, module_.GetContext().NextTemp()));
}
if (is_mul) return static_cast<ir::Value*>(builder_.CreateBinary(ir::Opcode::Mul, lhs, rhs, module_.GetContext().NextTemp()));
else return static_cast<ir::Value*>(builder_.CreateBinary(ir::Opcode::Div, lhs, rhs, module_.GetContext().NextTemp()));
}
#define DEFINE_CMP_VISITOR(name, int_opcode, float_opcode, cmp_op) \
std::any IRGenImpl::visit##name##Exp(SysYParser::name##ExpContext* ctx) { \
ir::Value* lhs = EvalExpr(*ctx->exp(0)); \
ir::Value* rhs = EvalExpr(*ctx->exp(1)); \
const auto common_ty = CommonArithType(lhs, rhs); \
lhs = CastValue(*this, builder_, module_, lhs, common_ty); \
rhs = CastValue(*this, builder_, module_, rhs, common_ty); \
if (auto* lconst = dynamic_cast<ir::ConstantValue*>(lhs)) { \
if (auto* rconst = dynamic_cast<ir::ConstantValue*>(rhs)) { \
const bool result = common_ty->IsFloat() ? (AsFloat(lconst) cmp_op AsFloat(rconst)) \
: (AsInt(lconst) cmp_op AsInt(rconst)); \
return static_cast<ir::Value*>(module_.GetContext().GetConstInt(result ? 1 : 0)); \
} \
} \
if (common_ty->IsFloat()) { \
return static_cast<ir::Value*>(builder_.CreateFCmp(ir::Opcode::float_opcode, lhs, rhs, module_.GetContext().NextTemp())); \
} \
return static_cast<ir::Value*>(builder_.CreateICmp(ir::Opcode::int_opcode, lhs, rhs, module_.GetContext().NextTemp())); \
std::any IRGenImpl::visitAddSubExp(SysYParser::AddSubExpContext* ctx) {
ir::Value* lhs = EvalExpr(*ctx->exp(0));
ir::Value* rhs = EvalExpr(*ctx->exp(1));
const auto common_ty = CommonArithType(lhs, rhs);
lhs = CastValue(*this, builder_, module_, lhs, common_ty);
rhs = CastValue(*this, builder_, module_, rhs, common_ty);
bool is_sub = ctx->SUB() != nullptr;
if (auto* lconst = dynamic_cast<ir::ConstantValue*>(lhs)) {
if (auto* rconst = dynamic_cast<ir::ConstantValue*>(rhs)) {
if (common_ty->IsFloat()) {
const float lv = AsFloat(lconst);
const float rv = AsFloat(rconst);
return static_cast<ir::Value*>(module_.GetContext().GetConstFloat(is_sub ? lv - rv : lv + rv));
}
const int lv = AsInt(lconst);
const int rv = AsInt(rconst);
return static_cast<ir::Value*>(module_.GetContext().GetConstInt(is_sub ? lv - rv : lv + rv));
}
}
DEFINE_CMP_VISITOR(Lt, ICmpLT, FCmpLT, <)
DEFINE_CMP_VISITOR(Le, ICmpLE, FCmpLE, <=)
DEFINE_CMP_VISITOR(Gt, ICmpGT, FCmpGT, >)
DEFINE_CMP_VISITOR(Ge, ICmpGE, FCmpGE, >=)
DEFINE_CMP_VISITOR(Eq, ICmpEQ, FCmpEQ, ==)
DEFINE_CMP_VISITOR(Ne, ICmpNE, FCmpNE, !=)
if (common_ty->IsFloat()) {
if (is_sub) return static_cast<ir::Value*>(builder_.CreateFSub(lhs, rhs, module_.GetContext().NextTemp()));
else return static_cast<ir::Value*>(builder_.CreateFAdd(lhs, rhs, module_.GetContext().NextTemp()));
}
if (is_sub) return static_cast<ir::Value*>(builder_.CreateBinary(ir::Opcode::Sub, lhs, rhs, module_.GetContext().NextTemp()));
else return static_cast<ir::Value*>(builder_.CreateBinary(ir::Opcode::Add, lhs, rhs, module_.GetContext().NextTemp()));
}
std::any IRGenImpl::visitRelExp(SysYParser::RelExpContext* ctx) {
ir::Value* lhs = EvalExpr(*ctx->exp(0));
ir::Value* rhs = EvalExpr(*ctx->exp(1));
const auto common_ty = CommonArithType(lhs, rhs);
lhs = CastValue(*this, builder_, module_, lhs, common_ty);
rhs = CastValue(*this, builder_, module_, rhs, common_ty);
ir::Opcode int_op = ir::Opcode::ICmpLT;
ir::Opcode float_op = ir::Opcode::FCmpLT;
bool is_lt = ctx->LT() != nullptr;
bool is_le = ctx->LE() != nullptr;
bool is_gt = ctx->GT() != nullptr;
bool is_ge = ctx->GE() != nullptr;
if (is_lt) { int_op = ir::Opcode::ICmpLT; float_op = ir::Opcode::FCmpLT; }
else if (is_le) { int_op = ir::Opcode::ICmpLE; float_op = ir::Opcode::FCmpLE; }
else if (is_gt) { int_op = ir::Opcode::ICmpGT; float_op = ir::Opcode::FCmpGT; }
else if (is_ge) { int_op = ir::Opcode::ICmpGE; float_op = ir::Opcode::FCmpGE; }
if (auto* lconst = dynamic_cast<ir::ConstantValue*>(lhs)) {
if (auto* rconst = dynamic_cast<ir::ConstantValue*>(rhs)) {
bool result = false;
if (common_ty->IsFloat()) {
float lv = AsFloat(lconst);
float rv = AsFloat(rconst);
if (is_lt) result = lv < rv;
else if (is_le) result = lv <= rv;
else if (is_gt) result = lv > rv;
else if (is_ge) result = lv >= rv;
} else {
int lv = AsInt(lconst);
int rv = AsInt(rconst);
if (is_lt) result = lv < rv;
else if (is_le) result = lv <= rv;
else if (is_gt) result = lv > rv;
else if (is_ge) result = lv >= rv;
}
return static_cast<ir::Value*>(module_.GetContext().GetConstInt(result ? 1 : 0));
}
}
if (common_ty->IsFloat()) {
return static_cast<ir::Value*>(builder_.CreateFCmp(float_op, lhs, rhs, module_.GetContext().NextTemp()));
}
return static_cast<ir::Value*>(builder_.CreateICmp(int_op, lhs, rhs, module_.GetContext().NextTemp()));
}
std::any IRGenImpl::visitEqNeExp(SysYParser::EqNeExpContext* ctx) {
ir::Value* lhs = EvalExpr(*ctx->exp(0));
ir::Value* rhs = EvalExpr(*ctx->exp(1));
const auto common_ty = CommonArithType(lhs, rhs);
lhs = CastValue(*this, builder_, module_, lhs, common_ty);
rhs = CastValue(*this, builder_, module_, rhs, common_ty);
ir::Opcode int_op = ir::Opcode::ICmpEQ;
ir::Opcode float_op = ir::Opcode::FCmpEQ;
bool is_eq = ctx->EQ() != nullptr;
if (is_eq) { int_op = ir::Opcode::ICmpEQ; float_op = ir::Opcode::FCmpEQ; }
else { int_op = ir::Opcode::ICmpNE; float_op = ir::Opcode::FCmpNE; }
if (auto* lconst = dynamic_cast<ir::ConstantValue*>(lhs)) {
if (auto* rconst = dynamic_cast<ir::ConstantValue*>(rhs)) {
bool result = false;
if (common_ty->IsFloat()) {
float lv = AsFloat(lconst);
float rv = AsFloat(rconst);
result = is_eq ? (lv == rv) : (lv != rv);
} else {
int lv = AsInt(lconst);
int rv = AsInt(rconst);
result = is_eq ? (lv == rv) : (lv != rv);
}
return static_cast<ir::Value*>(module_.GetContext().GetConstInt(result ? 1 : 0));
}
}
if (common_ty->IsFloat()) {
return static_cast<ir::Value*>(builder_.CreateFCmp(float_op, lhs, rhs, module_.GetContext().NextTemp()));
}
return static_cast<ir::Value*>(builder_.CreateICmp(int_op, lhs, rhs, module_.GetContext().NextTemp()));
}
std::any IRGenImpl::visitAndExp(SysYParser::AndExpContext* ctx) {
if (!builder_.GetInsertBlock()) {
@@ -601,7 +706,8 @@ ir::Value* IRGenImpl::DecayArrayPtr(SysYParser::LValueContext* ctx) {
const auto base_ty = GetDefType(def);
if (dynamic_cast<SysYParser::FuncFParamContext*>(def)) {
if (ctx->exp().empty()) return base_ptr;
ir::Value* loaded_base = builder_.CreateLoad(base_ptr, module_.GetContext().NextTemp());
if (ctx->exp().empty()) return loaded_base;
ir::Value* offset = CastValue(*this, builder_, module_, EvalExpr(*ctx->exp(0)),
ir::Type::GetInt32Type());
@@ -618,7 +724,7 @@ ir::Value* IRGenImpl::DecayArrayPtr(SysYParser::LValueContext* ctx) {
module_.GetContext().NextTemp());
cur_ty = arr_ty->GetElementType();
}
return builder_.CreateGEP(ScalarPointerType(cur_ty), base_ptr, {offset},
return builder_.CreateGEP(ScalarPointerType(cur_ty), loaded_base, {offset},
module_.GetContext().NextTemp());
}

View File

@@ -6,6 +6,7 @@
#include "frontend/SyntaxTreePrinter.h"
#if !COMPILER_PARSE_ONLY
#include "ir/IR.h"
#include "ir/PassManager.h"
#include "irgen/IRGen.h"
#include "mir/MIR.h"
#include "sem/Sema.h"
@@ -36,6 +37,7 @@ int main(int argc, char** argv) {
auto sema = RunSema(*comp_unit);
auto module = GenerateIR(*comp_unit, sema);
ir::RunOptimizationPasses(*module);
if (opts.emit_ir) {
ir::IRPrinter printer;
if (need_blank_line) {
@@ -46,13 +48,18 @@ int main(int argc, char** argv) {
}
if (opts.emit_asm) {
auto machine_func = mir::LowerToMIR(*module);
mir::RunRegAlloc(*machine_func);
mir::RunFrameLowering(*machine_func);
if (need_blank_line) {
std::cout << "\n";
mir::PrintGlobals(*module, std::cout);
auto machine_funcs = mir::LowerToMIR(*module);
for (auto& machine_func : machine_funcs) {
mir::RunRegAlloc(*machine_func);
mir::RunFrameLowering(*machine_func);
mir::RunPeephole(*machine_func);
if (need_blank_line) {
std::cout << "\n";
}
mir::PrintAsm(*machine_func, std::cout);
need_blank_line = true;
}
mir::PrintAsm(*machine_func, std::cout);
}
#else
if (opts.emit_ir || opts.emit_asm) {

View File

@@ -1,7 +1,11 @@
#include "mir/MIR.h"
#include "ir/IR.h"
#include <ostream>
#include <stdexcept>
#include <cstdint>
#include <vector>
#include <cstring>
#include "utils/Log.h"
@@ -16,10 +20,48 @@ const FrameSlot& GetFrameSlot(const MachineFunction& function,
return function.GetFrameSlot(operand.GetFrameIndex());
}
bool IsFloatReg(PhysReg reg) {
return reg >= PhysReg::S0 && reg <= PhysReg::S15;
}
void PrintStackAccess(std::ostream& os, const char* mnemonic, PhysReg reg,
int offset) {
os << " " << mnemonic << " " << PhysRegName(reg) << ", [x29, #" << offset
<< "]\n";
int offset, int frame_size) {
bool is_float = IsFloatReg(reg);
const char* ldr_cmd = is_float ? "ldr" : "ldr";
const char* str_cmd = is_float ? "str" : "str";
const char* base_mnemonic = (std::strcmp(mnemonic, "ldur") == 0) ? ldr_cmd : str_cmd;
if (offset >= -256 && offset <= 255) {
if (is_float) {
os << " " << base_mnemonic << " " << PhysRegName(reg) << ", [x29, #" << offset << "]\n";
} else {
os << " " << mnemonic << " " << PhysRegName(reg) << ", [x29, #" << offset << "]\n";
}
} else {
int sp_offset = frame_size + offset;
int access_size = 4;
if ((reg >= PhysReg::X0 && reg <= PhysReg::X28) ||
reg == PhysReg::X29 || reg == PhysReg::X30 ||
reg == PhysReg::SP) {
access_size = 8;
}
int max_offset = access_size == 8 ? 32760 : 16380;
if (sp_offset >= 0 && sp_offset <= max_offset &&
sp_offset % access_size == 0) {
os << " " << base_mnemonic << " " << PhysRegName(reg)
<< ", [sp, #" << sp_offset << "]\n";
} else {
os << " ldr x10, =" << offset << "\n";
os << " " << base_mnemonic << " " << PhysRegName(reg) << ", [x29, x10]\n";
}
}
}
std::string GetBlockLabel(const std::string& func_name, const std::string& block_name) {
if (block_name == "entry" || block_name.empty()) {
return func_name;
}
return ".L_" + func_name + "_" + block_name;
}
} // namespace
@@ -28,51 +70,313 @@ void PrintAsm(const MachineFunction& function, std::ostream& os) {
os << ".text\n";
os << ".global " << function.GetName() << "\n";
os << ".type " << function.GetName() << ", %function\n";
os << function.GetName() << ":\n";
for (const auto& inst : function.GetEntry().GetInstructions()) {
const auto& ops = inst.GetOperands();
switch (inst.GetOpcode()) {
case Opcode::Prologue:
os << " stp x29, x30, [sp, #-16]!\n";
os << " mov x29, sp\n";
if (function.GetFrameSize() > 0) {
os << " sub sp, sp, #" << function.GetFrameSize() << "\n";
struct FloatConstant {
std::string label;
float value;
};
std::vector<FloatConstant> float_constants;
for (size_t b = 0; b < function.GetBlocks().size(); ++b) {
const auto& block = function.GetBlocks()[b];
// Print the block label
if (b == 0) {
os << function.GetName() << ":\n";
} else {
os << GetBlockLabel(function.GetName(), block.GetName()) << ":\n";
}
for (const auto& inst : block.GetInstructions()) {
const auto& ops = inst.GetOperands();
switch (inst.GetOpcode()) {
case Opcode::Prologue:
os << " stp x29, x30, [sp, #-16]!\n";
os << " mov x29, sp\n";
if (function.GetFrameSize() > 0) {
int size = function.GetFrameSize();
if (size <= 4095) {
os << " sub sp, sp, #" << size << "\n";
} else {
os << " ldr x9, =" << size << "\n";
os << " sub sp, sp, x9\n";
}
}
break;
case Opcode::Epilogue:
if (function.GetFrameSize() > 0) {
int size = function.GetFrameSize();
if (size <= 4095) {
os << " add sp, sp, #" << size << "\n";
} else {
os << " ldr x9, =" << size << "\n";
os << " add sp, sp, x9\n";
}
}
os << " ldp x29, x30, [sp], #16\n";
break;
case Opcode::MovImm: {
PhysReg dst = ops.at(0).GetReg();
if (IsFloatReg(dst)) {
// Load float constant
int bits = ops.at(1).GetImm();
float val;
std::memcpy(&val, &bits, sizeof(float));
std::string flabel = ".LC_" + function.GetName() + "_" + std::to_string(float_constants.size());
float_constants.push_back({flabel, val});
os << " adrp x8, " << flabel << "\n";
os << " ldr " << PhysRegName(dst) << ", [x8, :lo12:" << flabel << "]\n";
} else {
int imm = ops.at(1).GetImm();
if (imm >= 0 && imm <= 65535) {
os << " mov " << PhysRegName(dst) << ", #" << imm << "\n";
} else {
os << " ldr " << PhysRegName(dst) << ", =" << imm << "\n";
}
}
break;
}
break;
case Opcode::Epilogue:
if (function.GetFrameSize() > 0) {
os << " add sp, sp, #" << function.GetFrameSize() << "\n";
case Opcode::LoadStack: {
const auto& slot = GetFrameSlot(function, ops.at(1));
PrintStackAccess(os, "ldur", ops.at(0).GetReg(), slot.offset,
function.GetFrameSize());
break;
}
os << " ldp x29, x30, [sp], #16\n";
break;
case Opcode::MovImm:
os << " mov " << PhysRegName(ops.at(0).GetReg()) << ", #"
<< ops.at(1).GetImm() << "\n";
break;
case Opcode::LoadStack: {
const auto& slot = GetFrameSlot(function, ops.at(1));
PrintStackAccess(os, "ldur", ops.at(0).GetReg(), slot.offset);
break;
case Opcode::StoreStack: {
const auto& slot = GetFrameSlot(function, ops.at(1));
PrintStackAccess(os, "stur", ops.at(0).GetReg(), slot.offset,
function.GetFrameSize());
break;
}
case Opcode::AddRR:
os << " add " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << ", "
<< PhysRegName(ops.at(2).GetReg()) << "\n";
break;
case Opcode::SubRR:
os << " sub " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << ", "
<< PhysRegName(ops.at(2).GetReg()) << "\n";
break;
case Opcode::MulRR:
os << " mul " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << ", "
<< PhysRegName(ops.at(2).GetReg()) << "\n";
break;
case Opcode::SDivRR:
os << " sdiv " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << ", "
<< PhysRegName(ops.at(2).GetReg()) << "\n";
break;
case Opcode::MSubRRRR:
os << " msub " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << ", "
<< PhysRegName(ops.at(2).GetReg()) << ", "
<< PhysRegName(ops.at(3).GetReg()) << "\n";
break;
case Opcode::FAddRRR:
os << " fadd " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << ", "
<< PhysRegName(ops.at(2).GetReg()) << "\n";
break;
case Opcode::FSubRRR:
os << " fsub " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << ", "
<< PhysRegName(ops.at(2).GetReg()) << "\n";
break;
case Opcode::FMulRRR:
os << " fmul " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << ", "
<< PhysRegName(ops.at(2).GetReg()) << "\n";
break;
case Opcode::FDivRRR:
os << " fdiv " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << ", "
<< PhysRegName(ops.at(2).GetReg()) << "\n";
break;
case Opcode::CmpRR:
os << " cmp " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << "\n";
break;
case Opcode::FCmpRR:
os << " fcmp " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << "\n";
break;
case Opcode::Cset:
os << " cset " << PhysRegName(ops.at(0).GetReg()) << ", "
<< ops.at(1).GetCondCode() << "\n";
break;
case Opcode::B:
os << " b " << GetBlockLabel(function.GetName(), ops.at(0).GetLabelName()) << "\n";
break;
case Opcode::BCond:
os << " b." << ops.at(0).GetCondCode() << " "
<< GetBlockLabel(function.GetName(), ops.at(1).GetLabelName()) << "\n";
break;
case Opcode::Call:
os << " bl " << ops.at(0).GetGlobalName() << "\n";
break;
case Opcode::Ret:
os << " ret\n";
break;
case Opcode::MovReg:
if (IsFloatReg(ops.at(0).GetReg()) || IsFloatReg(ops.at(1).GetReg())) {
os << " fmov " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << "\n";
} else {
os << " mov " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << "\n";
}
break;
case Opcode::Adrp:
os << " adrp " << PhysRegName(ops.at(0).GetReg()) << ", "
<< ops.at(1).GetGlobalName() << "\n";
break;
case Opcode::AddRegImm: {
PhysReg dst = ops.at(0).GetReg();
PhysReg src = ops.at(1).GetReg();
if (ops.at(2).GetKind() == Operand::Kind::FrameIndex) {
const auto& slot = function.GetFrameSlot(ops.at(2).GetFrameIndex());
int offset = slot.offset;
if (offset >= -4095 && offset <= 4095) {
if (offset >= 0) {
os << " add " << PhysRegName(dst) << ", " << PhysRegName(src) << ", #" << offset << "\n";
} else {
os << " sub " << PhysRegName(dst) << ", " << PhysRegName(src) << ", #" << (-offset) << "\n";
}
} else {
os << " ldr x9, =" << offset << "\n";
os << " add " << PhysRegName(dst) << ", " << PhysRegName(src) << ", x9\n";
}
} else if (ops.at(2).GetKind() == Operand::Kind::Global) {
os << " add " << PhysRegName(dst) << ", " << PhysRegName(src) << ", :lo12:" << ops.at(2).GetGlobalName() << "\n";
} else {
int imm = ops.at(2).GetImm();
if (imm >= -4095 && imm <= 4095) {
if (imm >= 0) {
os << " add " << PhysRegName(dst) << ", " << PhysRegName(src) << ", #" << imm << "\n";
} else {
os << " sub " << PhysRegName(dst) << ", " << PhysRegName(src) << ", #" << (-imm) << "\n";
}
} else {
os << " ldr x9, =" << imm << "\n";
os << " add " << PhysRegName(dst) << ", " << PhysRegName(src) << ", x9\n";
}
}
break;
}
case Opcode::LslImm:
os << " lsl " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << ", #"
<< ops.at(2).GetImm() << "\n";
break;
case Opcode::LdrRegReg: {
PhysReg reg = ops.at(0).GetReg();
const char* ldr_cmd = IsFloatReg(reg) ? "ldr" : "ldr";
os << " " << ldr_cmd << " " << PhysRegName(reg) << ", ["
<< PhysRegName(ops.at(1).GetReg()) << "]\n";
break;
}
case Opcode::StrRegReg: {
PhysReg reg = ops.at(0).GetReg();
const char* str_cmd = IsFloatReg(reg) ? "str" : "str";
os << " " << str_cmd << " " << PhysRegName(reg) << ", ["
<< PhysRegName(ops.at(1).GetReg()) << "]\n";
break;
}
case Opcode::SIToFP:
os << " scvtf " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << "\n";
break;
case Opcode::FPToSI:
os << " fcvtzs " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << "\n";
break;
case Opcode::ZExt:
if (ops.at(0).GetReg() >= PhysReg::X0 && ops.at(0).GetReg() <= PhysReg::X28) {
os << " sxtw " << PhysRegName(ops.at(0).GetReg()) << ", " << PhysRegName(ops.at(1).GetReg()) << "\n";
} else {
os << " and " << PhysRegName(ops.at(0).GetReg()) << ", " << PhysRegName(ops.at(1).GetReg()) << ", #1\n";
}
break;
}
case Opcode::StoreStack: {
const auto& slot = GetFrameSlot(function, ops.at(1));
PrintStackAccess(os, "stur", ops.at(0).GetReg(), slot.offset);
break;
}
case Opcode::AddRR:
os << " add " << PhysRegName(ops.at(0).GetReg()) << ", "
<< PhysRegName(ops.at(1).GetReg()) << ", "
<< PhysRegName(ops.at(2).GetReg()) << "\n";
break;
case Opcode::Ret:
os << " ret\n";
break;
}
}
os << ".size " << function.GetName() << ", .-" << function.GetName()
<< "\n";
os << ".size " << function.GetName() << ", .-" << function.GetName() << "\n";
// Print read-only data segment if there are float constants
if (!float_constants.empty()) {
os << ".section .rodata\n";
os << ".align 2\n";
for (const auto& fc : float_constants) {
os << fc.label << ":\n";
uint32_t bits;
std::memcpy(&bits, &fc.value, sizeof(float));
os << " .word " << bits << " // float " << fc.value << "\n";
}
}
}
static uint32_t GetTypeSize(const ir::Type* type) {
if (type->IsInt32() || type->IsFloat()) {
return 4;
}
if (type->IsPtrInt32() || type->IsPtrFloat()) {
return 8;
}
if (type->IsArray()) {
auto* arr_ty = const_cast<ir::Type*>(type)->GetAsArrayType().get();
return arr_ty->GetNumElements() * GetTypeSize(arr_ty->GetElementType().get());
}
return 4;
}
void PrintGlobals(const ir::Module& module, std::ostream& os) {
for (const auto& gv : module.GetGlobalValues()) {
os << ".global " << gv->GetName() << "\n";
std::shared_ptr<ir::Type> actual_ty = gv->GetType();
if (actual_ty->IsPtrInt32()) actual_ty = ir::Type::GetInt32Type();
else if (actual_ty->IsPtrFloat()) actual_ty = ir::Type::GetFloatType();
uint32_t actual_size = GetTypeSize(actual_ty.get());
if (gv->GetInitializer()) {
os << ".data\n";
os << ".align 2\n";
os << ".size " << gv->GetName() << ", " << actual_size << "\n";
os << gv->GetName() << ":\n";
if (actual_ty->IsFloat()) {
float val = 0.0f;
if (auto* cf = dynamic_cast<const ir::ConstantFloat*>(gv->GetInitializer())) {
val = cf->GetValue();
} else if (auto* ci = dynamic_cast<const ir::ConstantInt*>(gv->GetInitializer())) {
val = static_cast<float>(ci->GetValue());
}
uint32_t bits;
std::memcpy(&bits, &val, sizeof(float));
os << " .word " << bits << " // float " << val << "\n";
} else {
int val = 0;
if (auto* ci = dynamic_cast<const ir::ConstantInt*>(gv->GetInitializer())) {
val = ci->GetValue();
} else if (auto* cf = dynamic_cast<const ir::ConstantFloat*>(gv->GetInitializer())) {
val = static_cast<int>(cf->GetValue());
}
os << " .word " << val << "\n";
}
} else {
os << ".bss\n";
os << ".align 4\n";
os << ".size " << gv->GetName() << ", " << actual_size << "\n";
os << gv->GetName() << ":\n";
os << " .zero " << actual_size << "\n";
}
os << "\n";
}
}
} // namespace mir

View File

@@ -18,10 +18,10 @@ void RunFrameLowering(MachineFunction& function) {
int cursor = 0;
for (const auto& slot : function.GetFrameSlots()) {
cursor += slot.size;
if (-cursor < -256) {
throw std::runtime_error(FormatError("mir", "暂不支持过大的栈帧"));
}
}
// Align stack frames to 16 bytes for AArch64
cursor = AlignTo(cursor, 16);
cursor = 0;
for (const auto& slot : function.GetFrameSlots()) {
@@ -30,16 +30,40 @@ void RunFrameLowering(MachineFunction& function) {
}
function.SetFrameSize(AlignTo(cursor, 16));
auto& insts = function.GetEntry().GetInstructions();
std::vector<MachineInstr> lowered;
lowered.emplace_back(Opcode::Prologue);
for (const auto& inst : insts) {
if (inst.GetOpcode() == Opcode::Ret) {
lowered.emplace_back(Opcode::Epilogue);
auto& blocks = function.GetBlocks();
if (blocks.empty()) return;
bool has_call = false;
for (const auto& block : blocks) {
for (const auto& inst : block.GetInstructions()) {
if (inst.GetOpcode() == Opcode::Call) {
has_call = true;
break;
}
}
lowered.push_back(inst);
if (has_call) break;
}
if (function.GetFrameSize() == 0 && !has_call) {
return;
}
// Insert Prologue at the start of the first block
auto& entry_insts = blocks.front().GetInstructions();
entry_insts.insert(entry_insts.begin(), MachineInstr(Opcode::Prologue));
// Insert Epilogue before every Ret in all blocks
for (auto& block : blocks) {
auto& insts = block.GetInstructions();
std::vector<MachineInstr> lowered;
for (const auto& inst : insts) {
if (inst.GetOpcode() == Opcode::Ret) {
lowered.emplace_back(Opcode::Epilogue);
}
lowered.push_back(inst);
}
insts = std::move(lowered);
}
insts = std::move(lowered);
}
} // namespace mir

View File

@@ -2,122 +2,701 @@
#include <stdexcept>
#include <unordered_map>
#include <unordered_set>
#include <vector>
#include <cstring>
#include "ir/IR.h"
#include "utils/Log.h"
#include <iostream>
namespace mir {
namespace {
using ValueSlotMap = std::unordered_map<const ir::Value*, int>;
uint32_t GetTypeSize(const ir::Type* type) {
if (type->IsInt32() || type->IsFloat()) {
return 4;
}
if (type->IsPtrInt32() || type->IsPtrFloat()) {
return 8; // 64-bit pointers
}
if (type->IsArray()) {
auto* arr_ty = const_cast<ir::Type*>(type)->GetAsArrayType().get();
return arr_ty->GetNumElements() * GetTypeSize(arr_ty->GetElementType().get());
}
return 4;
}
std::unordered_set<const ir::Value*> IdentifyPointerValues(const ir::Function& function) {
std::unordered_set<const ir::Value*> pointers;
// 1. Arguments that are pointers
for (const auto& arg : function.GetArguments()) {
if (arg->GetType()->IsPtrInt32() || arg->GetType()->IsPtrFloat()) {
pointers.insert(arg.get());
}
}
// 2. Alloca instructions that store a pointer argument
for (const auto& bbPtr : function.GetBlocks()) {
for (const auto& instPtr : bbPtr->GetInstructions()) {
const auto* inst = instPtr.get();
if (inst->GetOpcode() == ir::Opcode::Alloca) {
bool stores_ptr = false;
auto* parent_bb = inst->GetParent();
if (parent_bb) {
auto* parent_func = parent_bb->GetParent();
if (parent_func) {
for (const auto& other_bb : parent_func->GetBlocks()) {
for (const auto& other_inst : other_bb->GetInstructions()) {
if (other_inst->GetOpcode() == ir::Opcode::Store) {
auto* store = static_cast<const ir::StoreInst*>(other_inst.get());
if (store->GetPtr() == inst) {
auto* val = store->GetValue();
if (val->GetType()->IsPtrInt32() || val->GetType()->IsPtrFloat() || pointers.find(val) != pointers.end()) {
stores_ptr = true;
break;
}
}
}
}
if (stores_ptr) break;
}
}
}
if (stores_ptr) {
pointers.insert(inst);
}
}
}
}
// 3. GEP instructions
for (const auto& bbPtr : function.GetBlocks()) {
for (const auto& instPtr : bbPtr->GetInstructions()) {
const auto* inst = instPtr.get();
if (inst->GetOpcode() == ir::Opcode::GEP) {
pointers.insert(inst);
}
}
}
// 4. Load instructions that load from those pointer-storing allocas
for (const auto& bbPtr : function.GetBlocks()) {
for (const auto& instPtr : bbPtr->GetInstructions()) {
const auto* inst = instPtr.get();
if (inst->GetOpcode() == ir::Opcode::Load) {
auto* load = static_cast<const ir::LoadInst*>(inst);
if (pointers.find(load->GetPtr()) != pointers.end()) {
if (auto* alloca = dynamic_cast<const ir::Instruction*>(load->GetPtr())) {
if (alloca->GetOpcode() == ir::Opcode::Alloca) {
pointers.insert(inst);
}
}
}
}
}
}
return pointers;
}
uint32_t GetAllocaSize(const ir::Instruction& inst, const std::unordered_set<const ir::Value*>& pointers) {
auto type = inst.GetType();
if (pointers.find(&inst) != pointers.end()) {
return 8; // Stores a 64-bit pointer
}
return GetTypeSize(type.get());
}
bool IsPowerOfTwo(uint32_t value) {
return value != 0 && (value & (value - 1)) == 0;
}
int Log2(uint32_t value) {
int shift = 0;
while (value > 1) {
value >>= 1;
shift++;
}
return shift;
}
std::vector<uint32_t> GetGepStrides(const ir::GetElementPtrInst& gep) {
std::vector<uint32_t> strides;
auto curr_type = gep.GetPtr()->GetType();
if (curr_type->IsPtrInt32() || curr_type->IsPtrFloat()) {
strides.push_back(4);
} else if (curr_type->IsArray()) {
strides.push_back(GetTypeSize(curr_type.get()));
for (size_t i = 2; i < gep.GetNumOperands(); ++i) {
curr_type = curr_type->GetAsArrayType()->GetElementType();
strides.push_back(GetTypeSize(curr_type.get()));
}
}
return strides;
}
void EmitAddressToReg(const ir::Value* value, PhysReg target,
const ValueSlotMap& slots, MachineBasicBlock& block) {
if (auto* alloca = dynamic_cast<const ir::Instruction*>(value)) {
if (alloca->GetOpcode() == ir::Opcode::Alloca) {
auto it = slots.find(value);
if (it == slots.end()) {
throw std::runtime_error(FormatError("mir", "找不到局部变量的栈槽: " + value->GetName()));
}
block.Append(Opcode::AddRegImm, {Operand::Reg(target), Operand::Reg(PhysReg::X29), Operand::FrameIndex(it->second)});
return;
}
}
if (value->IsGlobalValue()) {
block.Append(Opcode::Adrp, {Operand::Reg(target), Operand::Global(value->GetName())});
block.Append(Opcode::AddRegImm, {Operand::Reg(target), Operand::Reg(target), Operand::Global(value->GetName())});
return;
}
// Otherwise, the address itself is stored in a stack slot
auto it = slots.find(value);
if (it == slots.end()) {
throw std::runtime_error(FormatError("mir", "找不到指针的值槽: " + value->GetName()));
}
block.Append(Opcode::LoadStack, {Operand::Reg(target), Operand::FrameIndex(it->second)});
}
void EmitValueToReg(const ir::Value* value, PhysReg target,
const ValueSlotMap& slots, MachineBasicBlock& block) {
if (auto* constant = dynamic_cast<const ir::ConstantInt*>(value)) {
block.Append(Opcode::MovImm,
{Operand::Reg(target), Operand::Imm(constant->GetValue())});
block.Append(Opcode::MovImm, {Operand::Reg(target), Operand::Imm(constant->GetValue())});
return;
}
if (auto* constant = dynamic_cast<const ir::ConstantFloat*>(value)) {
float fval = constant->GetValue();
int bits;
std::memcpy(&bits, &fval, sizeof(float));
block.Append(Opcode::MovImm, {Operand::Reg(target), Operand::Imm(bits)});
return;
}
if (value->IsGlobalValue()) {
EmitAddressToReg(value, target, slots, block);
return;
}
auto it = slots.find(value);
if (it == slots.end()) {
throw std::runtime_error(
FormatError("mir", "找不到值对应的栈槽: " + value->GetName()));
throw std::runtime_error(FormatError("mir", "找不到值对应的栈槽: " + value->GetName()));
}
block.Append(Opcode::LoadStack,
{Operand::Reg(target), Operand::FrameIndex(it->second)});
block.Append(Opcode::LoadStack, {Operand::Reg(target), Operand::FrameIndex(it->second)});
}
void LowerInstruction(const ir::Instruction& inst, MachineFunction& function,
ValueSlotMap& slots) {
auto& block = function.GetEntry();
ValueSlotMap& slots, MachineBasicBlock& block,
const std::unordered_set<const ir::Value*>& pointers) {
switch (inst.GetOpcode()) {
case ir::Opcode::Alloca: {
slots.emplace(&inst, function.CreateFrameIndex());
slots.emplace(&inst, function.CreateFrameIndex(GetAllocaSize(inst, pointers)));
return;
}
case ir::Opcode::Store: {
auto& store = static_cast<const ir::StoreInst&>(inst);
auto dst = slots.find(store.GetPtr());
if (dst == slots.end()) {
throw std::runtime_error(
FormatError("mir", "暂不支持对非栈变量地址进行写入"));
if (auto* alloca = dynamic_cast<const ir::Instruction*>(store.GetPtr())) {
if (alloca->GetOpcode() == ir::Opcode::Alloca) {
auto it = slots.find(alloca);
if (it != slots.end()) {
bool is_ptr = store.GetValue()->GetType()->IsPtrInt32() ||
store.GetValue()->GetType()->IsPtrFloat() ||
pointers.find(store.GetValue()) != pointers.end() ||
store.GetValue()->IsGlobalValue();
PhysReg val_reg = store.GetValue()->GetType()->IsFloat() ? PhysReg::S8 :
is_ptr ? PhysReg::X8 : PhysReg::W8;
EmitValueToReg(store.GetValue(), val_reg, slots, block);
block.Append(Opcode::StoreStack, {Operand::Reg(val_reg), Operand::FrameIndex(it->second)});
return;
}
}
}
EmitValueToReg(store.GetValue(), PhysReg::W8, slots, block);
block.Append(Opcode::StoreStack,
{Operand::Reg(PhysReg::W8), Operand::FrameIndex(dst->second)});
// Dynamic store
bool is_ptr = store.GetValue()->GetType()->IsPtrInt32() ||
store.GetValue()->GetType()->IsPtrFloat() ||
pointers.find(store.GetValue()) != pointers.end() ||
store.GetValue()->IsGlobalValue();
PhysReg val_reg = store.GetValue()->GetType()->IsFloat() ? PhysReg::S8 :
is_ptr ? PhysReg::X8 : PhysReg::W8;
EmitValueToReg(store.GetValue(), val_reg, slots, block);
EmitAddressToReg(store.GetPtr(), PhysReg::X9, slots, block);
block.Append(Opcode::StrRegReg, {Operand::Reg(val_reg), Operand::Reg(PhysReg::X9)});
return;
}
case ir::Opcode::Load: {
auto& load = static_cast<const ir::LoadInst&>(inst);
auto src = slots.find(load.GetPtr());
if (src == slots.end()) {
throw std::runtime_error(
FormatError("mir", "暂不支持对非栈变量地址进行读取"));
}
int dst_slot = function.CreateFrameIndex();
block.Append(Opcode::LoadStack,
{Operand::Reg(PhysReg::W8), Operand::FrameIndex(src->second)});
block.Append(Opcode::StoreStack,
{Operand::Reg(PhysReg::W8), Operand::FrameIndex(dst_slot)});
bool is_ptr = load.GetType()->IsPtrInt32() ||
load.GetType()->IsPtrFloat() ||
pointers.find(&inst) != pointers.end();
int dst_slot = function.CreateFrameIndex(is_ptr ? 8 : GetTypeSize(load.GetType().get()));
slots.emplace(&inst, dst_slot);
if (auto* alloca = dynamic_cast<const ir::Instruction*>(load.GetPtr())) {
if (alloca->GetOpcode() == ir::Opcode::Alloca) {
auto it = slots.find(alloca);
if (it != slots.end()) {
PhysReg val_reg = load.GetType()->IsFloat() ? PhysReg::S8 :
is_ptr ? PhysReg::X8 : PhysReg::W8;
block.Append(Opcode::LoadStack, {Operand::Reg(val_reg), Operand::FrameIndex(it->second)});
block.Append(Opcode::StoreStack, {Operand::Reg(val_reg), Operand::FrameIndex(dst_slot)});
return;
}
}
}
// Dynamic load
PhysReg val_reg = load.GetType()->IsFloat() ? PhysReg::S8 :
is_ptr ? PhysReg::X8 : PhysReg::W8;
EmitAddressToReg(load.GetPtr(), PhysReg::X9, slots, block);
block.Append(Opcode::LdrRegReg, {Operand::Reg(val_reg), Operand::Reg(PhysReg::X9)});
block.Append(Opcode::StoreStack, {Operand::Reg(val_reg), Operand::FrameIndex(dst_slot)});
return;
}
case ir::Opcode::Add: {
case ir::Opcode::Add:
case ir::Opcode::Sub:
case ir::Opcode::Mul:
case ir::Opcode::Div:
case ir::Opcode::Mod: {
auto& bin = static_cast<const ir::BinaryInst&>(inst);
int dst_slot = function.CreateFrameIndex();
int dst_slot = function.CreateFrameIndex(4);
slots.emplace(&inst, dst_slot);
if (inst.GetOpcode() == ir::Opcode::Add) {
const ir::Value* variable = nullptr;
int constant = 0;
if (auto* lhs_const = dynamic_cast<const ir::ConstantInt*>(bin.GetLhs())) {
variable = bin.GetRhs();
constant = lhs_const->GetValue();
} else if (auto* rhs_const = dynamic_cast<const ir::ConstantInt*>(bin.GetRhs())) {
variable = bin.GetLhs();
constant = rhs_const->GetValue();
}
if (variable && constant >= -4095 && constant <= 4095) {
EmitValueToReg(variable, PhysReg::W8, slots, block);
block.Append(Opcode::AddRegImm, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W8), Operand::Imm(constant)});
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::W8), Operand::FrameIndex(dst_slot)});
return;
}
} else if (inst.GetOpcode() == ir::Opcode::Sub) {
if (auto* rhs_const = dynamic_cast<const ir::ConstantInt*>(bin.GetRhs())) {
int constant = rhs_const->GetValue();
if (constant >= -4095 && constant <= 4095) {
EmitValueToReg(bin.GetLhs(), PhysReg::W8, slots, block);
block.Append(Opcode::AddRegImm, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W8), Operand::Imm(-constant)});
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::W8), Operand::FrameIndex(dst_slot)});
return;
}
}
}
if (inst.GetOpcode() == ir::Opcode::Mul) {
const ir::Value* variable = nullptr;
int constant = 0;
if (auto* lhs_const = dynamic_cast<const ir::ConstantInt*>(bin.GetLhs())) {
variable = bin.GetRhs();
constant = lhs_const->GetValue();
} else if (auto* rhs_const = dynamic_cast<const ir::ConstantInt*>(bin.GetRhs())) {
variable = bin.GetLhs();
constant = rhs_const->GetValue();
}
if (variable && constant > 1) {
EmitValueToReg(variable, PhysReg::W8, slots, block);
if (IsPowerOfTwo(static_cast<uint32_t>(constant))) {
block.Append(Opcode::LslImm, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W8), Operand::Imm(Log2(static_cast<uint32_t>(constant)))});
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::W8), Operand::FrameIndex(dst_slot)});
return;
}
if (IsPowerOfTwo(static_cast<uint32_t>(constant - 1))) {
block.Append(Opcode::LslImm, {Operand::Reg(PhysReg::W9), Operand::Reg(PhysReg::W8), Operand::Imm(Log2(static_cast<uint32_t>(constant - 1)))});
block.Append(Opcode::AddRR, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W9), Operand::Reg(PhysReg::W8)});
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::W8), Operand::FrameIndex(dst_slot)});
return;
}
}
}
EmitValueToReg(bin.GetLhs(), PhysReg::W8, slots, block);
EmitValueToReg(bin.GetRhs(), PhysReg::W9, slots, block);
block.Append(Opcode::AddRR, {Operand::Reg(PhysReg::W8),
Operand::Reg(PhysReg::W8),
Operand::Reg(PhysReg::W9)});
block.Append(Opcode::StoreStack,
{Operand::Reg(PhysReg::W8), Operand::FrameIndex(dst_slot)});
if (inst.GetOpcode() == ir::Opcode::Add) {
block.Append(Opcode::AddRR, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W9)});
} else if (inst.GetOpcode() == ir::Opcode::Sub) {
block.Append(Opcode::SubRR, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W9)});
} else if (inst.GetOpcode() == ir::Opcode::Mul) {
block.Append(Opcode::MulRR, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W9)});
} else if (inst.GetOpcode() == ir::Opcode::Div) {
block.Append(Opcode::SDivRR, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W9)});
} else if (inst.GetOpcode() == ir::Opcode::Mod) {
block.Append(Opcode::SDivRR, {Operand::Reg(PhysReg::W10), Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W9)});
block.Append(Opcode::MSubRRRR, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W10), Operand::Reg(PhysReg::W9), Operand::Reg(PhysReg::W8)});
}
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::W8), Operand::FrameIndex(dst_slot)});
return;
}
case ir::Opcode::FAdd:
case ir::Opcode::FSub:
case ir::Opcode::FMul:
case ir::Opcode::FDiv: {
auto& bin = static_cast<const ir::BinaryInst&>(inst);
int dst_slot = function.CreateFrameIndex(4);
slots.emplace(&inst, dst_slot);
EmitValueToReg(bin.GetLhs(), PhysReg::S8, slots, block);
EmitValueToReg(bin.GetRhs(), PhysReg::S9, slots, block);
if (inst.GetOpcode() == ir::Opcode::FAdd) {
block.Append(Opcode::FAddRRR, {Operand::Reg(PhysReg::S8), Operand::Reg(PhysReg::S8), Operand::Reg(PhysReg::S9)});
} else if (inst.GetOpcode() == ir::Opcode::FSub) {
block.Append(Opcode::FSubRRR, {Operand::Reg(PhysReg::S8), Operand::Reg(PhysReg::S8), Operand::Reg(PhysReg::S9)});
} else if (inst.GetOpcode() == ir::Opcode::FMul) {
block.Append(Opcode::FMulRRR, {Operand::Reg(PhysReg::S8), Operand::Reg(PhysReg::S8), Operand::Reg(PhysReg::S9)});
} else if (inst.GetOpcode() == ir::Opcode::FDiv) {
block.Append(Opcode::FDivRRR, {Operand::Reg(PhysReg::S8), Operand::Reg(PhysReg::S8), Operand::Reg(PhysReg::S9)});
}
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::S8), Operand::FrameIndex(dst_slot)});
return;
}
case ir::Opcode::ICmpEQ:
case ir::Opcode::ICmpNE:
case ir::Opcode::ICmpLT:
case ir::Opcode::ICmpGT:
case ir::Opcode::ICmpLE:
case ir::Opcode::ICmpGE: {
auto& cmp = static_cast<const ir::BinaryInst&>(inst);
int dst_slot = function.CreateFrameIndex(4);
slots.emplace(&inst, dst_slot);
EmitValueToReg(cmp.GetLhs(), PhysReg::W8, slots, block);
EmitValueToReg(cmp.GetRhs(), PhysReg::W9, slots, block);
block.Append(Opcode::CmpRR, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W9)});
std::string cond;
switch (inst.GetOpcode()) {
case ir::Opcode::ICmpEQ: cond = "eq"; break;
case ir::Opcode::ICmpNE: cond = "ne"; break;
case ir::Opcode::ICmpLT: cond = "lt"; break;
case ir::Opcode::ICmpGT: cond = "gt"; break;
case ir::Opcode::ICmpLE: cond = "le"; break;
case ir::Opcode::ICmpGE: cond = "ge"; break;
default: break;
}
block.Append(Opcode::Cset, {Operand::Reg(PhysReg::W8), Operand::Cond(cond)});
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::W8), Operand::FrameIndex(dst_slot)});
return;
}
case ir::Opcode::FCmpEQ:
case ir::Opcode::FCmpNE:
case ir::Opcode::FCmpLT:
case ir::Opcode::FCmpGT:
case ir::Opcode::FCmpLE:
case ir::Opcode::FCmpGE: {
auto& cmp = static_cast<const ir::BinaryInst&>(inst);
int dst_slot = function.CreateFrameIndex(4);
slots.emplace(&inst, dst_slot);
EmitValueToReg(cmp.GetLhs(), PhysReg::S8, slots, block);
EmitValueToReg(cmp.GetRhs(), PhysReg::S9, slots, block);
block.Append(Opcode::FCmpRR, {Operand::Reg(PhysReg::S8), Operand::Reg(PhysReg::S9)});
std::string cond;
switch (inst.GetOpcode()) {
case ir::Opcode::FCmpEQ: cond = "eq"; break;
case ir::Opcode::FCmpNE: cond = "ne"; break;
case ir::Opcode::FCmpLT: cond = "mi"; break;
case ir::Opcode::FCmpGT: cond = "gt"; break;
case ir::Opcode::FCmpLE: cond = "ls"; break;
case ir::Opcode::FCmpGE: cond = "ge"; break;
default: break;
}
block.Append(Opcode::Cset, {Operand::Reg(PhysReg::W8), Operand::Cond(cond)});
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::W8), Operand::FrameIndex(dst_slot)});
return;
}
case ir::Opcode::ZExt: {
auto& cast = static_cast<const ir::CastInst&>(inst);
int dst_slot = function.CreateFrameIndex(4);
slots.emplace(&inst, dst_slot);
EmitValueToReg(cast.GetValue(), PhysReg::W8, slots, block);
block.Append(Opcode::ZExt, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W8)});
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::W8), Operand::FrameIndex(dst_slot)});
return;
}
case ir::Opcode::SIToFP: {
auto& cast = static_cast<const ir::CastInst&>(inst);
int dst_slot = function.CreateFrameIndex(4);
slots.emplace(&inst, dst_slot);
EmitValueToReg(cast.GetValue(), PhysReg::W8, slots, block);
block.Append(Opcode::SIToFP, {Operand::Reg(PhysReg::S8), Operand::Reg(PhysReg::W8)});
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::S8), Operand::FrameIndex(dst_slot)});
return;
}
case ir::Opcode::FPToSI: {
auto& cast = static_cast<const ir::CastInst&>(inst);
int dst_slot = function.CreateFrameIndex(4);
slots.emplace(&inst, dst_slot);
EmitValueToReg(cast.GetValue(), PhysReg::S8, slots, block);
block.Append(Opcode::FPToSI, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::S8)});
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::W8), Operand::FrameIndex(dst_slot)});
return;
}
case ir::Opcode::Br: {
auto& br = static_cast<const ir::BranchInst&>(inst);
auto emit_phi_copies = [&](const ir::BasicBlock* succ) {
if (!succ) return;
for (const auto& succ_inst : succ->GetInstructions()) {
if (succ_inst->GetOpcode() == ir::Opcode::Phi) {
auto* phi = static_cast<const ir::PhiInst*>(succ_inst.get());
const ir::Value* incoming_val = nullptr;
for (size_t i = 0; i < phi->GetNumIncoming(); ++i) {
if (phi->GetIncomingBlock(i) == inst.GetParent()) {
incoming_val = phi->GetIncomingValue(i);
break;
}
}
if (incoming_val) {
auto slot_it = slots.find(phi);
if (slot_it != slots.end()) {
int phi_slot = slot_it->second;
bool is_ptr = phi->GetType()->IsPtrInt32() ||
phi->GetType()->IsPtrFloat() ||
pointers.find(phi) != pointers.end() ||
(incoming_val && (pointers.find(incoming_val) != pointers.end() || incoming_val->IsGlobalValue()));
PhysReg val_reg = phi->GetType()->IsFloat() ? PhysReg::S8 :
is_ptr ? PhysReg::X8 : PhysReg::W8;
EmitValueToReg(incoming_val, val_reg, slots, block);
block.Append(Opcode::StoreStack, {Operand::Reg(val_reg), Operand::FrameIndex(phi_slot)});
}
}
}
}
};
if (br.IsConditional()) {
emit_phi_copies(br.GetIfTrue());
emit_phi_copies(br.GetIfFalse());
EmitValueToReg(br.GetCondition(), PhysReg::W8, slots, block);
block.Append(Opcode::MovImm, {Operand::Reg(PhysReg::W9), Operand::Imm(0)});
block.Append(Opcode::CmpRR, {Operand::Reg(PhysReg::W8), Operand::Reg(PhysReg::W9)});
block.Append(Opcode::BCond, {Operand::Cond("ne"), Operand::Label(br.GetIfTrue()->GetName())});
block.Append(Opcode::B, {Operand::Label(br.GetIfFalse()->GetName())});
} else {
emit_phi_copies(br.GetDest());
block.Append(Opcode::B, {Operand::Label(br.GetDest()->GetName())});
}
return;
}
case ir::Opcode::Phi: {
return;
}
case ir::Opcode::Ret: {
auto& ret = static_cast<const ir::ReturnInst&>(inst);
EmitValueToReg(ret.GetValue(), PhysReg::W0, slots, block);
if (ret.GetValue()) {
bool is_ptr = ret.GetValue()->GetType()->IsPtrInt32() ||
ret.GetValue()->GetType()->IsPtrFloat() ||
pointers.find(ret.GetValue()) != pointers.end() ||
ret.GetValue()->IsGlobalValue();
PhysReg ret_reg = ret.GetValue()->GetType()->IsFloat() ? PhysReg::S0 :
is_ptr ? PhysReg::X0 : PhysReg::W0;
EmitValueToReg(ret.GetValue(), ret_reg, slots, block);
}
block.Append(Opcode::Ret);
return;
}
case ir::Opcode::Sub:
case ir::Opcode::Mul:
throw std::runtime_error(FormatError("mir", "暂不支持该二元运算"));
case ir::Opcode::Call: {
auto& call = static_cast<const ir::CallInst&>(inst);
bool is_ret_ptr = call.GetType()->IsPtrInt32() ||
call.GetType()->IsPtrFloat() ||
pointers.find(&inst) != pointers.end();
int dst_slot = -1;
if (!call.GetType()->IsVoid()) {
dst_slot = function.CreateFrameIndex(is_ret_ptr ? 8 : GetTypeSize(call.GetType().get()));
slots.emplace(&inst, dst_slot);
}
int int_idx = 0;
int float_idx = 0;
for (size_t i = 1; i < call.GetNumOperands(); ++i) {
auto* arg = call.GetOperand(i);
if (arg->GetType()->IsFloat()) {
PhysReg reg = static_cast<PhysReg>(static_cast<int>(PhysReg::S0) + float_idx);
EmitValueToReg(arg, reg, slots, block);
float_idx++;
} else {
bool is_arg_ptr = arg->GetType()->IsPtrInt32() ||
arg->GetType()->IsPtrFloat() ||
pointers.find(arg) != pointers.end() ||
arg->IsGlobalValue();
PhysReg reg = is_arg_ptr
? static_cast<PhysReg>(static_cast<int>(PhysReg::X0) + int_idx)
: static_cast<PhysReg>(static_cast<int>(PhysReg::W0) + int_idx);
EmitValueToReg(arg, reg, slots, block);
int_idx++;
}
}
block.Append(Opcode::Call, {Operand::Global(call.GetFunction()->GetName())});
if (dst_slot != -1) {
if (call.GetType()->IsFloat()) {
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::S0), Operand::FrameIndex(dst_slot)});
} else {
PhysReg ret_reg = is_ret_ptr ? PhysReg::X0 : PhysReg::W0;
block.Append(Opcode::StoreStack, {Operand::Reg(ret_reg), Operand::FrameIndex(dst_slot)});
}
}
return;
}
case ir::Opcode::GEP: {
auto& gep = static_cast<const ir::GetElementPtrInst&>(inst);
int dst_slot = function.CreateFrameIndex(8);
slots.emplace(&inst, dst_slot);
// Load base pointer address into X8
if (gep.GetPtr()->IsGlobalValue()) {
EmitAddressToReg(gep.GetPtr(), PhysReg::X8, slots, block);
} else if (auto* alloca = dynamic_cast<const ir::AllocaInst*>(gep.GetPtr())) {
if (alloca->GetType()->IsArray()) {
EmitAddressToReg(gep.GetPtr(), PhysReg::X8, slots, block);
} else {
EmitValueToReg(gep.GetPtr(), PhysReg::X8, slots, block);
}
} else {
EmitValueToReg(gep.GetPtr(), PhysReg::X8, slots, block);
}
auto strides = GetGepStrides(gep);
for (size_t i = 1; i < gep.GetNumOperands(); ++i) {
auto* idx = gep.GetOperand(i);
uint32_t stride = strides.at(i - 1);
if (auto* ci = dynamic_cast<const ir::ConstantInt*>(idx)) {
int64_t offset = static_cast<int64_t>(ci->GetValue()) * stride;
if (offset != 0) {
block.Append(Opcode::AddRegImm, {Operand::Reg(PhysReg::X8), Operand::Reg(PhysReg::X8), Operand::Imm(offset)});
}
continue;
}
EmitValueToReg(idx, PhysReg::W9, slots, block);
bool shifted = false;
if (stride > 1 && IsPowerOfTwo(stride)) {
block.Append(Opcode::ZExt, {Operand::Reg(PhysReg::X9), Operand::Reg(PhysReg::W9)});
block.Append(Opcode::LslImm, {Operand::Reg(PhysReg::X9), Operand::Reg(PhysReg::X9), Operand::Imm(Log2(stride))});
shifted = true;
} else if (stride > 1) {
block.Append(Opcode::MovImm, {Operand::Reg(PhysReg::W10), Operand::Imm(stride)});
block.Append(Opcode::MulRR, {Operand::Reg(PhysReg::W9), Operand::Reg(PhysReg::W9), Operand::Reg(PhysReg::W10)});
}
// Extend W9 to X9 and add to base address X8
if (!shifted) {
block.Append(Opcode::ZExt, {Operand::Reg(PhysReg::X9), Operand::Reg(PhysReg::W9)});
}
block.Append(Opcode::AddRR, {Operand::Reg(PhysReg::X8), Operand::Reg(PhysReg::X8), Operand::Reg(PhysReg::X9)});
}
// Store address into GEP's stack slot
block.Append(Opcode::StoreStack, {Operand::Reg(PhysReg::X8), Operand::FrameIndex(dst_slot)});
return;
}
}
throw std::runtime_error(FormatError("mir", "暂不支持该 IR 指令"));
throw std::runtime_error(FormatError("mir", "暂不支持该 IR 指令: " + std::to_string(static_cast<int>(inst.GetOpcode()))));
}
} // namespace
std::unique_ptr<MachineFunction> LowerToMIR(const ir::Module& module) {
std::vector<std::unique_ptr<MachineFunction>> LowerToMIR(const ir::Module& module) {
DefaultContext();
std::vector<std::unique_ptr<MachineFunction>> mfuncs;
if (module.GetFunctions().size() != 1) {
throw std::runtime_error(FormatError("mir", "暂不支持多个函数"));
for (const auto& funcPtr : module.GetFunctions()) {
const auto& func = *funcPtr;
if (func.GetBlocks().empty()) continue; // skip declarations
auto machine_func = std::make_unique<MachineFunction>(func.GetName());
ValueSlotMap slots;
auto pointers = IdentifyPointerValues(func);
// First, create all basic blocks in MachineFunction
std::unordered_map<const ir::BasicBlock*, MachineBasicBlock*> bb_map;
machine_func->GetBlocks().reserve(func.GetBlocks().size());
for (const auto& bbPtr : func.GetBlocks()) {
auto& mbb = machine_func->CreateBlock(bbPtr->GetName());
bb_map[bbPtr.get()] = &mbb;
}
// Pre-allocate stack slots for all Phi instructions in the function
for (const auto& bbPtr : func.GetBlocks()) {
for (const auto& inst : bbPtr->GetInstructions()) {
if (inst->GetOpcode() == ir::Opcode::Phi) {
bool is_phi_ptr = inst->GetType()->IsPtrInt32() ||
inst->GetType()->IsPtrFloat() ||
pointers.find(inst.get()) != pointers.end();
int slot = machine_func->CreateFrameIndex(is_phi_ptr ? 8 : GetTypeSize(inst->GetType().get()));
slots.emplace(inst.get(), slot);
}
}
}
auto& entry_block = *bb_map.at(func.GetEntry());
// Lower function arguments at the start of the entry block
const auto& args = func.GetArguments();
int int_idx = 0;
int float_idx = 0;
for (const auto& arg : args) {
bool is_arg_ptr = arg->GetType()->IsPtrInt32() ||
arg->GetType()->IsPtrFloat() ||
pointers.find(arg.get()) != pointers.end();
int slot = machine_func->CreateFrameIndex(is_arg_ptr ? 8 : GetTypeSize(arg->GetType().get()));
slots.emplace(arg.get(), slot);
if (arg->GetType()->IsFloat()) {
PhysReg reg = static_cast<PhysReg>(static_cast<int>(PhysReg::S0) + float_idx);
entry_block.Append(Opcode::StoreStack, {Operand::Reg(reg), Operand::FrameIndex(slot)});
float_idx++;
} else {
PhysReg reg = is_arg_ptr
? static_cast<PhysReg>(static_cast<int>(PhysReg::X0) + int_idx)
: static_cast<PhysReg>(static_cast<int>(PhysReg::W0) + int_idx);
entry_block.Append(Opcode::StoreStack, {Operand::Reg(reg), Operand::FrameIndex(slot)});
int_idx++;
}
}
// Now, lower all instructions block by block
for (const auto& bbPtr : func.GetBlocks()) {
auto& mbb = *bb_map.at(bbPtr.get());
for (const auto& inst : bbPtr->GetInstructions()) {
LowerInstruction(*inst, *machine_func, slots, mbb, pointers);
}
}
mfuncs.push_back(std::move(machine_func));
}
const auto& func = *module.GetFunctions().front();
if (func.GetName() != "main") {
throw std::runtime_error(FormatError("mir", "暂不支持非 main 函数"));
}
auto machine_func = std::make_unique<MachineFunction>(func.GetName());
ValueSlotMap slots;
const auto* entry = func.GetEntry();
if (!entry) {
throw std::runtime_error(FormatError("mir", "IR 函数缺少入口基本块"));
}
for (const auto& inst : entry->GetInstructions()) {
LowerInstruction(*inst, *machine_func, slots);
}
return machine_func;
return mfuncs;
}
} // namespace mir

View File

@@ -8,7 +8,12 @@
namespace mir {
MachineFunction::MachineFunction(std::string name)
: name_(std::move(name)), entry_("entry") {}
: name_(std::move(name)) {}
MachineBasicBlock& MachineFunction::CreateBlock(std::string name) {
blocks_.emplace_back(std::move(name));
return blocks_.back();
}
int MachineFunction::CreateFrameIndex(int size) {
int index = static_cast<int>(frame_slots_.size());

View File

@@ -4,10 +4,12 @@
namespace mir {
Operand::Operand(Kind kind, PhysReg reg, int imm)
: kind_(kind), reg_(reg), imm_(imm) {}
Operand::Operand(Kind kind, PhysReg reg, int imm, std::string str)
: kind_(kind), reg_(reg), imm_(imm), str_(std::move(str)) {}
Operand Operand::Reg(PhysReg reg) { return Operand(Kind::Reg, reg, 0); }
Operand Operand::Reg(PhysReg reg) {
return Operand(Kind::Reg, reg, 0);
}
Operand Operand::Imm(int value) {
return Operand(Kind::Imm, PhysReg::W0, value);
@@ -17,6 +19,18 @@ Operand Operand::FrameIndex(int index) {
return Operand(Kind::FrameIndex, PhysReg::W0, index);
}
Operand Operand::Global(std::string name) {
return Operand(Kind::Global, PhysReg::W0, 0, std::move(name));
}
Operand Operand::Label(std::string name) {
return Operand(Kind::Label, PhysReg::W0, 0, std::move(name));
}
Operand Operand::Cond(std::string cond) {
return Operand(Kind::Cond, PhysReg::W0, 0, std::move(cond));
}
MachineInstr::MachineInstr(Opcode opcode, std::vector<Operand> operands)
: opcode_(opcode), operands_(std::move(operands)) {}

View File

@@ -8,26 +8,19 @@ namespace mir {
namespace {
bool IsAllowedReg(PhysReg reg) {
switch (reg) {
case PhysReg::W0:
case PhysReg::W8:
case PhysReg::W9:
case PhysReg::X29:
case PhysReg::X30:
case PhysReg::SP:
return true;
}
return false;
return true; // We allow all defined physical registers
}
} // namespace
void RunRegAlloc(MachineFunction& function) {
for (const auto& inst : function.GetEntry().GetInstructions()) {
for (const auto& operand : inst.GetOperands()) {
if (operand.GetKind() == Operand::Kind::Reg &&
!IsAllowedReg(operand.GetReg())) {
throw std::runtime_error(FormatError("mir", "寄存器分配失败"));
for (const auto& block : function.GetBlocks()) {
for (const auto& inst : block.GetInstructions()) {
for (const auto& operand : inst.GetOperands()) {
if (operand.GetKind() == Operand::Kind::Reg &&
!IsAllowedReg(operand.GetReg())) {
throw std::runtime_error(FormatError("mir", "寄存器分配失败"));
}
}
}
}

View File

@@ -1,6 +1,7 @@
#include "mir/MIR.h"
#include <stdexcept>
#include <string>
#include "utils/Log.h"
@@ -8,18 +9,77 @@ namespace mir {
const char* PhysRegName(PhysReg reg) {
switch (reg) {
case PhysReg::W0:
return "w0";
case PhysReg::W8:
return "w8";
case PhysReg::W9:
return "w9";
case PhysReg::X29:
return "x29";
case PhysReg::X30:
return "x30";
case PhysReg::SP:
return "sp";
case PhysReg::W0: return "w0";
case PhysReg::W1: return "w1";
case PhysReg::W2: return "w2";
case PhysReg::W3: return "w3";
case PhysReg::W4: return "w4";
case PhysReg::W5: return "w5";
case PhysReg::W6: return "w6";
case PhysReg::W7: return "w7";
case PhysReg::W8: return "w8";
case PhysReg::W9: return "w9";
case PhysReg::W10: return "w10";
case PhysReg::W11: return "w11";
case PhysReg::W12: return "w12";
case PhysReg::W13: return "w13";
case PhysReg::W14: return "w14";
case PhysReg::W15: return "w15";
case PhysReg::W19: return "w19";
case PhysReg::W20: return "w20";
case PhysReg::W21: return "w21";
case PhysReg::W22: return "w22";
case PhysReg::W23: return "w23";
case PhysReg::W24: return "w24";
case PhysReg::W25: return "w25";
case PhysReg::W26: return "w26";
case PhysReg::W27: return "w27";
case PhysReg::W28: return "w28";
case PhysReg::X0: return "x0";
case PhysReg::X1: return "x1";
case PhysReg::X2: return "x2";
case PhysReg::X3: return "x3";
case PhysReg::X4: return "x4";
case PhysReg::X5: return "x5";
case PhysReg::X6: return "x6";
case PhysReg::X7: return "x7";
case PhysReg::X8: return "x8";
case PhysReg::X9: return "x9";
case PhysReg::X10: return "x10";
case PhysReg::X11: return "x11";
case PhysReg::X12: return "x12";
case PhysReg::X13: return "x13";
case PhysReg::X14: return "x14";
case PhysReg::X15: return "x15";
case PhysReg::X19: return "x19";
case PhysReg::X20: return "x20";
case PhysReg::X21: return "x21";
case PhysReg::X22: return "x22";
case PhysReg::X23: return "x23";
case PhysReg::X24: return "x24";
case PhysReg::X25: return "x25";
case PhysReg::X26: return "x26";
case PhysReg::X27: return "x27";
case PhysReg::X28: return "x28";
case PhysReg::S0: return "s0";
case PhysReg::S1: return "s1";
case PhysReg::S2: return "s2";
case PhysReg::S3: return "s3";
case PhysReg::S4: return "s4";
case PhysReg::S5: return "s5";
case PhysReg::S6: return "s6";
case PhysReg::S7: return "s7";
case PhysReg::S8: return "s8";
case PhysReg::S9: return "s9";
case PhysReg::S10: return "s10";
case PhysReg::S11: return "s11";
case PhysReg::S12: return "s12";
case PhysReg::S13: return "s13";
case PhysReg::S14: return "s14";
case PhysReg::S15: return "s15";
case PhysReg::X29: return "x29";
case PhysReg::X30: return "x30";
case PhysReg::SP: return "sp";
}
throw std::runtime_error(FormatError("mir", "未知物理寄存器"));
}

View File

@@ -1,4 +1,319 @@
// 窥孔优化Peephole
// - 删除冗余 move、合并常见指令模式
// - 提升最终汇编质量(按实现范围裁剪)
#include "mir/MIR.h"
#include <unordered_map>
#include <unordered_set>
#include <vector>
namespace mir {
namespace {
int AlignTo(int value, int align) {
return ((value + align - 1) / align) * align;
}
PhysReg NormalizeReg(PhysReg reg) {
int r = static_cast<int>(reg);
// Map 64-bit X0-X28 registers to 32-bit W0-W28 registers to handle aliasing
if (r >= static_cast<int>(PhysReg::X0) && r <= static_cast<int>(PhysReg::X28)) {
return static_cast<PhysReg>(r - static_cast<int>(PhysReg::X0) + static_cast<int>(PhysReg::W0));
}
return reg;
}
PhysReg MatchRegSize(PhysReg target, PhysReg src) {
int t = static_cast<int>(target);
int s = static_cast<int>(src);
bool target_is_64 = (t >= static_cast<int>(PhysReg::X0) && t <= static_cast<int>(PhysReg::X28)) ||
t == static_cast<int>(PhysReg::X29) ||
t == static_cast<int>(PhysReg::X30) ||
t == static_cast<int>(PhysReg::SP);
bool src_is_64 = (s >= static_cast<int>(PhysReg::X0) && s <= static_cast<int>(PhysReg::X28)) ||
s == static_cast<int>(PhysReg::X29) ||
s == static_cast<int>(PhysReg::X30) ||
s == static_cast<int>(PhysReg::SP);
if (target_is_64 && !src_is_64) {
if (s >= static_cast<int>(PhysReg::W0) && s <= static_cast<int>(PhysReg::W28)) {
return static_cast<PhysReg>(s - static_cast<int>(PhysReg::W0) + static_cast<int>(PhysReg::X0));
}
} else if (!target_is_64 && src_is_64) {
if (s >= static_cast<int>(PhysReg::X0) && s <= static_cast<int>(PhysReg::X28)) {
return static_cast<PhysReg>(s - static_cast<int>(PhysReg::X0) + static_cast<int>(PhysReg::W0));
}
}
return src;
}
bool IsFloatReg(PhysReg reg) {
return reg >= PhysReg::S0 && reg <= PhysReg::S15;
}
bool SameReg(const Operand& lhs, const Operand& rhs) {
return lhs.GetKind() == Operand::Kind::Reg &&
rhs.GetKind() == Operand::Kind::Reg &&
NormalizeReg(lhs.GetReg()) == NormalizeReg(rhs.GetReg());
}
bool IsZeroImm(const Operand& operand) {
return operand.GetKind() == Operand::Kind::Imm && operand.GetImm() == 0;
}
std::vector<MachineInstr> SimplifyCompareToBranch(
const std::vector<MachineInstr>& insts) {
std::vector<MachineInstr> simplified;
for (size_t i = 0; i < insts.size();) {
if (i + 4 < insts.size() &&
(insts[i].GetOpcode() == Opcode::CmpRR ||
insts[i].GetOpcode() == Opcode::FCmpRR) &&
insts[i + 1].GetOpcode() == Opcode::Cset &&
insts[i + 2].GetOpcode() == Opcode::MovImm &&
insts[i + 3].GetOpcode() == Opcode::CmpRR &&
insts[i + 4].GetOpcode() == Opcode::BCond) {
const auto& cset_ops = insts[i + 1].GetOperands();
const auto& mov_ops = insts[i + 2].GetOperands();
const auto& cmp2_ops = insts[i + 3].GetOperands();
const auto& br_ops = insts[i + 4].GetOperands();
if (mov_ops.size() == 2 && cmp2_ops.size() == 2 && br_ops.size() == 2 &&
SameReg(cset_ops.at(0), cmp2_ops.at(0)) &&
SameReg(mov_ops.at(0), cmp2_ops.at(1)) &&
IsZeroImm(mov_ops.at(1)) &&
br_ops.at(0).GetKind() == Operand::Kind::Cond &&
br_ops.at(0).GetCondCode() == "ne") {
simplified.push_back(insts[i]);
simplified.emplace_back(Opcode::BCond,
std::vector<Operand>{
Operand::Cond(cset_ops.at(1).GetCondCode()),
br_ops.at(1)});
i += 5;
continue;
}
}
simplified.push_back(insts[i]);
i++;
}
return simplified;
}
void CompactFrameSlots(MachineFunction& function) {
std::unordered_set<int> used_slots;
for (const auto& block : function.GetBlocks()) {
for (const auto& inst : block.GetInstructions()) {
for (const auto& opnd : inst.GetOperands()) {
if (opnd.GetKind() == Operand::Kind::FrameIndex) {
used_slots.insert(opnd.GetFrameIndex());
}
}
}
}
int cursor = 0;
for (const auto& slot : function.GetFrameSlots()) {
if (used_slots.find(slot.index) == used_slots.end()) {
continue;
}
cursor += slot.size;
function.GetFrameSlot(slot.index).offset = -cursor;
}
function.SetFrameSize(AlignTo(cursor, 16));
}
} // namespace
void RunPeephole(MachineFunction& function) {
for (auto& block : function.GetBlocks()) {
auto& insts = block.GetInstructions();
insts = SimplifyCompareToBranch(insts);
std::vector<MachineInstr> optimized;
// Map from FrameIndex to the normalized physical register that currently holds its value
std::unordered_map<int, PhysReg> slot_to_reg;
for (const auto& inst : insts) {
Opcode op = inst.GetOpcode();
const auto& ops = inst.GetOperands();
// 1. Handle register move elimination (e.g. mov w8, w8)
if (op == Opcode::MovReg) {
if (NormalizeReg(ops.at(0).GetReg()) == NormalizeReg(ops.at(1).GetReg())) {
continue; // Delete redundant self-moves
}
}
// 2. Handle redundant Load after Store
if (op == Opcode::LoadStack) {
int fi = ops.at(1).GetFrameIndex();
auto it = slot_to_reg.find(fi);
if (it != slot_to_reg.end()) {
PhysReg source_reg = it->second;
PhysReg dest_reg = NormalizeReg(ops.at(0).GetReg());
if (source_reg == dest_reg) {
// Loading the same register that already has the value - completely redundant!
continue;
} else {
// Replace LoadStack dest_reg, fi with MovReg dest_reg, matched_source
PhysReg matched_source = MatchRegSize(ops.at(0).GetReg(), it->second);
optimized.push_back(MachineInstr(Opcode::MovReg, {Operand::Reg(ops.at(0).GetReg()), Operand::Reg(matched_source)}));
// Invalidate any other slots mapping to dest_reg because dest_reg is written
std::vector<int> to_remove;
for (const auto& pair : slot_to_reg) {
if (NormalizeReg(pair.second) == dest_reg) {
to_remove.push_back(pair.first);
}
}
for (int key : to_remove) {
slot_to_reg.erase(key);
}
// Add new mapping (normalized)
slot_to_reg[fi] = dest_reg;
continue;
}
}
}
// 3. Track and optimize stores
if (op == Opcode::StoreStack) {
PhysReg src = NormalizeReg(ops.at(0).GetReg());
int fi = ops.at(1).GetFrameIndex();
auto it = slot_to_reg.find(fi);
if (it != slot_to_reg.end() && NormalizeReg(it->second) == src) {
continue; // Delete redundant store
}
slot_to_reg[fi] = src;
}
// 4. Invalidate register mappings on writes
bool writes_reg = false;
PhysReg written_reg = PhysReg::W0; // dummy
switch (op) {
case Opcode::MovImm:
if (!ops.empty() && ops.at(0).GetKind() == Operand::Kind::Reg) {
writes_reg = true;
written_reg = NormalizeReg(ops.at(0).GetReg());
// Under the hood, MovImm to a float register implicitly writes to x8/w8
if (IsFloatReg(ops.at(0).GetReg())) {
PhysReg implicitly_written = NormalizeReg(PhysReg::X8);
std::vector<int> to_remove;
for (const auto& pair : slot_to_reg) {
if (NormalizeReg(pair.second) == implicitly_written) {
to_remove.push_back(pair.first);
}
}
for (int key : to_remove) {
slot_to_reg.erase(key);
}
}
}
break;
case Opcode::LoadStack:
case Opcode::AddRR:
case Opcode::SubRR:
case Opcode::MulRR:
case Opcode::SDivRR:
case Opcode::MSubRRRR:
case Opcode::FAddRRR:
case Opcode::FSubRRR:
case Opcode::FMulRRR:
case Opcode::FDivRRR:
case Opcode::Cset:
case Opcode::MovReg:
case Opcode::Adrp:
case Opcode::AddRegImm:
case Opcode::LslImm:
case Opcode::LdrRegReg:
case Opcode::SIToFP:
case Opcode::FPToSI:
case Opcode::ZExt:
if (!ops.empty() && ops.at(0).GetKind() == Operand::Kind::Reg) {
writes_reg = true;
written_reg = NormalizeReg(ops.at(0).GetReg());
}
break;
case Opcode::Call:
// A function call destroys all temporary/scratch registers.
slot_to_reg.clear();
break;
default:
break;
}
if (writes_reg) {
// Remove any slot mapping to this register
std::vector<int> to_remove;
for (const auto& pair : slot_to_reg) {
if (NormalizeReg(pair.second) == written_reg) {
to_remove.push_back(pair.first);
}
}
for (int key : to_remove) {
slot_to_reg.erase(key);
}
}
optimized.push_back(inst);
}
insts = std::move(optimized);
}
// 5. Eliminate Dead Stack Slots (stores to slots that are never loaded or address-taken)
// Count loads and address-taken operations
std::unordered_map<int, int> load_count;
std::unordered_map<int, int> address_taken_count;
for (const auto& block : function.GetBlocks()) {
for (const auto& inst : block.GetInstructions()) {
Opcode op = inst.GetOpcode();
const auto& ops = inst.GetOperands();
for (const auto& opnd : ops) {
if (opnd.GetKind() == Operand::Kind::FrameIndex) {
int fi = opnd.GetFrameIndex();
if (op == Opcode::LoadStack) {
load_count[fi]++;
} else if (op != Opcode::StoreStack) {
address_taken_count[fi]++;
}
}
}
}
}
// Identify dead slots
std::unordered_set<int> dead_slots;
for (size_t i = 0; i < function.GetFrameSlots().size(); ++i) {
int fi = static_cast<int>(i);
if (load_count[fi] == 0 && address_taken_count[fi] == 0) {
dead_slots.insert(fi);
}
}
// Remove StoreStack to dead slots
for (auto& block : function.GetBlocks()) {
auto& insts = block.GetInstructions();
std::vector<MachineInstr> optimized;
for (const auto& inst : insts) {
if (inst.GetOpcode() == Opcode::StoreStack) {
int fi = inst.GetOperands().at(1).GetFrameIndex();
if (dead_slots.find(fi) != dead_slots.end()) {
continue; // Delete this store
}
}
optimized.push_back(inst);
}
insts = std::move(optimized);
}
CompactFrameSlots(function);
}
} // namespace mir

View File

@@ -211,67 +211,25 @@ class SemaVisitor final : public SysYBaseVisitor {
return ctx->exp()->accept(this);
}
std::any visitMulExp(SysYParser::MulExpContext* ctx) override {
std::any visitMulDivModExp(SysYParser::MulDivModExpContext* ctx) override {
ctx->exp(0)->accept(this);
ctx->exp(1)->accept(this);
return {};
}
std::any visitDivExp(SysYParser::DivExpContext* ctx) override {
std::any visitAddSubExp(SysYParser::AddSubExpContext* ctx) override {
ctx->exp(0)->accept(this);
ctx->exp(1)->accept(this);
return {};
}
std::any visitModExp(SysYParser::ModExpContext* ctx) override {
std::any visitRelExp(SysYParser::RelExpContext* ctx) override {
ctx->exp(0)->accept(this);
ctx->exp(1)->accept(this);
return {};
}
std::any visitAddExp(SysYParser::AddExpContext* ctx) override {
ctx->exp(0)->accept(this);
ctx->exp(1)->accept(this);
return {};
}
std::any visitSubExp(SysYParser::SubExpContext* ctx) override {
ctx->exp(0)->accept(this);
ctx->exp(1)->accept(this);
return {};
}
std::any visitLtExp(SysYParser::LtExpContext* ctx) override {
ctx->exp(0)->accept(this);
ctx->exp(1)->accept(this);
return {};
}
std::any visitLeExp(SysYParser::LeExpContext* ctx) override {
ctx->exp(0)->accept(this);
ctx->exp(1)->accept(this);
return {};
}
std::any visitGtExp(SysYParser::GtExpContext* ctx) override {
ctx->exp(0)->accept(this);
ctx->exp(1)->accept(this);
return {};
}
std::any visitGeExp(SysYParser::GeExpContext* ctx) override {
ctx->exp(0)->accept(this);
ctx->exp(1)->accept(this);
return {};
}
std::any visitEqExp(SysYParser::EqExpContext* ctx) override {
ctx->exp(0)->accept(this);
ctx->exp(1)->accept(this);
return {};
}
std::any visitNeExp(SysYParser::NeExpContext* ctx) override {
std::any visitEqNeExp(SysYParser::EqNeExpContext* ctx) override {
ctx->exp(0)->accept(this);
ctx->exp(1)->accept(this);
return {};

View File

@@ -1,4 +1,77 @@
// SysY 运行库实现:
// - 按实验/评测规范提供 I/O 等函数实现
// - 与编译器生成的目标代码链接,支撑运行时行为
#include <stdio.h>
#include <sys/time.h>
int getint() {
int x;
if (scanf("%d", &x) != 1) return 0;
return x;
}
int getch() {
return getchar();
}
float getfloat() {
double x;
if (scanf("%lf", &x) != 1) return 0.0f;
return (float)x;
}
int getarray(int a[]) {
int n;
if (scanf("%d", &n) != 1) return 0;
for (int i = 0; i < n; i++) {
if (scanf("%d", &a[i]) != 1) break;
}
return n;
}
int getfarray(float a[]) {
int n;
if (scanf("%d", &n) != 1) return 0;
for (int i = 0; i < n; i++) {
double val;
if (scanf("%lf", &val) != 1) break;
a[i] = (float)val;
}
return n;
}
void putint(int x) {
printf("%d", x);
}
void putch(int x) {
putchar(x);
}
void putfloat(float x) {
printf("%a", x);
}
void putarray(int n, int a[]) {
printf("%d:", n);
for (int i = 0; i < n; i++) {
printf(" %d", a[i]);
}
printf("\n");
}
void putfarray(int n, float a[]) {
printf("%d:", n);
for (int i = 0; i < n; i++) {
printf(" %a", a[i]);
}
printf("\n");
}
struct timeval start, stop;
void starttime() {
gettimeofday(&start, NULL);
}
void stoptime() {
gettimeofday(&stop, NULL);
long long duration = (stop.tv_sec - start.tv_sec) * 1000000LL + (stop.tv_usec - start.tv_usec);
printf("timer: %lld us\n", duration);
}