torch.fx.experimental.proxy_tensor.maybe_enable_thunkify#

torch.fx.experimental.proxy_tensor.maybe_enable_thunkify()[source]#

在此上下文管理器中，如果您正在进行 `make_fx` 跟踪，我们将对所有 `SymNode` 计算进行“thunkify”处理，并避免将其跟踪到图中，除非实际需要。您应该尽量避免使用此功能，因为 `SymNode` 跟踪的惰性求值可能导致长链的 thunks，如果对其求值，可能会导致堆栈溢出。但是，这目前有时是必要的，因为 PT2 中存在一些有 bug 的部分，由于 `SymNode` 计算跟踪不足而会因“s0 未用代理跟踪”错误而失败。