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(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)N PYTORCH_JITTz> Using PyTorch JITz> PyTorch JIT DISABLED) parse_envenabledselfr N/home/ubuntu/SoloSpeech/.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)rr r r r __bool__+szEnabledProxy.__bool__N)__name__ __module__ __qualname____doc__r rr"r r r r rs  rcC dt_dS)NF_enabledrr 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_classEsr2cCs(t|dd}|durt|}t|dS)N_jit_override_qualname)getattr_get_python_classr-r)r0overrider r r _get_script_classJs  r7cCs t|dSr!)r.rr/r r r r5Qs r5cCsttdSr!)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 rDcCsdd|Dt|<dS)NcSsg|]}|jqSr r8)r?fnr r r rAosz+_set_jit_overload_cache..)r<)rB compiled_fnsr r r _set_jit_overload_cachensrGcCs2t|dddur dSt|d}|rt|SdS)N __disable_jit_function_caching__FT)r4r;rr=r>)rBr@r r r _try_get_jit_cached_functionrs   rIcCs t|tjjs J|jt|<dSr!) isinstancetorchjitScriptFunctionr/r;)rBr r r r _set_jit_function_cache|srN)r&rweakreftypingrrrrKrr)r*r,_CCompilationUnitr=r-__annotations__r.strr2r7r5r:WeakKeyDictionaryr;r<rDrGrIrNr r r r s,