69833dad5b
* 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> |
||
---|---|---|
.. | ||
__init__.py | ||
dist.py | ||
parsing.py | ||
test_apply_func.py | ||
test_apply_func_torchtext.py | ||
test_dtype_device_mixin.py | ||
test_upgrade_checkpoint.py | ||
test_xla_device_utils.py |