U <c@s ddlZddlZddlZddlmZmZmZddlmZddl m Z ddl m Z ddl mZmZmZddlmZmZedeeefd Zeeeeed d d Zedd ddZeeeddddZeeeddddZddddZedkreddedS)N)SequenceTypeVarUnion)set_simple_logging)gen)native_function_manager) DispatchKeyNativeFunctionsGroupNativeFunctionsViewGroup)config generator NativeGroupT)bound)grouped_native_functionsreturncsV|sgSg}tttftdddfdd|D}ddtj|dd d D}|S) N)grc Ss*t|t|W5QRSQRXdSN)rr is_supportedrrR/tmp/pip-unpacked-wheel-gikjz4vx/torchgen/static_runtime/gen_static_runtime_ops.pyrs z0group_functions_by_op_name..is_supportedc3s|]}|r|VqdSrr.0rrrr "sz-group_functions_by_op_name..cSsg|]\}}t|qSr)list)rkgrouprrr #sz.group_functions_by_op_name..cSs t|Sr)r Zfunc_name_base_strrrrr(z,group_functions_by_op_name..)key)rr r bool itertoolsgroupby)rgroupsZ eligible_opsrrrgroup_functions_by_op_names r&) cpp_file_pathrcCsddl}|dd|gdS)Nrz clang-formatz-i) subprocessrun)r'r(rrr clang_format0sr*)cpp_ops file_pathrc CsBd|}d|d}t|d}||W5QRXt|dS)N a// @lint-ignore-every CLANGTIDY HOWTOEVEN // AUTO-GENERATED FROM: torchgen/static_runtime/gen_static_runtime_ops.py #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include namespace torch { namespace jit { z* } // namespace jit } // namespace torch wjoinopenwriter*r+r,code generatedfrrr write_cpp6s (- r7c CsBd|}d|d}t|d}||W5QRXt|dS)Nr-aq// @lint-ignore-every CLANGTIDY HOWTOEVEN // AUTO-GENERATED FROM: torchgen/static_runtime/gen_static_runtime_ops.py #include #include #include #include "test_utils.h" using namespace caffe2; using namespace torch; using namespace torch::jit; using namespace torch::jit::test; using c10::IValue;  r.r/r3rrrwrite_test_cppjs  r9)rcstjdd}|jddddd|jdd d d d|jd d d dd|}tj|jd}tj|jd}t ||}|j |j }t t ddt|D}t|}fdd|D}fdd|D} ddt|D} t| } fdd| D} fdd| D} |dg| }| dg| }t||jt||jtdtt|tdt|tdd|D}td|tdt| tdd| D}td|td ||dS)!NzGenerate ATen source files) descriptionz-sz --source-pathz!path to source directory for ATenzcaffe2/aten/src/ATen)helpdefaultz-pz--generated-ops-cpp-pathz5path to directory to generate op dispatcher .cpp filez6caffe2/torch/csrc/jit/runtime/static/generated_ops.cppz-tz--generated-ops-test-cpp-pathz6caffe2/benchmarks/static_runtime/test_generated_ops.ccznative/native_functions.yamlznative/tags.yamlcSsg|]}t|tr|qSr) isinstancer rrrrrs zmain..csg|]}|tjqSr) out_variantrCPUrr%backend_indices op_generatorrrrscsg|]}|qSr)r>r@test_case_generatorrrrscSsg|]}t|tr|qSr)r=r rrrrrs csg|]}|tjqSr)viewrr?r@rArrrscsg|]}|qSr)rFr@rDrrrsr8z total grouped native ops: %dz'grouped native ops with out variant: %dcSsg|] }t|qSrlenr@rrrrsz/generated functions groups with out variant: %dz view grouped native ops: %dcSsg|] }t|qSrrGr@rrrrsz#generated functions view groups: %dz overall generated : %d)argparseArgumentParser add_argument parse_argsospathr0 source_pathrZparse_native_yamlnative_functionsrBr ZGenOpDispatcherZ GenOpTestCaseZget_grouped_native_functionsr&Z$get_grouped_by_view_native_functionsr7Zgenerated_ops_cpp_pathr9Zgenerated_ops_test_cpp_pathprintrHsum)parseroptionsZnative_yaml_pathZtags_yaml_pathZ parsed_yamlrPZnative_functions_groupsZsupported_functions_groupsZout_variant_op_resultZout_variant_test_resultZnative_functions_view_groupsZsupported_functions_view_groupsZview_op_resultZview_test_result op_resultZ test_resultZsupported_functions_numZsupported_view_functions_numr)rBrCrErmains             rV__main__F)Zescape_newlines)rIr#rMtypingrrrZ libfb.py.logrZtorchgenrZtorchgen.contextrZtorchgen.modelrr r Ztorchgen.static_runtimer r r r&strr*r7r9rV__name__rrrrs,     4_