From a3180b2e5d6a332b48087813825edbaf621d0c84 Mon Sep 17 00:00:00 2001 From: Mason Chang Date: Thu, 29 Feb 2024 14:25:49 +0000 Subject: [PATCH] Run YAPF on xla_model.py --- torch_xla/core/xla_model.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/torch_xla/core/xla_model.py b/torch_xla/core/xla_model.py index b5f82675647..b1c7d4c3023 100755 --- a/torch_xla/core/xla_model.py +++ b/torch_xla/core/xla_model.py @@ -18,10 +18,12 @@ import torch_xla.utils.utils as xu import torch_xla.utils.closures as xc + def _lazy_get_device(): runtime._maybe_select_default_device() return torch_xla._XLAC._xla_get_devices() + _DEVICES = xu.LazyProperty(_lazy_get_device) REDUCE_SUM = 'sum'