torch.fx.experimental.proxy_tensor.maybe_enable_thunkify#

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

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