Motivation

PaddlePaddle 目前正在对底层静态图 IR 进行升级，也即 PIR 项目，相关背景可参考 #55205 及一些相关 tracking issue。

为了推动 PIR 的正式落地，我们首先开发了 ProgramTranslator 来将老 IR Program 转换为 PIR Program，这保证了在前端组网不变的情况下可以快速验证 PIR 及其执行器的功能正确性。也就是图上的 Dy2St -> legacy static API branch -> legacy IR -> PIR -> PIR executor 这条链路（也称「中间态」）。目前这条链路已经打通，也通过单测和模型进行了验证。

graph TB
    A[Dy2St] --> B[legacy static API branch]
    B --> C[legacy IR]
    C --> D[legacy IR Executor]

    E[PIR Dy2St] --> F[PIR static API branch]
    F --> G[PIR]
    G --> H[PIR Executor]

    C -->|ProgramTranslator| G

我们最终的目标是直接通过 PIR 的 API 进行组网，当然作为静态图的主要出口，动转静也需要进行相关适配，也就是 PIR Dy2St -> PIR static API branch -> PIR -> PIR Executor 这条「最终态」（也称「理想态」）链路。

在这条链路上，有两个关键结点，一是针对于 PIR 的动转静模块，一是 Paddle API 的静态图分支，后者目前已经在 #57097、#58067 逐步展开，而前者就是本任务的内容。

Design

如上所述，目前 PIR 最终态链路的跑通还需要 PIR Dy2St 和 PIR static API branch 两大组件的完善。针对于 PIR static API branch，我们是使用 API 单测，使用静态图直接组网进行验证。而针对于 PIR Dy2St，我们则会使用动转静单测（test/dygraph_to_static）来进行验证。

在 #58356 我们已经对动转静单测的机制进行了统一，由于新机制易于扩展，因此我们可以方便地开始最终态的验证工作。

在 #58356 我们已经将大多单测开启了 SOT + AST / legacy IR + 「中间态」 共 2x2=4 种模式的验证，而本任务则是需要开启第三种 IR 模式，即「最终态」的验证，也就是 2x3=6 种模式的验证。

graph TB
    A[SOT] --> B[legacy IR]
    A --> C[PT]
    A --> D[PIR]

    E[AST] --> B
    E[AST] --> C
    E[AST] --> D

SOT 中尚未对相关模块进行适配，因此基本上一定会报错，我们考虑先对基础组件 AST 模式进行验证，因此前期我们会关掉 SOT+「最终态」 这种 case，以独立验证「最终态」的正确性。在 AST 模式基本验证过一遍后，再统一修复 SOT+「最终态」中遇到的问题，开启单测。

Schedule

• 2023.11.01-2023.11.30: 熟悉动转静及其单测机制，动转静最终态覆盖部分单测，保证 30% 单测验证通过（具体百分比需要在首次摸底之后确定）
• ...

Tasks

• 摸底动转静单测在「最终态」下的成功率，启用通过的动转静单测
• 统计并修复（协同）「最终态」动转静暴露出的问题，保障「最终态」基础功能的完善

Details

整体进度（56/111）：

状态解释：

• ✅：已经完全修复，所有单测都OK！
• 🟢：待合入，合入之后完全修复！
• 🟡：当前阶段不需要人力继续跟进，下阶段推进（Save、控制流、AMP、编译器、PyFuncOp、TrainStep整体导出）
• ❌：单测没有通过，需要人力跟进。
• 🚧：待分析，单测还没有进行分析，可以进行分析或者是修复。

单测问题列表:

需要确认的文件:

这类文件没有继承Dy2StTestBase但需要单独确认是否需要运行在 PIR API 和 SOT 下