o
    ߗi                     @   sv   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d Zd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                  c   s0    zt  } td d V  W t|  d S t|  w )NT)r   r   )
prev_state r   T/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/torch/_functorch/utils.pyr	      s   r	   c                 C   s   t dd | D }|S )Nc                 s   s(    | ]}t |tjrt|n|V  qd S )N)
isinstancetorchTensorr   ).0argr   r   r   	<genexpr>"   s    
z'unwrap_dead_wrappers.<locals>.<genexpr>)tuple)argsresultr   r   r   r
       s   r
   .)
contextlibtypingr   r   r   torch._C._functorchr   r   r   torch.utils._exposed_inr   __all__contextmanagerr	   r
   intr   r   r   r   r   <module>   s   
	