PaddleSOT icon indicating copy to clipboard operation
PaddleSOT copied to clipboard

扩展 Fallback 场景,允许用户标记某一个函数来强制跑动态图

Open SigureMo opened this issue 2 years ago • 0 comments

刚刚 @Aurelius84 提到的,暂时记一下~

增加一个装饰器,以使某一个特定函数跑动态图

@fallback_to_dygraph
def b():
    ...

def foo()
    a()   # <---- 跑正常 Symbolic Trace 逻辑
    b()   # <---- 强制跑动态图
    c()   # <---- 跑正常 Symbolic Trace 逻辑

看样子这个函数好像和 print 没什么区别?应该可以复用 print 对应的 Fallback 逻辑?

~~本需求和 #87 是否相似,接口上能否统一什么的?~~ 好像不太一样

SigureMo avatar May 25 '23 02:05 SigureMo