一个轻量级 AI 图编译器原型项目,支持:
- Python 导出 toy model 计算图
- C++ 解析 JSON 并构建 Graph IR
- 图优化 Pass(DCE、算子融合)
- Runtime 解释执行
- C++ Codegen 编译执行
- Runtime vs Codegen 正确性对比
本项目用于模拟 AI 编译器/工具链的最小闭环,覆盖以下核心流程:
- Python 定义/导出计算图
- 导出 JSON
- C++ 解析并构建 Graph IR
- 运行图优化 Pass
- Runtime 解释执行
- Codegen 生成 C++ Kernel
- 自动比较 Runtime 与 Compiled Kernel 输出一致性
InputMatMulAddBiasReluFusedMatMulAddBiasFusedMatMulAddBiasRelu
- Dead Code Elimination (DCE)
MatMul + AddBias -> FusedMatMulAddBiasFusedMatMulAddBias + Relu -> FusedMatMulAddBiasRelu
MiniGraph-AI-Compiler
├── graphs/ # 导出的示例图
├── include/mg/ # 头文件
├── python/ # Python 图导出脚本
├── src/ # C++ 实现
├── third_party/nlohmann/ # json 库
├── CMakeLists.txt
└── README.md