?iUdZddlZddlZddlmZddlZGddZeZddZddZ ej Z iZ eeeeefed <iZeeeefed <dd Zd Zd ZddZejZejed<ejZejed<dZddZdZddZdS)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,eZdZdZddZdZdefdZdS) 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 C/var/www/icac/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 |rrN)__name__ __module__ __qualname____doc__rr boolr&rr rrs_     JJJ $rrrcdt_dS)NF_enabledr r-rr disabler13sHrcdt_dS)NTr/r-rr enabler37sHr_script_classes_name_to_pyclasscR|t|<|t|<dSr%)r4r5qualified_name) python_class script_classs r _add_script_classr:Fs)$0OL!6B\0022333rczt|dd}|t|}t|S)N_jit_override_qualname)getattr_get_python_classr4r)r8overrides r _get_script_classr@Ks;|%=tDDH(22   | , ,,rc6t|Sr%)r5rr7s r r>r>Rs    / //rcjttdSr%)r4clearr5r-rr _clear_class_staterEVs,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)rGr)key qual_namess r _try_get_jit_cached_overloadsrQgs8/33C>>JPPZPPPPtrc.d|Dt|<dS)Ncg|] }|j Sr-rB)rLfns r rNz+_set_jit_overload_cache..ps*T*T*T2+<*T*T*Tr)rG)rO compiled_fnss r _set_jit_overload_cacherVos!*T*T|*T*T*T"3'''rct|dddurdSt|d}|rt|SdS)N __disable_jit_function_caching__FT)r=rFrrJrK)rOrMs r _try_get_jit_cached_functionrYssUs6>>$FFt"&&sD11I'' 222trcft|tjjsJ|jt |<dSr%) isinstancetorchjitScriptFunctionr7rF)rOr#s r _set_jit_function_cacher_}s1 eUY5 6 6666#2srr')r+rweakreftypingrr\rr0r1r3_CCompilationUnitrJr4dicttype__annotations__r5strr:r@r>rEWeakKeyDictionaryrFrGrQrVrYr_r-rr ris  > <>>X % % ' ' /1d3ic*+000)+$sDI~&+++CCCC ---0001J0I0K0KG-KKK