pytorch/test/distributed/_composable
Andrew Gu 90fb7c36ab [FSDP2] Clamp reduce_dtype in lazy init (#143297)
fixes https://github.com/pytorch/pytorch/issues/143277 by moving the clamp of `reduce_dtype` to `None` to lazy init (same place as where `param_dtype` can be clamped to `None`)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/143297
Approved by: https://github.com/weifengpy
2024-12-17 00:25:08 +00:00
..
fsdp [FSDP2] Clamp reduce_dtype in lazy init (#143297) 2024-12-17 00:25:08 +00:00
test_composability [distributed] Remove unused variable in test_composability/test_pp_composability.py (#143191) 2024-12-14 12:23:44 +00:00
test_checkpoint.py Fix unused Python variables in test/[a-d]* (#134665) 2024-12-13 22:13:12 +00:00
test_contract.py
test_replicate.py [FSDP2] Move to public torch.distributed.fsdp (#141868) 2024-12-07 01:24:28 +00:00
test_replicate_with_compiler.py Fix unused Python variables in test/[a-d]* (#134665) 2024-12-13 22:13:12 +00:00