U ʹjA@sUdZddlZddlZddlZGdddZeZddZddZej Z iZ iZ d d Z d d Zd dZddZeZejed<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`. Nc@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_envenabledselfrO/var/www/html/bwave/python/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>1yestrueT>0falsenoFZ1vZ0vzUnknown setting of z. Try using 0 or 1.)osenvirongetlowerprint ValueError)rnamedefaultZ true_messageZ false_messagevaluerrrrs   zEnabledProxy.parse_envcCs|jSN)rrrrr__bool__)szEnabledProxy.__bool__N)__name__ __module__ __qualname____doc__r rrrrrrrsrcCs dt_dS)NF_enabledrrrrrdisable0sr!cCs dt_dS)NTrrrrrenable4sr"cCs|t|<|t|<dSr)_script_classes_name_to_pyclassqualified_name) python_classZ script_classrrr_add_script_classCsr'cCs(t|dd}|dk rt|}t|dS)NZ_jit_override_qualname)getattr_get_python_classr#r)r&overriderrr_get_script_classHs r+cCs t|dSr)r$rr%rrrr)Osr)cCsttdSr)r#clearr$rrrr_clear_class_stateSsr._jit_caching_layer_jit_function_overload_cachingcCs&t|d}|rdd|DSdSdS)NcSsg|]}t|qSr) _python_cu find_function).0 qual_namerrr gsz1_try_get_jit_cached_overloads..)r0r)keyZ qual_namesrrr_try_get_jit_cached_overloadsds r7cCsdd|Dt|<dS)NcSsg|] }|jqSrr,)r3fnrrrr5msz+_set_jit_overload_cache..)r0)r6Z compiled_fnsrrr_set_jit_overload_cachelsr9cCs6t|dddkrdSt|d}|r.t|SdSdS)NZ __disable_jit_function_caching__FT)r(r/rr1r2)r6r4rrr_try_get_jit_cached_functionps   r:cCs t|tjjst|jt|<dSr) isinstancetorchZjitZScriptFunctionAssertionErrorr%r/)r6rrrr_set_jit_function_cachezsr>)rrweakrefr<rr r!r"Z_CZCompilationUnitr1r#r$r'r+r)r.WeakKeyDictionaryr/__annotations__r0r7r9r:r>rrrrs(