
    Og                         d dl Z d dlmZ d dlmZ d dlmZ d dlm	Z	 deege	f   deege	f   fdZ
deeege	f   deeege	f   fd	Zy)
    N)Callable)'NativeFunctionWithDifferentiabilityInfo)native_function_manager)Tfuncreturnc                 \     t        j                         dt        dt        f fd       }|S )Nfr   c                 j    t        | j                        5   |       cd d d        S # 1 sw Y   y xY wNr   r   )r
   r   s    i/var/www/html/suriana-translation/venv/lib/python3.12/site-packages/torchgen/packaged/autograd/context.pywrapperzAwith_native_function_with_differentiability_info.<locals>.wrapper   s+    $QVV, 	7	 	 	s   )2)	functoolswrapsNFWDIr   r   r   s   ` r   0with_native_function_with_differentiability_infor      s5     __T5 Q   N    c                 h     t        j                         dt        dt        dt        f fd       }|S )Nr
   keyr   c                 l    t        | j                        5   | |      cd d d        S # 1 sw Y   y xY wr   r   )r
   r   r   s     r   r   zIwith_native_function_with_differentiability_info_and_key.<locals>.wrapper   s-    $QVV, 	 3<	  	  	 s   	*3)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   