o i@sUdZddlZddlZddlmZddlZGdddZeZdddZdd d Z ej Z iZ eeeeefed <iZeeeefed <dd dZddZddZdddZeZejed<eZejed<ddZdddZddZdddZdS) 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@s0eZdZdZd ddZddZdefdd 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 R/sda-disk/www/egybert/egybert_env/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 rboolr!r r r r rs  rrcC 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_classFsr3cCs&t|dd}|durt|}t|S)N_jit_override_qualname)getattr_get_python_classr.r)r1overrider r r _get_script_classKs  r8cCs t|Sr )r/rr0r r r r6Rr,r6cCsttdSr )r.clearr/r r r r _clear_class_stateVs r;_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 rEcCsdd|Dt|<dS)NcSsg|]}|jqSr r9)r@fnr r r rBpsz+_set_jit_overload_cache..)r=)rC compiled_fnsr r r _set_jit_overload_cacheosrHcCs2t|dddur dSt|d}|rt|SdS)N __disable_jit_function_caching__FT)r5r<rr>r?)rCrAr r r _try_get_jit_cached_functionss   rJcCs t|tjjs J|jt|<dSr ) isinstancetorchjitScriptFunctionr0r<)rCrr r r _set_jit_function_cache}srOr")r&rweakreftypingrrLrr*r+r-_CCompilationUnitr>r.dicttype__annotations__r/strr3r8r6r;WeakKeyDictionaryr<r=rErHrJrOr r r r s,