
    sh                     \   S SK r S SKrS SKJr  S SKJr  S SKr\R                  " 5       rS\	4S jr
\ R                  S\\SS4   4S j5       rS\R                  R                  SS4S jrS\R                  R                  S\	4S	 jrS
\R"                  SS4S jrS
\R"                  S\	4S jrg)    N)	Generator)Anyreturnc                  $    [        [        SS5      $ )Nfreezing_activeF)getattr_TLS     r/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/_inductor/freezing_utils.py_freezing_activer      s    4*E22r   c               #   v   #    [        5       n S[        l         Sv   U [        l        g! U [        l        f = f7f)z7
Context manager to designate when freezing is active.
TN)r   r	   r   )prevs    r   enter_freezingr      s/     
 DD$#ts   9) 969gmc                     SU l         g)z&
Mark the gm as having frozen params.
TN)_has_frozen_paramsr   s    r   record_has_frozen_paramsr      s     !Br   c                     [        U SS5      $ )z.
Return True if the gm has frozen parameters.
r   Fr   r   s    r   has_frozen_paramsr   $   s     2+U33r   tc                 2    [        5       (       a  SU l        gg)zC
Mark the provided tensor as a frozen param if freezing is active.
TN)r   _is_frozen_paramr   s    r   maybe_set_is_frozen_paramr   +   s     ! r   c                     [        U SS5      $ )z.
Return True if the tensor is a frozen param.
r   Fr   r   s    r   is_frozen_paramr   3   s     1(%00r   )
contextlib	threadingcollections.abcr   typingr   torchlocalr	   boolr   contextmanagerr   fxGraphModuler   r   Tensorr   r   r
   r   r   <module>r+      s      %   3$ 3 	$	#tT/2 	$ 	$!!5!5 !$ !4%((.. 44 4" "$ "1u|| 1 1r   