Closed AsakusaRinne closed 1 year ago
Pytorch中的fx之所以能自动wrap所有的pytorch function是因为pytorch里面有个__pytorch_funtion__
的机制,由于one-fx只是个插件,不太能直接照抄,所以对于这个问题目前有两种方案:
'oneflow._oneflow_internal._C'
),如果是,那么就走wrap的分支。目前准备先采用第二种。
为了方便后续发包测试和添加文档,先merge掉这个PR,保留commit信息,后续有需要可以回退。
初始化框架,适配OneFlow0.8.0,目前的计划事项如下:
目前遇到的问题:
Importer
和ScriptObject
,会对某些情况造成影响。