From 777b97ffc306e63a415a1144ccdbf6bb6e1cdb21 Mon Sep 17 00:00:00 2001 From: Amithrajith Mamidala Date: Mon, 25 Mar 2024 09:36:50 -0700 Subject: [PATCH] Fix import --- torch_xla/core/xla_model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torch_xla/core/xla_model.py b/torch_xla/core/xla_model.py index 915db2e6e7b..00aa3f2cf28 100755 --- a/torch_xla/core/xla_model.py +++ b/torch_xla/core/xla_model.py @@ -17,6 +17,7 @@ import torch_xla.debug.metrics_saver as ms import torch_xla.utils.utils as xu import torch_xla.utils.closures as xc +import os _DEVICES = xu.LazyProperty(lambda: torch_xla._XLAC._xla_get_devices())