o Zh@sddlZddlZddlmZddlmZddlZeZde fddZ ej deeddffddZ d ej jddfd d Zd ej jde fd d ZdejddfddZdejde fddZdS)N) Generator)AnyreturncCs ttddS)Nfreezing_activeF)getattr_TLSrrM/var/www/auris/lib/python3.10/site-packages/torch/_inductor/freezing_utils.py_freezing_active s r ccs*t}dt_z dVW|t_dS|t_w)z? Context manager to designate when freezing is active. TN)r rr)prevrrr enter_freezings r gmcCs d|_dS)z. Mark the gm as having frozen params. TN)_has_frozen_paramsr rrr record_has_frozen_paramss rcC t|ddS)z6 Return True if the gm has frozen parameters. rFrrrrr has_frozen_params$ rtcCstrd|_dSdS)zK Mark the provided tensor as a frozen param if freezing is active. TN)r _is_frozen_paramrrrr maybe_set_is_frozen_param+s rcCr)z6 Return True if the tensor is a frozen param. rFrrrrr is_frozen_param3rr) contextlib threadingcollections.abcrtypingrZtorchlocalrboolr contextmanagerr ZfxZ GraphModulerrZTensorrrrrrr s