U
    •Ê¹j¯  ã                   @   sˆ   d dl Z d dlmZ d dlmZ d dlmZ d dlm	Z	 eege	f eege	f dœdd„Z
eeege	f eeege	f dœd	d
„ZdS )é    N)ÚCallable)Ú'NativeFunctionWithDifferentiabilityInfo)Únative_function_manager)ÚT)ÚfuncÚreturnc                    s"   t  ˆ ¡ttdœ‡ fdd„ƒ}|S )N)Úfr   c              
      s*   t | jƒ� ˆ | ƒW  5 Q R £ S Q R X d S ©N©r   r   )r   ©r   © úa/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torchgen/packaged/autograd/context.pyÚwrapper   s    zAwith_native_function_with_differentiability_info.<locals>.wrapper)Ú	functoolsÚwrapsÚNFWDIr   ©r   r   r   r   r   Ú0with_native_function_with_differentiability_info   s    r   c                    s$   t  ˆ ¡tttdœ‡ fdd„ƒ}|S )N)r   Úkeyr   c              
      s,   t | jƒ� ˆ | |ƒW  5 Q R £ S Q R X d S r	   r
   )r   r   r   r   r   r      s    zIwith_native_function_with_differentiability_info_and_key.<locals>.wrapper)r   r   r   Ústrr   r   r   r   r   Ú8with_native_function_with_differentiability_info_and_key   s    r   )r   Útypingr   Ztorchgen.api.autogradr   r   Ztorchgen.contextr   Ztorchgen.utilsr   r   r   r   r   r   r   r   Ú<module>   s   þþ