lightning/tests/utilities
Lezwon Castelino 69833dad5b
Added check to verify xla device is TPU (#3274)
* tpu device check

* replaced with xmp spawn

* Revert "replaced with xmp spawn"

This reverts commit 6835380f

* replaced all instances of XLA_AVAILABLE

* moved inner_f to global scope

* made refactors

* added changelog

* added TPU_AVAILABLE variable

* fix codefactor issues

* removed form trainer and early stopping

* add TORCHXLA_AVAILABLE check

* added tests

* refactoring

* Update pytorch_lightning/utilities/xla_device_utils.py

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* updated function names

* fixed bug

* updated CHANGELOG.md

* added todo

* added type hints

* isort and black

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-10-06 19:54:37 +02:00
..
__init__.py New metric classes (#1326) (#1877) 2020-05-19 11:05:07 -04:00
dist.py ref: part a of #3733 (#3766) 2020-10-01 08:15:23 -04:00
parsing.py Bugfix: Lr finder and hparams compatibility (#2821) 2020-08-07 00:34:48 +02:00
test_apply_func.py New metric classes (#1326) (#1877) 2020-05-19 11:05:07 -04:00
test_apply_func_torchtext.py Bugfix/torchtext include lengths (#2689) 2020-07-31 07:53:08 -04:00
test_dtype_device_mixin.py ref: move backends back to individual files (1/5) (ddp_cpu) (#3712) 2020-09-29 01:59:18 -04:00
test_upgrade_checkpoint.py callback method for on_save_checkpoint (#2501) 2020-08-28 16:50:52 +02:00
test_xla_device_utils.py Added check to verify xla device is TPU (#3274) 2020-10-06 19:54:37 +02:00