Ë
    éÓ^jš  ã                   ó$   — d Z ddlZddlZd„ Zd„ Zy)zP JIT scripting/tracing utils

Hacked together by / Copyright 2020 Ross Wightman
é    Nc                  ó   — t        t        j                  d«      sJ d«       ‚t        j                  j                  d«       t        j                  j	                  d«       t        j                  j                  d«       y)zÝ Set JIT executor to legacy w/ support for op fusion
    This is hopefully a temporary need in 1.5/1.5.1/1.6 to restore performance due to changes
    in the JIT executor. These API are not supported so could change.
    Ú_jit_set_profiling_executorzOld JIT behavior doesn't exist!FTN)ÚhasattrÚtorchÚ_Cr   Ú_jit_set_profiling_modeÚ_jit_override_can_fuse_on_gpu© ó    úY/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/timm/utils/jit.pyÚset_jit_legacyr   
   sU   € ô ”5—8‘8Ð:Ô;Ð^Ð=^Ó^Ð;Ü	‡H�H×(Ñ(¨Ô/Ü	‡H�H×$Ñ$ UÔ+Ü	‡H�H×*Ñ*¨4Õ0r   c                 óŒ  — | dk(  r¼t         j                  j                  d«       t         j                  j                  d«       t         j                  j	                  d«       t         j                  j                  d«       t         j                  j                  d«       	 t         j                  j                  d«       y | dk(  s| dk(  r�t         j                  j                  d«       t         j                  j                  d«       t         j                  j                  d«       t         j                  j                  d«       	 t         j                  j                  d«       y | dk(  s| dk(  �r)dt        j                  d	<   t         j                  j                  d«       t         j                  j                  d«       t         j                  j                  d«       t         j                  j                  «        t         j                  j                  «        t         j                  j	                  d«       t         j                  j                  d«       t         j                  j                  d«       t         j                  j                  d«       y J d
| › d�«       ‚# t        $ r Y y w xY w# t        $ r Y y w xY w)NÚteTFÚoldÚlegacyÚnvfuserÚnvfÚ1Ú PYTORCH_NVFUSER_DISABLE_FALLBACKzInvalid jit fuser (ú))r   r   r   r   Ú_jit_override_can_fuse_on_cpur	   Ú_jit_set_texpr_fuser_enabledÚ_jit_set_nvfuser_enabledÚ	ExceptionÚosÚenvironÚ_jit_can_fuse_on_cpuÚ_jit_can_fuse_on_gpuÚ_jit_set_nvfuser_guard_mode)Úfusers    r   Úset_jit_fuserr!      sã  € Ø�‚}ä�‰×,Ñ,¨TÔ2Ü�‰×(Ñ(¨Ô.Ü�‰×.Ñ.¨uÔ5Ü�‰×.Ñ.¨tÔ4Ü�‰×-Ñ-¨dÔ3ð	Ü�H‰H×-Ñ-¨eÕ4ð 
�%Š˜5 HÒ,Ü�‰×,Ñ,¨UÔ3Ü�‰×(Ñ(¨Ô/Ü�‰×.Ñ.¨tÔ4Ü�‰×-Ñ-¨eÔ4ð	Ü�H‰H×-Ñ-¨eÕ4ð 
�)Ò	˜u¨›~Ø9<Œ�
‰
Ð5Ñ6ô 	�‰×-Ñ-¨eÔ4Ü�‰×,Ñ,¨TÔ2Ü�‰×(Ñ(¨Ô.Ü�‰×%Ñ%Ô'Ü�‰×%Ñ%Ô'Ü�‰×.Ñ.¨uÔ5Ü�‰×.Ñ.¨uÔ5Ü�‰×,Ñ,¨TÔ2Ü�‰×)Ñ)¨$Õ/à4Ð+¨E¨7°!Ð4Ó4ˆuøô3 ò 	Ùð	ûô ò 	Ùð	ús$   Â"J( Å	J7 Ê(	J4Ê3J4Ê7	KËK)Ú__doc__r   r   r   r!   r
   r   r   ú<module>r#      s   ðñó 
ã ò	1ó#5r   