trunk/bb078e405cb2b06bba31998be30425de333d94b7: [aot_compile] Rebuild AOTCompilePickler on FunctionPicklerBase (#196685)
- PyTorch 近 90 天出现 833 次
- PyTorch 近 90 天第 822 次发版
- 上一次:同一天稍早 · trunk/658b2ed8491d2b0836cbe113ead4a97688cbece4
发生了什么
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…
摘要按规则整理自下方来源原文