search

Oneflow-Inc / one-fx

A toolkit for developers to simplify the transformation of nn.Module instances. It's now corresponding to Pytorch.fx.
Other
13 stars 1 forks source link

Optimize global wrap. #16

Closed AsakusaRinne closed 1 year ago

AsakusaRinne commented 1 year ago
  1. 优化global_wrap的实现
  2. 添加global_wrap_oneflow_all这个API（作为以后可能遇到问题的一种临时解决方案）
  3. 修复fx.GraphModule单独对Proxy进行执行会遇到代理问题的错误（#15）