diff --git a/torch/_dynamo/backends/torchxla.py b/torch/_dynamo/backends/torchxla.py index 6b5645d4b90..32e1aba8ec6 100644 --- a/torch/_dynamo/backends/torchxla.py +++ b/torch/_dynamo/backends/torchxla.py @@ -1,7 +1,6 @@ # mypy: ignore-errors import logging -import warnings from functorch.compile import make_boxed_func @@ -12,20 +11,6 @@ log = logging.getLogger(__name__) @register_experimental_backend -def torchxla_trivial(gm, fake_tensor_inputs): - return gm - - -@register_experimental_backend -def torchxla_trace_once(model, fake_tensor_inputs): - warnings.warn( - "This backend will be deprecated in 2.2, please use `openxla` backend instead" - ) - - return xla_backend_helper(model, fake_tensor_inputs) - - -@register_backend def openxla_eval(model, fake_tensor_inputs): return xla_backend_helper(model, fake_tensor_inputs, boxed=False) @@ -55,20 +40,6 @@ def xla_backend_helper(model, fake_tensor_inputs, boxed=False): return make_boxed_func(fwd) if boxed else fwd -aot_torchxla_trivial = aot_autograd( - fw_compiler=torchxla_trivial, -) -register_experimental_backend( - name="aot_torchxla_trivial", compiler_fn=aot_torchxla_trivial -) - -aot_torchxla_trace_once = aot_autograd( - fw_compiler=torchxla_trace_once, -) -register_experimental_backend( - name="aot_torchxla_trace_once", compiler_fn=aot_torchxla_trace_once -) - openxla = aot_autograd( fw_compiler=openxla_eval_boxed, )