
    3j                     P    S SK r S SKJr  S SKr " S S5      rSS\S-  S\4S jjrg)	    N)Anyc                   b    \ rS rSr% 0 r\\\4   \S'   \	R                  " 5       rS\SS4S jrSrg)DeviceContext   	_CONTEXTSdevicereturnNc                     Xl         g Nr   )selfr   s     T/home/wildlama/miniconda3/lib/python3.13/site-packages/torch/_lazy/device_context.py__init__DeviceContext.__init__   s        r   )__name__
__module____qualname____firstlineno__r   dictstrr   __annotations__	threadingLock_CONTEXTS_LOCKr   __static_attributes__ r   r   r   r      s3     "ItCH~"^^%Ns t r   r   r   r	   c                 L   U c)  [         R                  R                  R                  5       n O[	        U 5      n [
        R                     [
        R                  R                  U S 5      nUc  [        U 5      nU[
        R                  U '   UsS S S 5        $ ! , (       d  f       g = fr   )	torch_C_lazy_get_default_device_typer   r   r   r   get)r   devctxs     r   get_device_contextr%      su    ~88:V		%	%((,,VT:>"6*F.4M##F+ 
&	%	%s   AB
B#r   )r   typingr   torch._C._lazyr   r   r   r%   r   r   r   <module>r(      s/       
sTz 
] 
r   