torch.fx.experimental.proxy_tensor.maybe_enable_thunkify¶
- torch.fx.experimental.proxy_tensor.maybe_enable_thunkify()[源代码][源代码]¶
在此上下文管理器中,如果您正在进行 make_fx 追踪,我们将 thunkify 所有 SymNode 计算,并避免将其追踪到图中,除非实际需要。您应该尽可能避免使用此功能,因为 SymNode 追踪的延迟评估可能会导致长 thunk 链,如果您评估它们,将会导致堆栈溢出。但是,目前有时这是必要的,因为 PT2 的某些错误部分会因 SymNode 计算追踪不足而导致 “s0 is not tracked with proxy” 错误。
- 返回类型
Generator[None, None, None]