a HSicCã@s¼UdZddlZddlZddlZGdd„dƒZeƒZdd„Zdd„Zej  ¡Z iZ iZ d d „Z d d „Zd d„Zdd„Ze ¡Zejed<e ¡Zejed<dd„Zdd„Zdd„Zdd„ZdS)zÃJIT-related state This module stores various pieces of Python-global state relating to the JIT. This is not intended to be imported directly; please the exposed functionalities in `torch.jit`. éNc@s(eZdZdZdd„Zdd„Zdd„ZdS) Ú EnabledProxyzyStores whether the JIT is enabled or not. This is just a wrapper for a bool, so that we get reference semantics cCs| dddd¡|_dS)NZ PYTORCH_JITTz> Using PyTorch JITz> PyTorch JIT DISABLED)Ú parse_envÚenabled©Úself©rúL/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/jit/_state.pyÚ__init__sÿzEnabledProxy.__init__cCsrtj |¡}|dur|S| ¡dvr(dS| ¡dvr8dS|dkrLt|ƒdS|dkr`t|ƒdStd |¡ƒ‚dS)N>ÚtrueÚ1ÚyesT>Ú0ÚfalseÚnoFZ1vZ0vz(Unknown setting of {}. Try using 0 or 1.)ÚosÚenvironÚgetÚlowerÚprintÚ ValueErrorÚformat)rÚnameÚdefaultZ true_messageZ false_messageÚvaluerrrrs   zEnabledProxy.parse_envcCs|jS©N)rrrrrÚ__bool__'szEnabledProxy.__bool__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__r rrrrrrr srcCs dt_dS)NF©Ú_enabledrrrrrÚdisable.sr"cCs dt_dS)NTr rrrrÚenable2sr#cCs|t|<|t| ¡<dSr)Ú_script_classesÚ_name_to_pyclassÚqualified_name)Ú python_classÚ script_classrrrÚ_add_script_classAsr)cCs(t|ddƒ}|durt|ƒ}t |d¡S)NÚ_jit_override_qualname)ÚgetattrÚ_get_python_classr$r)r'ÚoverriderrrÚ_get_script_classFs r.cCs t |d¡Sr)r%r©r&rrrr,Msr,cCst ¡t ¡dSr)r$Úclearr%rrrrÚ_clear_class_stateQsr1Ú_jit_caching_layerÚ_jit_function_overload_cachingcCs&t |d¡}|rdd„|DƒSdSdS)NcSsg|]}t |¡‘qSr)Ú _python_cuÚ find_function)Ú.0Ú qual_namerrrÚ dóz1_try_get_jit_cached_overloads..)r3r)ÚkeyZ qual_namesrrrÚ_try_get_jit_cached_overloadsas r;cCsdd„|Dƒt|<dS)NcSsg|] }|j‘qSrr/)r6Úfnrrrr8ir9z+_set_jit_overload_cache..)r3)r:Ú compiled_fnsrrrÚ_set_jit_overload_cachehsr>cCs6t|ddƒdurdSt |d¡}|r.t |¡SdSdS)NZ __disable_jit_function_caching__FT)r+r2rr4r5)r:r7rrrÚ_try_get_jit_cached_functionks   r?cCs t|tjjƒsJ‚|jt|<dSr)Ú isinstanceÚtorchÚjitÚScriptFunctionr&r2)r:rrrrÚ_set_jit_function_cachetsrD)rrArÚweakrefrr!r"r#Ú_CÚCompilationUnitr4r$r%r)r.r,r1ÚWeakKeyDictionaryr2Ú__annotations__r3r;r>r?rDrrrrÚs(