
    ɯwg                     :    d dl Z ddlmZ  G d d      Z e       Zy)    N   )ExactWeakKeyDictionaryc                   z    e Zd ZddZdej
                  fdZdej
                  fdZdej
                  fdZd Z	y)	CodeContextDictNc                 "    t               | _        y N)r   code_contextselfs    _/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/torch/_dynamo/code_context.py__init__zCodeContextDict.__init__   s    24    codec                     || j                   v S r   )r	   )r   r   s     r   has_contextzCodeContextDict.has_context   s    t((((r   c                 b    | j                   j                  |      }|i }|| j                   |<   |S r   )r	   getr   r   ctxs      r   get_contextzCodeContextDict.get_context   s7    ##D);C&)Dd#
r   c                 p    | j                  |      }| j                  j                  t        |             |S r   )r   r	   
_remove_ididr   s      r   pop_contextzCodeContextDict.pop_context   s/    t$$$RX.
r   c                 8    | j                   j                          y r   )r	   clearr
   s    r   r   zCodeContextDict.clear   s    !r   )returnN)
__name__
__module____qualname__r   typesCodeTyper   r   r   r    r   r   r   r      s<    5) )  
"r   r   )r!   utilsr   r   r	   r#   r   r   <module>r%      s     )" ".  r   