support backward compatibility of the old imports

pytorch · Nov 10, 2023 · 2ac9dc0 · 2ac9dc0
1 parent 09cb218
commit 2ac9dc0
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/torch_xla/experimental/xla_sharded_tensor.py b/torch_xla/experimental/xla_sharded_tensor.py
@@ -0,0 +1,10 @@
+# Keep this for backward compatibility.
+# TODO(yeounoh) remove after 2.2 release.
+import warnings
+
+warnings.warn(
+    "Importing from `torch_xla.experimental.xla_sharded_tensor` will be deprecated "
+    "after 2.2 release. Please use `torch_xla.experimental.spmd` "
+    "instead.", DeprecationWarning, 2)
+
+from .spmd.xla_sharded_tensor import *
diff --git a/torch_xla/experimental/xla_sharding.py b/torch_xla/experimental/xla_sharding.py
@@ -0,0 +1,10 @@
+# Keep this for backward compatibility.
+# TODO(yeounoh) remove after 2.2 release.
+import warnings
+
+warnings.warn(
+    "Importing from `torch_xla.experimental.xla_sharding` will be deprecated "
+    "after 2.2 release. Please use `torch_xla.experimental.spmd` instead.",
+    DeprecationWarning, 2)
+
+from .spmd.xla_sharding import *