pytorch
e1d963e8 - model_dump: Fix memory computation when both constants and data tensors are present (#66006)

Commit
4 years ago
model_dump: Fix memory computation when both constants and data tensors are present (#66006) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/66006 Previously, this was resulting in a key collision and a crash. ghstack-source-id: 139342089 Test Plan: Ran webdriver test locally. Reviewed By: dhruvbird Differential Revision: D31281092 fbshipit-source-id: f31311726c681d6d7e0504ff8e84c888af9054f0
Author
Parents
Loading