
    קg                     H    d dl Z ddlmZ  G d d          Z e            ZdS )    N   )ExactWeakKeyDictionaryc                   ^    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	dS )
CodeContextDictreturnNc                 ,    t                      | _        d S N)r   code_contextselfs    V/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/_dynamo/code_context.py__init__zCodeContextDict.__init__   s    244    codec                     || j         v S r	   )r
   )r   r   s     r   has_contextzCodeContextDict.has_context   s    t(((r   c                 V    | j                             |          }|i }|| j         |<   |S r	   )r
   getr   r   ctxs      r   get_contextzCodeContextDict.get_context   s4    ##D));C&)Dd#
r   c                 ~    |                      |          }| j                            t          |                     |S r	   )r   r
   
_remove_ididr   s      r   pop_contextzCodeContextDict.pop_context   s7    t$$$$RXX...
r   c                 8    | j                                          d S r	   )r
   clearr   s    r   r   zCodeContextDict.clear   s    !!!!!r   )r   N)
__name__
__module____qualname__r   typesCodeTyper   r   r   r    r   r   r   r      s        5 5 5 5) ) ) ) )        
" " " " "r   r   )r!   utilsr   r   r
   r#   r   r   <module>r%      s[     ) ) ) ) ) )" " " " " " " ".   r   