`iddlZddlZddlmZddlmZddlZejZde fdZ ej deeddffdZ dej jddfdZdej jde fd Zd ejddfd Zd ejde fd ZdS) N) Generator)Anyreturnc.ttddS)Nfreezing_activeF)getattr_TLSr/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/_inductor/freezing_utils.py_freezing_activer s 4*E 2 22r c#Kt}dt_ dV|t_dS#|t_wxYw)z? Context manager to designate when freezing is active. TN)r r r)prevs r enter_freezingrsK   DD$ #t####s0>gmcd|_dS)z. Mark the gm as having frozen params. TN)_has_frozen_paramsrs r record_has_frozen_paramsrs!Br c$t|ddS)z6 Return True if the gm has frozen parameters. rFrrs r has_frozen_paramsr$s 2+U 3 33r tc4tr d|_dSdS)zK Mark the provided tensor as a frozen param if freezing is active. TN)r _is_frozen_paramrs r maybe_set_is_frozen_paramr+s*"!""r c$t|ddS)z6 Return True if the tensor is a frozen param. rFrrs r is_frozen_paramr3s 1(% 0 00r ) contextlib threadingcollections.abcrtypingrtorchlocalr boolr contextmanagerrfx GraphModulerrTensorrrr r r r+sL%%%%%% y3$3333  $ #tT/2 $ $ $ $!!5!$!!!!4%(.444444""$""""1u|1111111r