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 未用代理進行追蹤”的錯誤而失敗。