pytorch
56c0c0af - [ShardedTensor] Add `is_floating_point` (#85483)

Commit
3 years ago
[ShardedTensor] Add `is_floating_point` (#85483) This adds `is_floating_point()` support to `ShardedTensor`. This is needed for `ShardedTensor` + FSDP. Pull Request resolved: https://github.com/pytorch/pytorch/pull/85483 Approved by: https://github.com/wanchaol
Author
Andrew Gu
Committer
Parents
Loading