VisualDL适配Paddle PIR

[cse0001]

开源之夏 飞桨PaddlePaddle-PIR适配VisualDL模型可视化

相关链接

• 【开源之夏】PIR适配VisualDL模型可视化 PaddlePaddle/VisualDL#1268

主要改动

• 添加get_sub_var和get_sub_op获取子block的算子和变量
• 修改analyse_pir支持PIR 控制流算子可视化（ifop，whileop）
• 修改analyse_pir增加计算图边信息的分析
• 修改translate_graph支持输入PIR静态图和动态图
• 添加cond_test，while_test，cond_inside_cond_test，pir_graph_test测例
• 修改前后端接口，动态图界面更换模型功能支持json格式模型

其他改动

• 修改pir_translate测例为pir_program_test测例，测试PIR静态计算图可视化
• 修改component/graph/init.py，暴露analyse_pir API方便下一步支持json格式计算图可视化
• 修复其他已知BUG

运行测例

1. cd VisualDL
2. export FLAGS_enable_pir_api=1
3. python demo/components/pir_program_test.py （pir_graph_test，cond_test，while_test，cond_inside_cond_test）输出文件将在VisualDL/log/cond_test路径下
4. visualdl --logdir ./log/program_test/ --host 0.0.0.0 注意此时在VisualDL目录下

当前效果

运行测例后在http://0.0.0.0:8040/查看可视化计算图

pir_program_test（PIR静态计算图可视化）

pir_graph_test（输入PIR动态图可视化）

cond_test（ifop可视化）

全展开后

while_test（whileop可视化）

全展开后

cond_inside_cond_test（双层ifop嵌套结构可视化）

展开一层ifop

全展开后

[changeyoung98]

LGTM