
    3j                         S SK r S SKJr  S SKJr  S SKrS SKJrJrJ	r	  S SK
Jr  / SQr\ R                  S\S   4S	 j5       rS
\\S4   S\\S4   4S jr\\\S4   -  rg)    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               #   v   #     [        5       n [        S5        S v   [        U 5        g ! [        W 5        f = f7f)NT)r   r   )
prev_states    P/home/wildlama/miniconda3/lib/python3.13/site-packages/torch/_functorch/utils.pyr
   r
      s.     ??A
2482:>2:>s   9) 969args.c                 *    [        S U  5       5      nU$ )Nc              3   z   #    U  H1  n[        U[        R                  5      (       a  [        U5      OUv   M3     g 7f)N)
isinstancetorchTensorr   ).0args     r   	<genexpr>'unwrap_dead_wrappers.<locals>.<genexpr>"   s.      QU#z#u||<<s#EQUs   9;)tuple)r   results     r   r   r       s      QU F M    )
contextlibcollections.abcr   typingr   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?  %S/!	r   