Carlos Mocholí 8dac251273 [TPU] Fix PjRT tests (#17408)		2023-04-19 16:39:00 +02:00
..
launchers	Merge DDPStrategy and DDPSpawnStrategy in PL (#16809)	2023-02-27 14:43:23 -05:00
scripts	copyright Lightning AI team (#16647)	2023-02-06 15:26:51 +01:00
__init__.py	Future 4/n: test & legacy in test/ folder (#13295)	2022-06-15 18:10:49 -04:00
test_common.py	Remove DP (#16748)	2023-02-16 05:12:08 +00:00
test_custom_plugin.py	copyright Lightning AI team (#16647)	2023-02-06 15:26:51 +01:00
test_ddp.py	Call `_cuda_clearCublasWorkspaces` on teardown (#16907)	2023-03-06 17:52:02 +01:00
test_ddp_spawn.py	Add `Strategy.on_exception` (#16646)	2023-02-08 14:00:31 +00:00
test_ddp_spawn_strategy.py	Replace obsolete `_FakeQueue` in multiprocessing launcher (#16873)	2023-02-28 01:31:18 +00:00
test_ddp_strategy.py	[TPU] Fix PjRT tests (#17408)	2023-04-19 16:39:00 +02:00
test_ddp_strategy_with_comm_hook.py	Merge DDPStrategy and DDPSpawnStrategy in PL (#16809)	2023-02-27 14:43:23 -05:00
test_deepspeed_strategy.py	[TPU] Fix PjRT tests (#17408)	2023-04-19 16:39:00 +02:00
test_fsdp.py	Wraps sharded model for proper access to it `state_dict` in `FSDP` strategy (#16558)	2023-04-18 01:03:43 +02:00
test_registry.py	[TPU] v4 support (#17227)	2023-04-11 22:24:11 +00:00
test_single_device_strategy.py	Set a fixed stage in the evaluation loops (#17094)	2023-03-27 10:00:03 +02:00
test_xla.py	[TPU] Fix PjRT tests (#17408)	2023-04-19 16:39:00 +02:00