
    קg                         d dl Z d dlmZmZ d dlZd dlmZmZmZ d dl	m
Z
 g dZe j        d             Zd Zeeeedf         f         ZdS )	    N)TupleUnion)*get_single_level_autograd_function_allowed*set_single_level_autograd_function_allowedunwrap_if_dead)
exposed_in)r   	argnums_t%enable_single_level_autograd_functionunwrap_dead_wrappersc               #      K   	 t                      } t          d           d V  t          |            d S # t          |            w xY w)NT)r   r   )
prev_states    R/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/_functorch/utils.pyr
   r
      sV      ??AA
248882:>>>>>2:>>>>s	   !6 Ac                 8    t          d | D                       }|S )Nc              3   l   K   | ]/}t          |t          j                  rt          |          n|V  0d S )N)
isinstancetorchTensorr   ).0args     r   	<genexpr>z'unwrap_dead_wrappers.<locals>.<genexpr>"   sQ        JMz#u|<<Es#         )tuple)argsresults     r   r   r       s4      QU    F Mr   .)
contextlibtypingr   r   r   torch._C._functorchr   r   r   torch.utils._exposed_inr   __all__contextmanagerr
   r   intr	    r   r   <module>r#      s                         
 / . . . . .   ? ? ?   #uS#X&'			r   