o
    j9:j                     @   s   d dl Z d dlmZ d dl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df B ZdS )    N)	Generator)Any)*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   )
prev_state r   ]/home/nk/hobo-godmode/plappi-mvp/.venv/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torch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   torch._C._functorchr   r   r   torch.utils._exposed_inr   __all__contextmanagerr	   r   r
   intr   r   r   r   r   <module>   s    "	