hUdZddlZddlZddlmZddlZGddZeZdZdZ ej Z iZ eeeeefed<iZeeeefed <d Zd Zd Zd ZejZejed<ejZejed<dZdZdZdZdS)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&eZdZdZddZdZdZdS) EnabledProxyzyStores whether the JIT is enabled or not. This is just a wrapper for a bool, so that we get reference semantics returnNc@|dddd|_dS)N PYTORCH_JITTz> Using PyTorch JITz> PyTorch JIT DISABLED) parse_envenabledselfs X/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/torch/jit/_state.py__init__zEnabledProxy.__init__s&~~ 4!68P   c*tj|}||S|dvrdS|dvrdS|dkrt |dS|dkrt |dSt d|d) N>1yestrueT>0nofalseF1v0vzUnknown setting of z. Try using 0 or 1.)osenvirongetlowerprint ValueError)r namedefault true_message false_messagevalues r r zEnabledProxy.parse_envs t$$ =N ;;==0 0 04 [[]]2 2 25 D== ,   4 d]] - 5HtHHHIIIrc|jSN)r r s r __bool__zEnabledProxy.__bool__,s |r)rN)__name__ __module__ __qualname____doc__rr r&rr rrsS     JJJ rrcdt_dS)NF_enabledr r+rr disabler/3sHrcdt_dS)NTr-r+rr enabler17sHr_script_classes_name_to_pyclasscR|t|<|t|<dSr%)r2r3qualified_name) python_class script_classs r _add_script_classr8Fs)$0OL!6B\0022333rc|t|dd}|t|}t|dS)N_jit_override_qualname)getattr_get_python_classr2r)r6overrides r _get_script_classr>Ks=|%=tDDH(22   |T 2 22rc8t|dSr%)r3rr5s r r<r<Rs    5 55rcjttdSr%)r2clearr3r+rr _clear_class_staterCVs,r_jit_caching_layer_jit_function_overload_cachingcXt|d}|r d|DSdS)NcBg|]}t|Sr+) _python_cu find_function).0 qual_names r z1_try_get_jit_cached_overloads..js&PPP ((33PPPr)rEr)key qual_namess r _try_get_jit_cached_overloadsrOgs8/33C>>JPPZPPPPtrc.d|Dt|<dS)Ncg|] }|j Sr+r@)rJfns r rLz+_set_jit_overload_cache..ps*T*T*T2+<*T*T*Tr)rE)rM compiled_fnss r _set_jit_overload_cacherTos!*T*T|*T*T*T"3'''rct|dddurdSt|d}|rt|SdS)N __disable_jit_function_caching__FT)r;rDrrHrI)rMrKs r _try_get_jit_cached_functionrWssUs6>>$FFt"&&sD11I'' 222trcft|tjjsJ|jt |<dSr%) isinstancetorchjitScriptFunctionr5rD)rMr#s r _set_jit_function_cacher]}s1 eUY5 6 6666#2sr)r*rweakreftypingrrZrr.r/r1_CCompilationUnitrHr2dicttype__annotations__r3strr8r>r<rCWeakKeyDictionaryrDrErOrTrWr]r+rr rgs  > <>>X % % ' ' /1d3ic*+000)+$sDI~&+++CCC 3336661J0I0K0KG-KKK