trunk/bb078e405cb2b06bba31998be30425de333d94b7: [aot_compile] Rebuild AOTCompilePickler on FunctionPicklerBase (#196685)
- PyTorch: 832 events in the last 90 days
- PyTorch: 821th Release in the last 90 days
- Previous: earlier the same day · trunk/658b2ed8491d2b0836cbe113ead4a97688cbece4
What happened
torch.compile(fn).aot_compile(...).save_compiled_function(path) pickles everything the compiled function needs to run later, including the nested helpers it closes over. AOTCompilePickler had its own copies of the cell, module, code and bound-method reducers, and rebuilt a function from its code object only when its qualname carried : " marker def fn(x): return helper(x) + 1 torch.compile(fn, fullgraph=True).aot_com…
Summary assembled by rule from the sources below