o 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*eZdZdZd ddZddZdd 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)N PYTORCH_JITTz> Using PyTorch JITz> PyTorch JIT DISABLED) parse_envenabledselfr d/var/www/html/construction_image-detection-poc/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)r namedefault true_message false_messagevaluer r r rs   zEnabledProxy.parse_envcCs|jSN)rrr r r __bool__,szEnabledProxy.__bool__)rN)__name__ __module__ __qualname____doc__r rr!r 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_class script_classr r r _add_script_classFsr1cCs(t|dd}|durt|}t|dS)N_jit_override_qualname)getattr_get_python_classr,r)r/overrider r r _get_script_classKs  r6cCs t|dSr )r-rr.r r r r4Rs r4cCsttdSr )r,clearr-r r r r _clear_class_stateVs r9_jit_caching_layer_jit_function_overload_cachingcCs"t|d}|rdd|DSdS)NcSsg|]}t|qSr ) _python_cu find_function).0 qual_namer r r jsz1_try_get_jit_cached_overloads..)r;r)key qual_namesr r r _try_get_jit_cached_overloadsgs rCcCsdd|Dt|<dS)NcSsg|]}|jqSr r7)r>fnr r r r@psz+_set_jit_overload_cache..)r;)rA compiled_fnsr r r _set_jit_overload_cacheosrFcCs2t|dddur dSt|d}|rt|SdS)N __disable_jit_function_caching__FT)r3r:rr<r=)rAr?r r r _try_get_jit_cached_functionss   rHcCs t|tjjs J|jt|<dSr ) isinstancetorchjitScriptFunctionr.r:)rArr r r _set_jit_function_cache}srM)r%rweakreftypingrrJrr(r)r+_CCompilationUnitr<r,dicttype__annotations__r-strr1r6r4r9WeakKeyDictionaryr:r;rCrFrHrMr r r r s,