Commit be879f4
committed
Skip packing GraphModule.recompile() codegen bookkeeping as module state
torch.fx.GraphModule.recompile() writes `_code`/`_lineno_map` (and
`_in_spec`/`_out_spec` for pytree codegen) straight into the module's
__dict__. Thunder's attribute-modification tracking mistook these
codegen-cache internals for real module state and tried to pack them
into the epilogue trace, asserting when the value was a plain str/dict
instead of a Proxy. Fixes test_higher_order_inplace_alias_update
failing intermittently in CI depending on test order/dynamo cache
state.1 parent 5dd55d3 commit be879f4
1 file changed
Lines changed: 6 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
2063 | 2063 | | |
2064 | 2064 | | |
2065 | 2065 | | |
2066 | | - | |
| 2066 | + | |
| 2067 | + | |
| 2068 | + | |
| 2069 | + | |
| 2070 | + | |
| 2071 | + | |
2067 | 2072 | | |
2068 | 2073 | | |
2069 | 2074 | | |
| |||
0 commit comments