U ‰dCã@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ú4/tmp/pip-unpacked-wheel-ua33x9lu/torch/jit/_state.pyÚ__init__s ÿzEnabledProxy.__init__cCsrtj |¡}|dkr|S| ¡dkr(dS| ¡dkr8dS|dkrLt|ƒdS|dkr`t|ƒdStd |¡ƒ‚dS)N>ÚyesÚ1ÚtrueT>ÚfalseÚ0Ú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_classZ script_classrrrÚ_add_script_classAsr(cCs(t|ddƒ}|dk rt|ƒ}t |d¡S)NZ_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_stateQsr/Ú_jit_caching_layerÚ_jit_function_overload_cachingcCs&t |d¡}|rdd„|DƒSdSdS)NcSsg|]}t |¡‘qSr)Ú _python_cuÚ find_function)Ú.0Ú qual_namerrrÚ dsz1_try_get_jit_cached_overloads..)r1r)ÚkeyZ qual_namesrrrÚ_try_get_jit_cached_overloadsas r8cCsdd„|Dƒt|<dS)NcSsg|] }|j‘qSrr-)r4Úfnrrrr6isz+_set_jit_overload_cache..)r1)r7Z compiled_fnsrrrÚ_set_jit_overload_cachehsr:cCs6t|ddƒdkrdSt |d¡}|r.t |¡SdSdS)NZ __disable_jit_function_caching__FT)r)r0rr2r3)r7r5rrrÚ_try_get_jit_cached_functionks   r;cCs t|tjjƒst‚|jt|<dSr)Ú isinstanceÚtorchZjitZScriptFunctionÚAssertionErrorr&r0)r7rrrrÚ_set_jit_function_cachetsr?)rr=rÚweakrefrr!r"r#Z_CZCompilationUnitr2r$r%r(r,r*r/ÚWeakKeyDictionaryr0Ú__annotations__r1r8r:r;r?rrrrÚs(