P/jUdZddlZddlZddlmZmZmZddlZGddZeZ dZ dZ ejjZiZeeeeefed<iZeeeefed <d Zd Zd Zd Zej.Zej.ed<ej.Zej.ed<dZdZdZdZy)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$eZdZdZddZdZdZy) EnabledProxyzyStores whether the JIT is enabled or not. This is just a wrapper for a bool, so that we get reference semantics Nc6|jdddd|_y)N PYTORCH_JITTz> Using PyTorch JITz> PyTorch JIT DISABLED) parse_envenabledselfs S/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/jit/_state.py__init__zEnabledProxy.__init__s~~ 4!68P  ctjj|}||S|jdvry|jdvry|dk(r t |y|dk(r t |yt d|d) N>1yestrueT>0nofalseF1v0vzUnknown setting of z. Try using 0 or 1.)osenvirongetlowerprint ValueError)r namedefault true_message false_messagevalues rr zEnabledProxy.parse_envs| t$ =N ;;=0 0 [[]2 2 D= ,  d] - .tf4GHIIrc|jSN)r r s r__bool__zEnabledProxy.__bool__+s ||r)returnN)__name__ __module__ __qualname____doc__rr r'rrrrs  J rrcdt_y)NF_enabledr r-rrdisabler12s Hrcdt_y)NTr/r-rrenabler36s Hr_script_classes_name_to_pyclasscD|t|<|t|j<yr&)r4r5qualified_name) python_class script_classs r_add_script_classr:Es $0OL!6B\0023rcbt|dd}| t|}tj|dS)N_jit_override_qualname)getattr_get_python_classr4r)r8overrides r_get_script_classr@Js4|%=tDH(2   |T 22rc.tj|dSr&)r5r)r7s rr>r>Qs    55rcTtjtjyr&)r4clearr5r-rr_clear_class_staterDUsr_jit_caching_layer_jit_function_overload_cachingctj|d}|r#|Dcgc]}tj|c}Sycc}wr&)rFr _python_cu find_function)key qual_names qual_names r_try_get_jit_cached_overloadsrMfs</33C>JEOP ((3PPQs=cP|Dcgc]}|jc}t|<ycc}wr&)r7rF)rJ compiled_fnsfns r_set_jit_overload_cacherQns!GS*T2+<+<*T"3'*Ts#c~t|dddurytj|d}|rtj |Sy)N __disable_jit_function_caching__FT)r=rErrHrI)rJrLs r_try_get_jit_cached_functionrTrs@s6>$F"&&sD1I'' 22rcvt|tjjsJ|jt |<yr&) isinstancetorchjitScriptFunctionr7rE)rJr$s r_set_jit_function_cacherZ|s- eUYY55 66 6#22sr)r,rweakreftypingrrrrWrr0r1r3_CCompilationUnitrHr4__annotations__r5strr:r@r>rDWeakKeyDictionaryrErFrMrQrTrZr-rrrbs "" > >XX % % ' /1d3ic*+0)+$sDI~&+C 361J0I0I0KG--K