
    ȅi                         S SK r S SKJr  S SKJr  S SKrS/r\" S\S9 " S S\R                  R                  R                  5      5       r
g)    N)Any)
deprecatedautocastzi`torch.cpu.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cpu', args...)` instead.)categoryc                     ^  \ rS rSrSr\R                  S:  a*  S\R                  S4S\	S\R                  S\	4U 4S jjjrS	 rS\R                  S4S\	S\R                  S\	4U 4S
 jjjrU 4S jrS\S\S\4U 4S jjrU 4S jrSrU =r$ )r      z
See :class:`torch.autocast`.
``torch.cpu.amp.autocast(args...)`` is deprecated. Please use ``torch.amp.autocast("cpu", args...)`` instead.
)      Tenableddtypecache_enabledc                 "   > [         TU ]  U 5      $ N)super__new__)clsr   r   r   	__class__s       U/home/james-whalen/.local/lib/python3.13/site-packages/torch/cpu/amp/autocast_mode.pyr   autocast.__new__   s     7?3''    c                     g r    )r   s    r   __init_subclass__autocast.__init_subclass__'   s    r   c                    > [         R                  R                  5       (       a  Xl        SU l        X l        g [        TU ]  SXUS9  g )Ncpu)r   r   r   )torch_jit_internalis_scripting_enableddevice
fast_dtyper   __init__)selfr   r   r   r   s       r   r#   autocast.__init__*   sG     ++--#MDK#O7} 	 	
r   c                 j   > [         R                  R                  5       (       a  U $ [        TU ]  5       $ r   )r   r   r   r   	__enter__)r$   r   s    r   r'   autocast.__enter__9   s+    ++--Kw ""r   exc_typeexc_valexc_tbc                 l   > [         R                  R                  5       (       a  g [        TU ]  XU5      $ r   )r   r   r   r   __exit__)r$   r)   r*   r+   r   s       r   r-   autocast.__exit__?   s-    ++--w6::r   c                 l   > [         R                  R                  5       (       a  U$ [        TU ]  U5      $ r   )r   r   r   r   __call__)r$   funcr   s     r   r0   autocast.__call__D   s-    ++--Kw%%r   )r    r!   r"   )__name__
__module____qualname____firstlineno____doc__sysversion_infor   bfloat16boolr   r   r   r#   r'   r   r-   r0   __static_attributes____classcell__)r   s   @r   r   r      s     '! !!&"&		(	( ;;	(  		( 	(	
 "^^"	

 {{
 	
 
#; ;s ;C ;
& &r   )r8   typingr   typing_extensionsr   r   __all__FutureWarningampautocast_moder   r   r   r   <module>rD      sR    
  (  , ?
6&uyy&&// 6&
6&r   