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

对oneflow函数的自动代理加入cache #14

Closed AsakusaRinne closed 1 year ago

AsakusaRinne commented 1 year ago

当前在每次tracer初始化的时候默认代理oneflow, oneflow.nn.functionaloneflow._C这三个空间下的函数,虽然能work但是可能会比较影响效率(但应该不影响变换后的模型的效率),后续应该考虑在这里加入cache。