o i@sUdZddlZddlZddlmZmZmZddlZGdddZeZ ddZ dd Z ej ZiZeeeeefed <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)AnyDictTypec@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 N/home/ubuntu/transcripts/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)r r r r r __bool__+szEnabledProxy.__bool__)rN)__name__ __module__ __qualname____doc__rrr#r r r r rs   rcC dt_dS)NF_enabledr r r r r disable2 r+cCr()NTr)r r r r enable6r,r-_script_classes_name_to_pyclasscCs|t|<|t|<dSr")r.r/qualified_name) python_class script_classr r r _add_script_classEsr3cCs(t|dd}|durt|}t|dS)N_jit_override_qualname)getattr_get_python_classr.r)r1overrider r r _get_script_classJs  r8cCs t|dSr")r/rr0r r r r6Qs r6cCsttdSr")r.clearr/r r r r _clear_class_stateUs r;_jit_caching_layer_jit_function_overload_cachingcCs"t|d}|rdd|DSdS)NcSsg|]}t|qSr ) _python_cu find_function).0 qual_namer r r isz1_try_get_jit_cached_overloads..)r=r)key qual_namesr r r _try_get_jit_cached_overloadsfs rEcCsdd|Dt|<dS)NcSsg|]}|jqSr r9)r@fnr r r rBosz+_set_jit_overload_cache..)r=)rC compiled_fnsr r r _set_jit_overload_cachensrHcCs2t|dddur dSt|d}|rt|SdS)N __disable_jit_function_caching__FT)r5r<rr>r?)rCrAr r r _try_get_jit_cached_functionrs   rJcCs t|tjjs J|jt|<dSr") isinstancetorchjitScriptFunctionr0r<)rCr!r r r _set_jit_function_cache|srO)r'rweakreftypingrrrrLrr*r+r-_CCompilationUnitr>r.__annotations__r/strr3r8r6r;WeakKeyDictionaryr<r=rErHrJrOr r r r s,