U
    9%et                  	   @   s   d Z ddlZddlZddlmZ ddlmZ ddlmZm	Z	 ddl
mZ ddlmZ dd	lmZ dd
lmZmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ deeeeeeddddZeeeeddddZddddZedkre  dS )a~  
To run this file by hand from the root of the PyTorch
repository, run:

python -m tools.autograd.gen_autograd        aten/src/ATen/native/native_functions.yaml        aten/src/ATen/native/tags.yaml        $OUTPUT_DIR        tools/autograd

Where $OUTPUT_DIR is where you would like the files to be
generated.  In the full build system, OUTPUT_DIR is
torch/csrc/autograd/generated/
    N)List)cpp)match_differentiability_info'NativeFunctionWithDifferentiabilityInfo)parse_native_yaml)SelectiveBuilder   )gen_python_functions)gen_autograd_functions_libgen_autograd_functions_python)gen_inplace_or_view_type)gen_trace_type)gen_variable_factories)gen_variable_type)load_derivativesF)native_functions_path	tags_pathoutautograd_diroperator_selectordisable_autogradreturnc                 C   s   t tj|d| |\}}tj|d}t| |j}	tt|j|	dd d}
t	|
|}|st
|| |||| t|| ||| t||	| t||| t|| || d S )Nderivatives.yaml	templatesc                 S   s   t | jS )N)r   namefunc)f r   f/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torchgen/packaged/autograd/gen_autograd.py<lambda>E       zgen_autograd.<locals>.<lambda>)key)r   ospathjoinr   native_functionssortedfilterZ(is_native_function_selected_for_trainingr   r   r   r   r
   r   )r   r   r   r   r   r   differentiability_infosZused_dispatch_keystemplate_pathZnative_funcsfnsZfns_with_diff_infosr   r   r   gen_autograd1   sF    	   	    r+   )r   r   r   r   r   c                 C   sX   t tj|d| |\}}tj|d}t||| tj|d}t|| ||| d S )Nr   r   zdeprecated.yaml)r   r"   r#   r$   r   r	   gen)r   r   r   r   r(   _r)   Zdeprecated_pathr   r   r   gen_autograd_pythonc   s          r.   )r   c                  C   st   t jdd} | jdddd | jdddd | jd	d
dd | jdddd |  }t|j|j|j|jt	
  d S )Nz"Generate autograd C++ files script)descriptionr%   ZNATIVEzpath to native_functions.yaml)metavarhelptagszpath to tags.yamlr   ZOUTzpath to output directoryautogradZAUTOGRADzpath to autograd directory)argparseArgumentParseradd_argument
parse_argsr+   r%   r2   r   r3   r   Zget_nop_selector)parserargsr   r   r   mainy   s*        r:   __main__)F)__doc__r4   r"   typingr   Ztorchgen.apir   Ztorchgen.api.autogradr   r   Ztorchgen.genr   Z!torchgen.selective_build.selectorr    r	   Zgen_autograd_functionsr
   r   r   r   r   r   r   strboolr+   r.   r:   __name__r   r   r   r   <module>   sB   	 3