o
    ht                     @   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dededededededdfddZdededededdf
ddZd ddZedkre  dS 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	|
|}|sDt
|| |||| t|| ||| t||	| t||| t|| || d S )Nderivatives.yaml	templatesc                 S   s   t | jS )N)r   namefunc)f r   ]/var/www/html/ai/venv/lib/python3.10/site-packages/torchgen/packaged/autograd/gen_autograd.py<lambda>E   s    zgen_autograd.<locals>.<lambda>)key)r   ospathjoinr   native_functionssortedfilter(is_native_function_selected_for_trainingr   r   r   r   r
   r   )r   r   r   r   r   r   differentiability_infosused_dispatch_keystemplate_pathnative_funcsfnsfns_with_diff_infosr   r   r   gen_autograd1   s8   		
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*   deprecated_pathr   r   r   gen_autograd_pythonc   s   
r2   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$   NATIVEzpath to native_functions.yaml)metavarhelptagszpath to tags.yamlr   OUTzpath to output directoryautogradAUTOGRADzpath to autograd directory)argparseArgumentParseradd_argument
parse_argsr.   r$   r7   r   r9   r   get_nop_selector)parserargsr   r   r   mainy   s"   rB   __main__)F)r   N)__doc__r;   r!   typingr   torchgen.apir   torchgen.api.autogradr   r   torchgen.genr   !torchgen.selective_build.selectorr    r	   gen_autograd_functionsr
   r   r   r   r   r   r   strboolr.   r2   rB   __name__r   r   r   r   <module>   s\    	
2


