
    JTh                         S SK r S SKJr  S SKJr  S SKJr  S SKJ	r	  S\\/\	4   S\\/\	4   4S jr
S\\\/\	4   S\\\/\	4   4S	 jrg)
    N)Callable)'NativeFunctionWithDifferentiabilityInfo)native_function_manager)Tfuncreturnc                 b   ^  [         R                  " T 5      S[        S[        4U 4S jj5       nU$ )Nfr   c                 t   > [        U R                  5         T" U 5      sS S S 5        $ ! , (       d  f       g = fNr   r   )r
   r   s    Z/var/www/auris/envauris/lib/python3.13/site-packages/torchgen/packaged/autograd/context.pywrapperAwith_native_function_with_differentiability_info.<locals>.wrapper   s"    $QVV,7 -,,   )
7)	functoolswrapsNFWDIr   r   r   s   ` r   0with_native_function_with_differentiability_infor      s5     __T5 Q   N    c                 n   ^  [         R                  " T 5      S[        S[        S[        4U 4S jj5       nU$ )Nr
   keyr   c                 t   > [        U R                  5         T" X5      sS S S 5        $ ! , (       d  f       g = fr   r   )r
   r   r   s     r   r   Iwith_native_function_with_differentiability_info_and_key.<locals>.wrapper   s"    $QVV,< -,,r   )r   r   r   strr   r   s   ` r   8with_native_function_with_differentiability_info_and_keyr      s<     __T 5  s  q     Nr   )r   typingr   torchgen.api.autogradr   r   torchgen.contextr   torchgen.utilsr   r   r   r    r   r   <module>r#      sn      R 4 

E7A:
ugqj
E3<?
#uclAor   