lightning/pytorch_lightning
thomas chaton 3a4c4246ee
[TPU] update is_tpu_exists utils internal logic to rely on xmp.spawn (#6719)
* update_logic

* update

* Update tests/utilities/test_xla_device_utils.py

* Update pytorch_lightning/utilities/xla_device.py

Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>

* Update pytorch_lightning/utilities/xla_device.py

Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>

* update test

* Update tests/utilities/test_xla_device_utils.py

* update

* Apply fix

* Docstring

* flake8

* update

Co-authored-by: Your Name <you@example.com>
Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
2021-03-29 18:59:20 +01:00
..
accelerators add copyr (#6661) 2021-03-24 14:29:46 +01:00
callbacks Remove legacy `Result` parameters (#6016) 2021-03-28 11:55:08 +02:00
core Remove legacy `Result` parameters (#6016) 2021-03-28 11:55:08 +02:00
distributed [bug] Update broadcast + reduce decision ModelCheckpoint] (#6410) 2021-03-14 17:14:27 +00:00
loggers Add artifcact_location arg to MLFlow logger (#6677) 2021-03-26 00:12:03 +01:00
metrics Simplify deprecations (#6620) 2021-03-25 15:26:38 +01:00
overrides Flash predict step (#6577) 2021-03-23 11:13:13 -04:00
plugins Fix checkpoint callback & Trainer.test(_) issue for TPUs (#6654) 2021-03-25 10:37:37 +00:00
profiler Do not describe when there's no summary (#6681) 2021-03-26 14:58:05 +00:00
trainer More explicit exception message when testing with fast_dev_run=True (#6667) 2021-03-29 13:29:54 +00:00
tuner Fix tuner.scale_batch_size not finding batch size attribute when using datamodule (#5968) 2021-03-14 09:16:19 +01:00
utilities [TPU] update is_tpu_exists utils internal logic to rely on xmp.spawn (#6719) 2021-03-29 18:59:20 +01:00
__init__.py refactoring setup (#6590) 2021-03-22 08:39:19 -04:00
info.py Docs/robots (#6658) 2021-03-24 10:46:56 +01:00
py.typed make PyTorch Lightning PEP 561 Compliant (#3187) 2020-09-09 13:37:03 -04:00
setup_tools.py refactoring setup (#6590) 2021-03-22 08:39:19 -04:00