torch.fx.experimental.proxy_tensor.maybe_enable_thunkify¶

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

在此上下文管理器中，如果您正在進行 make_fx 跟蹤，我們將對所有 SymNode 計算進行 thunkify 處理，並避免將其跟蹤到圖中，除非實際需要。您應儘量避免使用此功能，因為 SymNode 跟蹤的惰性求值可能導致冗長的 thunk 鏈，在求值時會發生堆疊溢位。然而，目前有時這是必要的，因為 PT2 中存在一些錯誤的部分，由於 SymNode 計算跟蹤不足，會導致出現“s0 is not tracked with proxy”錯誤。