
    קg                     H    d dl Z d dlmZmZ d dlZ G d d          ZddZdS )    N)AnyDictc                   R    e Zd ZU i Zeeef         ed<    ej	                    Z
d ZdS )DeviceContext	_CONTEXTSc                     || _         d S N)device)selfr
   s     V/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/_lazy/device_context.py__init__zDeviceContext.__init__   s        N)__name__
__module____qualname__r   r   strr   __annotations__	threadingLock_CONTEXTS_LOCKr    r   r   r   r      sL          "ItCH~"""#Y^%%N    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   r   torch._C._lazyr   r   r   r   r   r   <module>r"      s{                          
 
 
 
 
 
r   