New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

VisualDL适配Paddle PIR #1

Open

cse0001 wants to merge 2 commits into main from develop

Owner

cse0001 commented Sep 11, 2024

开源之夏飞桨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测例

其他改动

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

运行测例

cd VisualDL
export FLAGS_enable_pir_api=1
python demo/components/pir_program_test.py （pir_graph_test，cond_test，while_test，cond_inside_cond_test）输出文件将在VisualDL/log/cond_test路径下
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

全展开后

cse0001 added 2 commits

September 10, 2024 12:27


          VisualDL Adaptation for Paddle PIR

0c33fad


          [Doc] Add document for VisualDL Adaptation for Paddle PIR

1af4e05

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet