benchmark
94f54e87 - Add custom treespec fqn field

Commit
2 years ago
Add custom treespec fqn field Summary: X-link: https://github.com/pytorch/pytorch/pull/112428 Custom classes that are serialized with pytree are serialized by default with `f”{class.__module__}.{class.__name__}”`. This is a dependency from our serialized program directly into the outer Python environment. If a user moves the class to a different directory, the serialized program will be unable to be loaded. So, we will require users to pass in an FQN if they want to serialize their custom treespec type. cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx chenyang78 aakhundov kadeng avikchaudhuri gmagogsfm zhxchen17 tugsbayasgalan suo imported-using-ghimport Reviewed By: suo Differential Revision: D50886366 Pulled By: angelayi fbshipit-source-id: 276af112307e3d89137ff10af999ebeb5935a5ad
Author
Parents
Loading