pytorch

268160ee - Add context manager to save tensors on CPU

Add context manager to save tensors on CPU Fix #57100 ghstack-source-id: a9e56df9653a4240b4d382f7dbec3080a3a07a3d Pull Request resolved: https://github.com/pytorch/pytorch/pull/61928