Update debugging.py

pytorch · Nov 29, 2023 · d46323b · d46323b
1 parent f4e44ea
commit d46323b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch_xla/distributed/spmd/debugging.py b/torch_xla/distributed/spmd/debugging.py
@@ -10,6 +10,7 @@
 import torch_xla.core.xla_model as xm
 import torch_xla.runtime as xr
 import torch_xla.experimental.xla_sharding as xs
+import torch_xla.core.xla_env_vars as xenv
 import torch_xla.utils.utils as xu
 from torch_xla.experimental.xla_sharded_tensor import XLAShardedTensor