o hA@sUdZddlZddlZddlZGdddZeZddZddZej Z iZ iZ d d Z d d Zd dZddZeZejed<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`. Nc@s(eZdZdZddZddZddZdS) 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)N PYTORCH_JITTz> Using PyTorch JITz> PyTorch JIT DISABLED) parse_envenabledselfrF/var/www/html/ai/venv/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>1yestrueT>0nofalseF1v0vzUnknown setting of z. Try using 0 or 1.)osenvirongetlowerprint ValueError)rnamedefault true_message false_messagevaluerrr rs   zEnabledProxy.parse_envcCs|jSN)rrrrr __bool__)szEnabledProxy.__bool__N)__name__ __module__ __qualname____doc__r rrrrrr rs  rcC dt_dS)NF_enabledrrrrr disable0 r'cCr$)NTr%rrrr enable4r(r)cCs|t|<|t|<dSr)_script_classes_name_to_pyclassqualified_name) python_class script_classrrr _add_script_classCsr/cCs(t|dd}|durt|}t|dS)N_jit_override_qualname)getattr_get_python_classr*r)r-overriderrr _get_script_classHs  r4cCs t|dSr)r+rr,rrr r2Os r2cCsttdSr)r*clearr+rrrr _clear_class_stateSs r7_jit_caching_layer_jit_function_overload_cachingcCs"t|d}|rdd|DSdS)NcSsg|]}t|qSr) _python_cu find_function).0 qual_namerrr gsz1_try_get_jit_cached_overloads..)r9r)key qual_namesrrr _try_get_jit_cached_overloadsds rAcCsdd|Dt|<dS)NcSsg|]}|jqSrr5)r<fnrrr r>msz+_set_jit_overload_cache..)r9)r? compiled_fnsrrr _set_jit_overload_cachelsrDcCs2t|dddur dSt|d}|rt|SdS)N __disable_jit_function_caching__FT)r1r8rr:r;)r?r=rrr _try_get_jit_cached_functionps   rFcCs t|tjjs J|jt|<dSr) isinstancetorchjitScriptFunctionr,r8)r?rrrr _set_jit_function_cachezsrK)r#rweakrefrHrr&r'r)_CCompilationUnitr:r*r+r/r4r2r7WeakKeyDictionaryr8__annotations__r9rArDrFrKrrrr s*