benchmark
60a167b4 - Only store necessary tensor_dict fields in node meta (#132805)

Commit
1 year ago
Only store necessary tensor_dict fields in node meta (#132805) Summary: Fixes #132290 This PR attempts a more invasive / complete solution than the one from #132338, which removes immediate tensor fields from the `tensor_dict` copy stored in node meta. The approach taken here is to store only those fields of the `tensor_dict` which are absolutely utilized somewhere else. So far, this appears to be limited to: * `_dynamo_static_input_type` * `tag` (at least in the tests). Discussion at #94080 appears to indicate this is depended on for export (CI may point out more) X-link: https://github.com/pytorch/pytorch/pull/132805 Approved by: https://github.com/mlazos Reviewed By: izaitsevfb, PaliC Differential Revision: D60945621 Pulled By: jbschlosser fbshipit-source-id: ba29dba760c3fc492c8b4cfeb0ca8a8d70f4df1f
Author
Parents
Loading