U ʹj/ @s8ddlmZmZddlmZmZmZmZmZm Z ddl Z ddl m Z m Z ddlmZmZddlmZmZmZmZmZddlmZedd d gZd d d gZeeefee efdddZee dddZeeeeeeffdddZdeeeeee e eeeeeeefffdddZ!eddddZ"dS)) defaultdict namedtuple)AnyDictListOptionalSetTupleN) ETKernelIndex ETKernelKey) LineLoaderparse_native_yaml)BackendMetadata DispatchKeyFunctionSchemaNativeFunction OperatorName)NamespaceHelper ETParsedYamlnative_functionsZet_kernel_indiceskernels type_aliasdim_order_alias)eireturncCs|}|dd}dkr iS|di}|di}|ddi}|D]}|d}|dk rl|d|d}tj|dd } | jd d } t| jd | d d} |dkrtddd gn t |||} | D]0} | |kst dt | dt || || <qqL|S)a3Given a loaded yaml representing kernel assignment information, extract the mapping from `kernel keys` to `BackendMetadata` (the latter representing the kernel instance) Args: ei: Dict keys {kernels, type_alias, dim_order_alias} See ETKernelKey for description of arguments rNrrZ__line__arg_meta kernel_name) max_levelat)defaultFz::native)ZkernelZ structuredZ cpp_namespaceTzDuplicate kernel key:  ) copypopgetrfrom_namespaced_entityZget_cpp_namespacer entity_namer Z gen_from_yamlAssertionErrorstr)rerrrZkernel_mappingentryrrnamespace_helperZkernel_namespaceZbackend_metadataZ kernel_keysZ kernel_keyr!r!X/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torchgen/executorch/parse.pyparse_from_yamls@          r.)esrcCsi}|D]~}|}|d}t|ts6td|tj|dd}t|j j }||ksjtd|dt t |}dkr|||<qt |S)zGiven a loaded yaml representing a list of operators, for each op extract the mapping of `kernel keys` to `BackendMetadata` (the latter representing the kernel instance that should be used by the kernel key). func not a str: Znamespaced_entityrzDuplicate func found in yaml: z alreadyr)r#r$ isinstancer)r(rr&rparser'namelenr.r )r/indicesrr*funcsr,opnameindexr!r!r-parse_et_yaml_structJs  r<cCs|tt}|D]j}|d}t|ts2td|tj|dd}t |j j }t D]"}||}dk rR||||<qRq |S)zGiven a loaded yaml representing a list of operators, extract the kernel key related fields indexed by the operator name. r0r1r2r3N) rdictr%r4r)r(rr&rr5r'r6 ET_FIELDS)r/fieldsrr9r,r:fieldvaluer!r!r-extract_kernel_fieldsbs rBF)pathtags_yaml_path ignore_keysskip_native_fns_genrc CsNt|}tj|td}W5QRXt|}t|t|||||d}|j|fS)zParse native_functions.yaml into NativeFunctions and an Operator Indexed Dict of fields to persist from native_functions.yaml to functions.yaml )Loader)rFZ loaded_yaml)openyamlloadr rBstrip_et_fieldsr r)rCrDrErFfr/Z et_kernelZ native_yamlr!r!r- parse_et_yamlvs rMcCs$|D]}tD]}||dq qdS)zGiven a loaded yaml representing a list of operators, remove ET specific fields from every entries for BC compatibility N)r>r$)r/r+r@r!r!r-rKsrK)NF)# collectionsrrtypingrrrrrr rIZtorchgen.executorch.modelr r Z torchgen.genr r Ztorchgen.modelrrrrrZtorchgen.utilsrrr>r)objectr.r<rBboolrMrKr!r!r!r-s*    1