Skip to content

Latest commit

 

History

3 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MiniGraph AI Compiler

一个轻量级 AI 图编译器原型项目,支持:

  • Python 导出 toy model 计算图
  • C++ 解析 JSON 并构建 Graph IR
  • 图优化 Pass(DCE、算子融合)
  • Runtime 解释执行
  • C++ Codegen 编译执行
  • Runtime vs Codegen 正确性对比

项目目标

本项目用于模拟 AI 编译器/工具链的最小闭环,覆盖以下核心流程:

  1. Python 定义/导出计算图
  2. 导出 JSON
  3. C++ 解析并构建 Graph IR
  4. 运行图优化 Pass
  5. Runtime 解释执行
  6. Codegen 生成 C++ Kernel
  7. 自动比较 Runtime 与 Compiled Kernel 输出一致性

当前支持的算子

  • Input
  • MatMul
  • AddBias
  • Relu
  • FusedMatMulAddBias
  • FusedMatMulAddBiasRelu

当前支持的优化

  • Dead Code Elimination (DCE)
  • MatMul + AddBias -> FusedMatMulAddBias
  • FusedMatMulAddBias + Relu -> FusedMatMulAddBiasRelu

项目结构

MiniGraph-AI-Compiler
├── graphs/                  # 导出的示例图
├── include/mg/              # 头文件
├── python/                  # Python 图导出脚本
├── src/                     # C++ 实现
├── third_party/nlohmann/    # json 库
├── CMakeLists.txt
└── README.md

About

No description, website, or topics provided.

Resources

Stars

1 star

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages