torch.fx.experimental.proxy_tensor.maybe_enable_thunkify¶

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

在此上下文管理器中，如果您正在进行 make_fx 跟踪，我们将对 SymNode 计算进行 thunkify 处理，并避免将其跟踪到图中，除非实际需要。您应尽可能避免使用此功能，因为 SymNode 跟踪的懒加载可能会导致长链的 thunks，如果评估它们，可能会导致栈溢出。然而，由于 PT2 中存在一些有问题的部分，它们会因为 SymNode 计算跟踪不足而失败，并出现“s0 未通过代理跟踪”错误，因此目前有时有必要使用此功能。