pytorch
ef7d572a - Ensure ShardedTensor handles list/tuple appropriately as `size` parameter. (#62109)

Commit
4 years ago
Ensure ShardedTensor handles list/tuple appropriately as `size` parameter. (#62109) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62109 The `size` parameter only worked correctly for *args like invocation :10, 20 and not for list: [10, 20] and tuples: (10, 20). This PR ensures this works similar to `torch.empty`. ghstack-source-id: 134246166 Test Plan: 1) unit tests 2) waitforbuildbot Reviewed By: SciPioneer Differential Revision: D29884768 fbshipit-source-id: 7a4a3c5ed5d7c081344f6ead3170905b97fc652d
Author
Parents
Loading