oneflow
oneflow copied to clipboard
[bugfix] enable faster rcnn and sd model with oneflow backend
oneflow backend 对接 torch compile ,在关闭和打开动态形状的时候,跑通了 faster rcnn 和 sd 模型。相关 issue: oneflow backend 对接 torch compile ,运行 faster rcnn
主要改动包括:
- 修复 oneflow 模型转 torch 模型时, 部分 torch.nn.functional.func 转换失败的 bug
- 在 oneflow backend 中打开 nn.Graph 的动态形状支持,环境变量对其 oneflow compile
- 在 oneflow backend 中对推理场景添加了 flow.no_grad ,避免了编译时错误:RuntimeError: The gradient function for op fused_multi_head_attention_inference is not found. Please check whether it has been implemented and registered correctly.
- 补全了对 nn.Graph 的不同返回数据类型的处理