Carlos Mocholí 729e833935 [TPU] XLAFSDP checkpointing fixes (#18424) ... Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>		2023-08-31 05:47:11 -07:00
..
launchers	Improve error message when "if name == main" guard is needed (#18298)	2023-08-18 08:53:01 -04:00
__init__.py	…
test_ddp.py	docformatter: config with black (#18064)	2023-08-09 10:44:20 -04:00
test_ddp_integration.py	Fix state dict loading for ddp/dp in Fabric (#17997)	2023-07-06 13:47:17 +02:00
test_deepspeed.py	docformatter: config with black (#18064)	2023-08-09 10:44:20 -04:00
test_deepspeed_integration.py	docformatter: config with black (#18064)	2023-08-09 10:44:20 -04:00
test_dp.py	add & apply flake8-simplify (#17386)	2023-04-24 21:57:08 +00:00
test_fsdp.py	Fix saving FSDP checkpoint when world size = 1 and torch <= 2.0 (#18371)	2023-08-23 06:45:20 -04:00
test_fsdp_integration.py	Meta device initialization for FSDP in Fabric (#18122)	2023-08-02 07:58:32 -04:00
test_registry.py	[TPU] Add Fabric support for PyTorch XLA FSDP (#18126)	2023-08-02 12:56:00 -04:00
test_single_device.py	Support empty weight initialization in `Fabric.init_module()` (#17627)	2023-06-07 18:33:53 +00:00
test_strategy.py	docformatter: config with black (#18064)	2023-08-09 10:44:20 -04:00
test_xla.py	[TPU] Preserve the device with XLA's collectives (#18275)	2023-08-16 22:56:41 +02:00
test_xla_fsdp.py	[TPU] Support setting the XLAFSDP policy with a set (#18430)	2023-08-30 09:25:12 -07:00
test_xla_fsdp_integration.py	[TPU] XLAFSDP checkpointing fixes (#18424)	2023-08-31 05:47:11 -07:00