这里是官方使用手册的配套可运行源码。建议先按编号学习,再回到 对应章节完成练习。

学习路线

顺序示例学习重点对应章节
00最短 quickstart仓外安装包的 include、链接与首次运行仓外 C++ 项目
01Tensor 基础CPU Tensor 的分配、元数据与数据读写仓外 C++ 项目
02图编译建图、编译、整图执行与结果校验核心概念
03自定义算子schema、kernel 注册与标准编译路径新增算子
04nn 与数据静态 nn 构图与 CPU 批数据迭代nn 与数据
05训练前向、反向、SGD 更新图与 compile once训练
06ONNX 权重initializer 权重的保存、加载与验证C++ 与工具

构建与运行

dev preset 会构建七个示例:

cmake --preset dev
cmake --build --preset dev
ctest --preset dev -L examples --output-on-failure

单独运行 CPU 图编译示例:

./build/dev/examples/frame_example_02_graph_compile cpu

成功时程序返回 0,失败时返回非零。示例 02 可用 cuda 参数执行真实 CUDA 编译、H2D、整图执行与 D2H;CUDA 构建方法见后端章节。 示例 04、05、06 的教学主路径使用 CPU。示例 06 只交换 ONNX initializer, 不导入 ONNX 算子图,并会在校验后删除临时权重文件。