VhPddlZddlZddlmZddlmZddlZejZde fdZ ejdeeddffdZ dejjddfdZdejjde fd Zd ej"ddfd Zd ej"de fd Zy) N) Generator)Anyreturnc$ttddS)Nfreezing_activeF)getattr_TLSN/home/dcms/DCMS/lib/python3.12/site-packages/torch/_inductor/freezing_utils.py_freezing_activer s 4*E 22r c#vKt}dt_ d|t_y#|t_wxYww)z? Context manager to designate when freezing is active. TN)r r r)prevs r enter_freezingrs0  DD$ #ts9) 9 69gmcd|_y)z. Mark the gm as having frozen params. TN)_has_frozen_paramsrs r record_has_frozen_paramsrs !Br ct|ddS)z6 Return True if the gm has frozen parameters. rFrrs r has_frozen_paramsr$s 2+U 33r tc(trd|_yy)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 ct|ddS)z6 Return True if the tensor is a frozen param. rFrrs r is_frozen_paramr3s 1(% 00r ) contextlib threadingcollections.abcrtypingrtorchlocalr boolr contextmanagerrfx GraphModulerrTensorrrr r r r+s% y3$3  $ #tT/2 $ $!!5!5!$!4%((..444""$"1u||11r