U wh@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(eZdZdZddZddZddZdS) EnabledProxyzyStores whether the JIT is enabled or not. This is just a wrapper for a bool, so that we get reference semantics cCs|dddd|_dS)NZ PYTORCH_JITTz> Using PyTorch JITz> PyTorch JIT DISABLED) parse_envenabledselfr B/var/www/html/venv/lib/python3.8/site-packages/torch/jit/_state.py__init__s zEnabledProxy.__init__cCsttj|}|dkr|S|dkr(dS|dkr8dS|dkrLt|dS|dkr`t|dStd|ddS) N>yes1trueT>0falsenoFZ1vZ0vzUnknown setting of z. Try using 0 or 1.)osenvirongetlowerprint ValueError)r namedefaultZ true_messageZ false_messagevaluer r r rs   zEnabledProxy.parse_envcCs|jSN)rrr r r __bool__+szEnabledProxy.__bool__N)__name__ __module__ __qualname____doc__r rrr r r r rsrcCs dt_dS)NF_enabledrr r r r disable2sr$cCs dt_dS)NTr"r r r r enable6sr%_script_classes_name_to_pyclasscCs|t|<|t|<dSr)r&r'qualified_name) python_classZ script_classr r r _add_script_classEsr*cCs(t|dd}|dk rt|}t|dS)NZ_jit_override_qualname)getattr_get_python_classr&r)r)overrider r r _get_script_classJs r.cCs t|dSr)r'rr(r r r r,Qsr,cCsttdSr)r&clearr'r r r r _clear_class_stateUsr1_jit_caching_layer_jit_function_overload_cachingcCs&t|d}|rdd|DSdSdS)NcSsg|]}t|qSr ) _python_cu find_function).0 qual_namer r r isz1_try_get_jit_cached_overloads..)r3r)keyZ qual_namesr r r _try_get_jit_cached_overloadsfs r:cCsdd|Dt|<dS)NcSsg|] }|jqSr r/)r6fnr r r r8osz+_set_jit_overload_cache..)r3)r9Z compiled_fnsr r r _set_jit_overload_cachensr<cCs6t|dddkrdSt|d}|r.t|SdSdS)NZ __disable_jit_function_caching__FT)r+r2rr4r5)r9r7r r r _try_get_jit_cached_functionrs   r=cCs t|tjjst|jt|<dSr) isinstancetorchZjitZScriptFunctionAssertionErrorr(r2)r9rr r r _set_jit_function_cache|srA)r!rweakreftypingrrrr?rr#r$r%Z_CZCompilationUnitr4r&__annotations__r'strr*r.r,r1WeakKeyDictionaryr2r3r:r<r=rAr r r r s*