PaddleSOT
PaddleSOT copied to clipboard
优化 SIR call Layer 功能,实现黑白名单机制
为了能够让 Paddle 内置的 Layer 中的代码不进行 Fallback,打断子图,我们扩展了 SIR 的功能,使 SIR 能够支持 call Layer,这些 Layer 会直接动转静,而不会跑字节码收集组网代码,我们默认会将 Paddle 的内置 Layer 放到 SIR call Layer 中,可以充分利用了 Paddle 内置 Layer 动静统一的特性,而用户自定义的 Layer 则会仍然跑字节码
为了能够更加自由地控制一个 Layer 是否跑字节码,我们可以增加黑白名单机制,以更加确定性地指定一个 Layer 直接动转静还是跑字节码