oneflow icon indicating copy to clipboard operation
oneflow copied to clipboard

[bugfix] enable faster rcnn and sd model with oneflow backend

Open crazy-JiangDongHua opened this issue 1 year ago • 0 comments

oneflow backend 对接 torch compile ,在关闭和打开动态形状的时候,跑通了 faster rcnn 和 sd 模型。相关 issue: oneflow backend 对接 torch compile ,运行 faster rcnn

主要改动包括:

  1. 修复 oneflow 模型转 torch 模型时, 部分 torch.nn.functional.func 转换失败的 bug
  2. 在 oneflow backend 中打开 nn.Graph 的动态形状支持,环境变量对其 oneflow compile
  3. 在 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.
  4. 补全了对 nn.Graph 的不同返回数据类型的处理

crazy-JiangDongHua avatar Mar 05 '24 14:03 crazy-JiangDongHua