o i/@sJddlmZmZddlmZmZmZmZmZm Z ddl Z ddl m Z m Z ddlmZmZddlmZmZmZmZmZddlmZedd d gZgd Zd eeefd ee effddZded e fddZded eeeeefffddZ  ddededeeede d e eeeeeeeffff ddZ!ded dfddZ"dS)) defaultdict namedtuple)AnyDictListOptionalSetTupleN) ETKernelIndex ETKernelKey) LineLoaderparse_native_yaml)BackendMetadata DispatchKeyFunctionSchemaNativeFunction OperatorName)NamespaceHelper ETParsedYamlnative_functionsZet_kernel_indices)kernels type_aliasdim_order_aliaseireturncCs|}|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 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_yamlstr)rerrrZkernel_mappingentryrrnamespace_helperZkernel_namespaceZbackend_metadataZ kernel_keysZ kernel_keyr!r!h/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchgen/executorch/parse.pyparse_from_yamls@          r-escCsi}|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: Znamespaced_entityrzDuplicate func found in yaml: z alreadyr) r#r$ isinstancer(rr&rparser'namelenr-r )r.indicesrr)funcsr+opnameindexr!r!r,parse_et_yaml_structJs r;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/r0r1r2N) rdictr%r3r(rr&rr4r'r5 ET_FIELDS)r.fieldsrr8r+r9fieldvaluer!r!r,extract_kernel_fieldsbs  rAFpathtags_yaml_path ignore_keysskip_native_fns_gencCsbt|}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)rEZ loaded_yaml)openyamlloadr rAstrip_et_fieldsr r)rBrCrDrEfr.Z et_kernelZ native_yamlr!r!r, parse_et_yamlvs  rLcCs$|D] }tD]}||dqqdS)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,rJs rJ)NF)# collectionsrrtypingrrrrrr rHZtorchgen.executorch.modelr r Z torchgen.genr r Ztorchgen.modelrrrrrZtorchgen.utilsrrr=r(objectr-r;rAboolrLrJr!r!r!r,s4  "1"