
    Bj                     T    d dl Z d dlmZ d dlZ G d d          Zddedz  defdZdS )	    N)Anyc                   \    e Zd ZU i Zeeef         ed<    ej	                    Z
deddfdZdS )DeviceContext	_CONTEXTSdevicereturnNc                     || _         d S N)r   )selfr   s     ]/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/torch/_lazy/device_context.py__init__zDeviceContext.__init__   s        )__name__
__module____qualname__r   dictstrr   __annotations__	threadingLock_CONTEXTS_LOCKr    r   r   r   r      s_          "ItCH~"""#Y^%%Ns t      r   r   r   r   c                 <   | $t           j        j                                        } nt	          |           } t
          j        5  t
          j                            | d           }|t          |           }|t
          j        | <   |cd d d            S # 1 swxY w Y   d S r
   )	torch_C_lazy_get_default_device_typer   r   r   r   get)r   devctxs     r   get_device_contextr       s    ~88::V		%  (,,VT::>"6**F.4M#F+                 s   ABBBr
   )r   typingr   torch._C._lazyr   r   r   r    r   r   r   <module>r#      s                        
 
sTz 
] 
 
 
 
 
 
r   