
    %Vji                         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	            Zd
eedf         deedf         f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               #      K   	 t                      } t          d           d V  t          |            d S # t          |            w xY w)NT)r   r   )
prev_states    P/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/_functorch/utils.pyr   r      sV      ??AA
248882:>>>>>2:>>>>s	   !6 Aargs.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)r   results     r   r   r       s4      QU    F Mr   )
contextlibcollections.abcr   typingr   r   r   torch._C._functorchr   r   r   torch.utils._exposed_inr	   __all__contextmanagerr   r   r   intr
    r   r   <module>r&      s       % % % % % %                  
 / . . . . .   ?y9I/J ? ? ? ?uS#X 5c?     #uS#X&'			r   