Ë
    ÃÓ^j  ã                   óš  — d dl Z d dlmZ d dlmZ d dlZd dlmZ d dlmZ ddl	m
Z
 dd	lmZmZmZ  e j                  e«      Zed
ej$                  deej(                     defd„«       Zd
ej$                  deej(                     dedef   fd„Z	 dd
ej$                  deej(                     dededef   fd„Z e
e¬«      Z ede¬«       y)é    N)ÚCallable)ÚAny)Úmake_boxed_func)Úfxé   )Úaot_autogradé   )Ú
CompiledFnÚregister_backendÚregister_experimental_backendÚmodelÚfake_tensor_inputsÚreturnc                 ó   — t        | |d¬«      S )NF©Úboxed©Úxla_backend_helper©r   r   s     új/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/_dynamo/backends/torchxla.pyÚopenxla_evalr      s   € ô ˜eÐ%7¸uÔEÐEó    .c                 ó   — t        | |d¬«      S )NTr   r   r   s     r   Úopenxla_eval_boxedr      s   € ô ˜eÐ%7¸tÔDÐDr   r   c                 ó´   ‡ ‡‡— 	 dd l mc mŠ d Šdt        j
                  dt        fˆˆˆ fd„}|rt        |«      S |S # t        $ r}t        d«      |‚d }~ww xY w)Nr   z_Please follow the instruction in https://github.com/pytorch/xla#pytorchxla to install torch_xlaÚargsr   c                  ó8   •— ‰€‰j                  ‰| «      Š‹ ‰| Ž S )N)Úextract_compiled_graph)r   ÚbridgeÚcompiled_graphr   s    €€€r   Úfwdz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      se   ú€ ðß5Ð5ð €Nð%”5—<‘<ð %¤C÷ %ñ $)Œ?˜3ÓÐ1¨cÐ1øô ò ÜØmó
àð	ûðús   …	= ½	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ð €g×Ñ˜Ó!€ð ð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   