moved tpu_device_exists to lightning.py

Lightning-AI · Aug 31, 2020 · f52321a · f52321a
1 parent e150678
commit f52321a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py
@@ -39,8 +39,9 @@
 from pytorch_lightning.utilities.device_dtype_mixin import DeviceDtypeModuleMixin
 from pytorch_lightning.utilities.parsing import AttributeDict, collect_init_args, get_init_args
 from pytorch_lightning.core.step_result import TrainResult, EvalResult
-from pytorch_lightning.utilities.xla_device_utils import TPU_AVAILABLE
+from pytorch_lightning.utilities.xla_device_utils import tpu_device_exists
 
+TPU_AVAILABLE = tpu_device_exists()
 if TPU_AVAILABLE:
     import torch_xla.core.xla_model as xm
 

diff --git a/pytorch_lightning/utilities/xla_device_utils.py b/pytorch_lightning/utilities/xla_device_utils.py
@@ -46,5 +46,3 @@ def tpu_device_exists():
     else:
         return False
 
-
-TPU_AVAILABLE = tpu_device_exists()
Original file line number	Diff line number	Diff line change
Expand Up		@@ -46,5 +46,3 @@ def tpu_device_exists():
		else:
		return False


		TPU_AVAILABLE = tpu_device_exists()