torch.fx.experimental.proxy_tensor.maybe_enable_thunkify¶

torch.fx.experimental.proxy_tensor.maybe_enable_thunkify()[源代码]¶

在这个上下文管理器中，如果你正在进行 make_fx 跟踪，我们将对所有 SymNode 计算进行 thunkify，并避免将其跟踪到图中，除非它实际上需要。你应该尽量避免使用它，因为 SymNode 跟踪的延迟评估会导致很长的 thunk 链，如果你评估它们，就会导致堆栈溢出。但是，目前有时需要这样做，因为 PT2 中存在一些错误，由于 SymNode 计算的跟踪不足，这些错误会因“s0 未通过代理跟踪”错误而失败。