lightning/tests/accelerators
four4fish cec2d7946b
3/n Move accelerator into Strategy (#11022)
* remove training_step() from accelerator

* remove test, val, predict step

* move

* wip

* accelerator references

* cpu training

* rename occurrences in tests

* update tests

* pull from adrian's commit

* fix changelog merge pro

* fix accelerator_connector and other updates

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* fix doc build and some mypy

* fix lite

* fix gpu setup environment

* support customized ttp and accelerator

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* fix tpu error check

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* fix precision_plugin initialization to recognisze cusomized plugin

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update bug_report_model.py

* Update accelerator_connector.py

* update changelog

* allow shorthand typing references to pl.Accelerator

* rename helper method and add docstring

* fix typing

* Update pytorch_lightning/trainer/connectors/accelerator_connector.py

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>

* Update tests/accelerators/test_accelerator_connector.py

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>

* Update tests/accelerators/test_cpu.py

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* fix pre commit complaint

* update typing to long ugly path

* spacing in flow diagram

* remove todo comments

* docformatter

* Update pytorch_lightning/plugins/training_type/training_type_plugin.py

* revert test changes

* improve custom plugin examples

* remove redundant call to ttp attribute

it is no longer a property

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Apply suggestions from code review

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2021-12-16 04:41:34 +00:00
..
__init__.py Automatically find and run special tests (#6669) 2021-03-26 17:04:59 +00:00
ddp_model.py Update strategy flag in docs (#10000) 2021-10-20 21:02:53 +05:30
test_accelerator_connector.py 3/n Move accelerator into Strategy (#11022) 2021-12-16 04:41:34 +00:00
test_common.py Add `auto_device_count` method to `Accelerators` (#10222) 2021-10-29 22:31:32 +02:00
test_cpu.py 3/n Move accelerator into Strategy (#11022) 2021-12-16 04:41:34 +00:00
test_ddp.py Rename special to standalone (#10779) 2021-11-26 17:13:14 +00:00
test_ddp_spawn.py Remove deprecated dataloader arguments in Trainer methods (#10325) 2021-11-04 11:03:39 +01:00
test_dp.py remove deprecated signature for `transfer_batch_to_device` (#10480) 2021-11-13 19:32:30 +00:00
test_gpu.py 3/n Move accelerator into Strategy (#11022) 2021-12-16 04:41:34 +00:00
test_ipu.py 3/n Move accelerator into Strategy (#11022) 2021-12-16 04:41:34 +00:00
test_multi_nodes_gpu.py Rename special to standalone (#10779) 2021-11-26 17:13:14 +00:00
test_tpu.py 3/n Move accelerator into Strategy (#11022) 2021-12-16 04:41:34 +00:00