
    sh                         S SK r S SKJr  SSKJr  SSKJrJr  \ R                  " \	5      r
\S 5       rS rSS	 jr\" \S
9r\" S\S9  g)    N)make_boxed_func   )aot_autograd   )register_backendregister_experimental_backendc                     [        XSS9$ )NFboxedxla_backend_helpermodelfake_tensor_inputss     s/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/_dynamo/backends/torchxla.pyopenxla_evalr      s    euEE    c                     [        XSS9$ )NTr
   r   r   s     r   openxla_eval_boxedr      s    etDDr   c                    ^ ^^  SS K Js  Jm  S mUUU 4S jnU(       a  [	        U5      $ U$ ! [         a  n[        S5      UeS nAff = f)Nr   z_Please follow the instruction in https://github.com/pytorch/xla#pytorchxla to install torch_xlac                  :   > Tc  TR                  TU 5      m@T" U 6 $ )N)extract_compiled_graph)argsbridgecompiled_graphr   s    r   fwdxla_backend_helper.<locals>.fwd!   s+     !#::5$GNt$$r   )torch_xla.core.dynamo_bridgecoredynamo_bridgeImportErrorr   )r   r   r   er   r   r   s   `    @@r   r   r      sS    55 N% $)?31c1  m
	s   	, 
AAA)fw_compileropenxla)namecompiler_fn)F)loggingfunctorch.compiler   backends.commonr   registryr   r   	getLogger__name__logr   r   r   r$    r   r   <module>r/      sa     - * E ! F FE2* " iW 5r   