o
    Zh                     @   s   d dl Z d dlm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ed fd	d
Zdeedf deedf fddZeeeedf f ZdS )    N)	Generator)Any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_wrappersreturn)NNNc                  c   s0    zt  } td d V  W t|  d S t|  w )NT)r   r   )Z
prev_state r   E/var/www/auris/lib/python3.10/site-packages/torch/_functorch/utils.pyr
      s   r
   args.c                 C   s   t dd | D }|S )Nc                 s   s(    | ]}t |tjrt|n|V  qd S )N)
isinstancetorchZTensorr   ).0argr   r   r   	<genexpr>"   s    
z'unwrap_dead_wrappers.<locals>.<genexpr>)tuple)r   resultr   r   r   r       s   r   )
contextlibcollections.abcr   typingr   r   r   Ztorch._C._functorchr   r   r   Ztorch.utils._exposed_inr   __all__contextmanagerr
   r   r   intr	   r   r   r   r   <module>   s    "	