a h@sUdZddlZddlZddlmZddlZGdddZeZddZdd Z ej Z iZ eeeeefed <iZeeeefed <d d ZddZddZddZeZejed<eZejed<ddZddZddZddZdS)zJIT-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.eZdZdZddddZddZdd ZdS) EnabledProxyzyStores whether the JIT is enabled or not. This is just a wrapper for a bool, so that we get reference semantics N)returncCs|dddd|_dS)NZ PYTORCH_JITTz> Using PyTorch JITz> PyTorch JIT DISABLED) parse_envenabledselfr >/var/www/auris/lib/python3.9/site-packages/torch/jit/_state.py__init__szEnabledProxy.__init__cCsttj|}|dur|S|dvr(dS|dvr8dS|dkrLt|dS|dkr`t|dStd|ddS) N>trueyes1T>nofalse0FZ1vZ0vzUnknown setting of z. Try using 0 or 1.)osenvirongetlowerprint ValueError)rnamedefaultZ true_messageZ false_messagevaluer r r rs   zEnabledProxy.parse_envcCs|jSN)rrr r r __bool__,szEnabledProxy.__bool__)__name__ __module__ __qualname____doc__r rrr r r r rsrcCs dt_dS)NF_enabledrr r r r disable3sr#cCs dt_dS)NTr!r r r r enable7sr$_script_classes_name_to_pyclasscCs|t|<|t|<dSr)r%r&qualified_name) python_classZ script_classr r r _add_script_classFsr)cCs(t|dd}|durt|}t|dS)NZ_jit_override_qualname)getattr_get_python_classr%r)r(overrider r r _get_script_classKs r-cCs t|dSr)r&rr'r r r r+Rsr+cCsttdSr)r%clearr&r r r r _clear_class_stateVsr0_jit_caching_layer_jit_function_overload_cachingcCs&t|d}|rdd|DSdSdS)NcSsg|]}t|qSr ) _python_cu find_function).0 qual_namer r r jz1_try_get_jit_cached_overloads..)r2r)keyZ qual_namesr r r _try_get_jit_cached_overloadsgs r:cCsdd|Dt|<dS)NcSsg|] }|jqSr r.)r5fnr r r r7pr8z+_set_jit_overload_cache..)r2)r9Z compiled_fnsr r r _set_jit_overload_cacheosr<cCs6t|dddurdSt|d}|r.t|SdSdS)NZ __disable_jit_function_caching__FT)r*r1rr3r4)r9r6r r r _try_get_jit_cached_functionss   r=cCs t|tjjsJ|jt|<dSr) isinstancetorchZjitZScriptFunctionr'r1)r9rr r r _set_jit_function_cache}sr@)r rweakreftypingrr?rr"r#r$Z_CZCompilationUnitr3r%dicttype__annotations__r&strr)r-r+r0WeakKeyDictionaryr1r2r:r<r=r@r r r r s*