o h.@sddlmZddlmZmZddlmZddlZddlm Z m Z ddl m Z m Z ddlmZmZmZmZmZddlmZed d d gZgd Zd)ddZd*ddZd+ddZ  d,d-d$d%Zd.d'd(ZdS)/) annotations) defaultdict namedtuple)AnyN) ETKernelIndex ETKernelKey) LineLoaderparse_native_yaml)BackendMetadata DispatchKeyFunctionSchemaNativeFunction OperatorName)NamespaceHelper ETParsedYamlnative_functionset_kernel_indices)kernels type_aliasdim_order_aliaseidict[str, object]return"dict[ETKernelKey, BackendMetadata]cCs|}|dd}duriS|di}|di}|ddi}|D]Y}|d}|dur6|d|d}tj|dd } | jd d } t| jd | d d} |dur]tddd gnt |||} | D]} | |vszJdt | dt || || <qfq&|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 rNrr__line__arg_meta kernel_name) max_levelat)defaultFz::native)kernel structured cpp_namespaceTzDuplicate kernel key:  ) copypopgetrfrom_namespaced_entityget_cpp_namespacer entity_namer gen_from_yamlstr)rerrrkernel_mappingentryrrnamespace_helperkernel_namespacebackend_metadata kernel_keys kernel_keyr$r$m/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torchgen/executorch/parse.pyparse_from_yamls@          r7esobjectrcCsi}|D]?}|}|d}t|tsJd|tj|dd}t|jj }||vs5Jd|dt t |}dkrC|||<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: namespaced_entityrzDuplicate func found in yaml: z alreadyr) r&r' isinstancer-rr)r parser+namelenr7r)r8indicesrr.funcsr1opnameindexr$r$r6parse_et_yaml_structLs rG"dict[OperatorName, dict[str, Any]]cCs|tt}|D]5}|d}t|tsJd|tj|dd}t|j j }t D]}||}dur:||||<q)q|S)zGiven a loaded yaml representing a list of operators, extract the kernel key related fields indexed by the operator name. r:r;r<r=N) rdictr(r?r-rr)r r@r+rA ET_FIELDS)r8fieldsrrDr1rEfieldvaluer$r$r6extract_kernel_fieldsds  rNFpathr-tags_yaml_path ignore_keysset[DispatchKey] | Noneskip_native_fns_genbool?tuple[list[NativeFunction], dict[OperatorName, dict[str, Any]]]cCsbt|}tj|td}Wdn1swYt|}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 )LoaderN)rS loaded_yaml)openyamlloadrrNstrip_et_fieldsr r)rOrPrQrSfr8 et_kernel native_yamlr$r$r6 parse_et_yamlxs  r_NonecCs$|D] }tD]}||dqqdS)zGiven a loaded yaml representing a list of operators, remove ET specific fields from every entries for BC compatibility N)rJr')r8r0rLr$r$r6r[s r[)rrrr)r8r9rr)r8r9rrH)NF) rOr-rPr-rQrRrSrTrrU)r8r9rr`) __future__r collectionsrrtypingrrYtorchgen.executorch.modelrr torchgen.genrr torchgen.modelr r r r rtorchgen.utilsrrrJr7rGrNr_r[r$r$r$r6s"     1