When Metal is used, the tensors are uploaded to the device in use. We don't Need to keep the host copy. This does wonder to RSS (roughly divides by two). Differential Revision: https://phabricator.services.mozilla.com/D324041
When Metal is used, the tensors are uploaded to the device in use. We don't Need to keep the host copy. This does wonder to RSS (roughly divides by two). Differential Revision: https://phabricator.services.mozilla.com/D324041