fix deprecation

pytorch · Jul 11, 2023 · c304e06 · c304e06
1 parent 955a0fb
commit c304e06
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_xla/experimental/pjrt.py b/torch_xla/experimental/pjrt.py
@@ -7,9 +7,7 @@
 
 aliases = [
     runtime.addressable_device_count,
-    runtime.physical_device_attributes,
     runtime.device_type,
-    runtime.global_physical_device_attributes,
     runtime.global_device_count,
     runtime.global_ordinal,
     runtime.local_device_count,
@@ -28,6 +26,8 @@
 ]
 
 rendezvous = deprecated(this_module, xm.xla_rendezvous)
+device_attributes = deprecated(this_module, runtime.runtime_device_attributes)
+global_device_attributes = deprecated(this_module, runtime.global_runtime_device_attributes)
 
 for alias in aliases:
   register_deprecated(this_module, alias)