o …ZŽh¯ã@sôUdZddlZddlZddlmZddlZGdd„dƒZeƒZdd„Zdd „Z ej ¡ZiZ eeeeefed <iZeeeefed<dd „Zdd„Zdd„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`. éN)ÚAnyc@s*eZdZdZd dd„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 ÚreturnNcCs| dddd¡|_dS)NZPYTORCH_JITTz> Using PyTorch JITz> PyTorch JIT DISABLED)Ú parse_envÚenabled©Úself©r ú?/var/www/auris/lib/python3.10/site-packages/torch/jit/_state.pyÚ__init__s ÿzEnabledProxy.__init__cCsptj |¡}|dur|S| ¡dvrdS| ¡dvrdS|dkr&t|ƒdS|dkr0t|ƒdStd|›dƒ‚) N>Ú1ÚyesÚtrueT>Ú0ÚnoÚfalseFZ1vZ0vzUnknown setting of z. Try using 0 or 1.)ÚosÚenvironÚgetÚlowerÚprintÚ ValueError)rÚnameÚdefaultZtrue_messageZ false_messageÚvaluer r r rszEnabledProxy.parse_envcCs|jS©N)rrr r r Ú__bool__,szEnabledProxy.__bool__)rN)Ú__name__Ú __module__Ú__qualname__Ú__doc__rrrr r r r rs rcCó dt_dS)NF©Ú_enabledrr r r r Údisable3ó r$cCr!)NTr"r r r r Úenable7r%r&Ú_script_classesÚ_name_to_pyclasscCs|t|<|t| ¡<dSr)r'r(Úqualified_name)Úpython_classZscript_classr r r Ú_add_script_classFsr+cCs(t|ddƒ}|durt|ƒ}t |d¡S)NZ_jit_override_qualname)ÚgetattrÚ_get_python_classr'r)r*Úoverrider r r Ú_get_script_classKsr/cCst |d¡Sr)r(r©r)r r r r-Rsr-cCst ¡t ¡dSr)r'Úclearr(r r r r Ú_clear_class_stateVsr2Ú_jit_caching_layerÚ_jit_function_overload_cachingcCs"t |d¡}|rdd„|DƒSdS)NcSsg|]}t |¡‘qSr )Ú _python_cuÚ find_function)Ú.0Ú qual_namer r r Ú jsz1_try_get_jit_cached_overloads..)r4r)ÚkeyZ qual_namesr r r Ú_try_get_jit_cached_overloadsgsr;cCsdd„|Dƒt|<dS)NcSsg|]}|j‘qSr r0)r7Úfnr r r r9psz+_set_jit_overload_cache..)r4)r:Zcompiled_fnsr r r Ú_set_jit_overload_cacheosr=cCs2t|ddƒdur dSt |d¡}|rt |¡SdS)NZ __disable_jit_function_caching__FT)r,r3rr5r6)r:r8r r r Ú_try_get_jit_cached_functionss r>cCs t|tjjƒs J‚|jt|<dSr)Ú isinstanceÚtorchZjitZScriptFunctionr)r3)r:rr r r Ú_set_jit_function_cache}srA)r rÚweakrefÚtypingrr@rr#r$r&Z_CZCompilationUnitr5r'ÚdictÚtypeÚ__annotations__r(Ústrr+r/r-r2ÚWeakKeyDictionaryr3r4r;r=r>rAr r r r Ús,