ó
    Eñi  ã                   ó–  • S SK r S SKJr  S SKJr  S SKrS SKJr  S SKJr  SSK	J
r
  SS	KJrJrJr  \ R                  " \5      r\S
\R$                  S\\R(                     S\4S j5       rS
\R$                  S\\R(                     S\S\4   4S jr SS
\R$                  S\\R(                     S\S\S\4   4S jjr\
" \S9r\" S\S9  g)é    N)ÚCallable)ÚAny)Úmake_boxed_func)Úfxé   )Úaot_autogradé   )Ú
CompiledFnÚregister_backendÚregister_experimental_backendÚmodelÚfake_tensor_inputsÚreturnc                 ó   • [        XSS9$ )NF©Úboxed©Úxla_backend_helper©r   r   s     Ú\/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/_dynamo/backends/torchxla.pyÚopenxla_evalr      s   € ô ˜e¸uÑEÐEó    .c                 ó   • [        XSS9$ )NTr   r   r   s     r   Úopenxla_eval_boxedr      s   € ô ˜e¸tÑDÐDr   r   c                 óÄ   ^ ^^•  SS K Js  Jm  S mS[        R
                  S[        4UUU 4S j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_xlaÚargsr   c                  ó:   >• Tc  TR                  TU 5      m@T" U 6 $ )N)Úextract_compiled_graph)r   ÚbridgeÚcompiled_graphr   s    €€€r   ÚfwdÚxla_backend_helper.<locals>.fwd)   s+   ø€ ð Ñ!Ø#×:Ñ:¸5À$ÓGˆNØÙ˜tÐ$Ð$r   )Útorch_xla.core.dynamo_bridgeÚcoreÚdynamo_bridgeÚImportErrorÚtorchÚTensorr   r   )r   r   r   Úer!   r   r    s   `    @@r   r   r      sj   ú€ ðß5Ð5ð €Nð%”5—<‘<ð %¤C÷ %ñ %ö $)Œ?˜3ÓÐ1¨cÐ1øô ó ÜØmó
àð	ûðús   …	A Á
AÁAÁA)Úfw_compilerÚopenxla)ÚnameÚcompiler_fn)F)ÚloggingÚcollections.abcr   Útypingr   r'   Úfunctorch.compiler   r   Úbackends.commonr   Úregistryr
   r   r   Ú	getLoggerÚ__name__ÚlogÚGraphModuleÚlistr(   r   r   Úboolr   r+   © r   r   Ú<module>r;      s  ðÛ Ý $Ý ã Ý -Ý å *ß QÑ Qð ×Ò˜Ó!€ð ðFØ�>‰>ðFØ/3°E·L±LÑ/AðFàóFó ðFðEØ�>‰>ðEØ/3°E·L±LÑ/AðEàˆc�3ˆhÑôEð RWñ2Ø�>‰>ð2Ø/3°E·L±LÑ/Að2ØJNð2àˆc�3ˆhÑõ2ñ. Ø"ñ€ñ �i¨WÓ 5r   