'}h/\ddlmZmZddlmZmZmZmZmZm Z ddl Z ddl m Z m Z ddlmZmZddlmZmZmZmZmZddlmZedd d gZgd Zd eeefd ee effdZded e fdZded eeeeefffdZ ddededeeede d e eeeeeeeffff dZ!ded dfdZ"y)) defaultdict namedtuple)AnyDictListOptionalSetTupleN) ETKernelIndex ETKernelKey) LineLoaderparse_native_yaml)BackendMetadata DispatchKeyFunctionSchemaNativeFunction OperatorName)NamespaceHelper ETParsedYamlnative_functionset_kernel_indices)kernels type_aliasdim_order_aliaseireturnct|j}|jddx}iS|jdi}|jdi}|jddi}|D]}|jd}||jd|jd}tj|d } | j d } t | jd | d z} |tdd gntj|||} | D]-} | |vs"Jdt| zdzt|z| || </|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_mappingentryrr namespace_helperkernel_namespacebackend_metadata kernel_keys kernel_keys X/var/www/html/test/engine/venv/lib/python3.12/site-packages/torchgen/executorch/parse.pyparse_from_yamlr;sj  A55D))2 '(uu\2'>J,-EE2CR,HO D)9;N:99Z(   LL $ii . *AA 1 ,==d=K*#//+j8 T * +**8ZQ  & :J^3 (3z?:S@3q6I 3*:N: &  :-:8 esc|i}|D]}|j}|jd}t|ts Jd|t j |d}t j|jj}||vs Jd|dtt|x}dk7s|||<t|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_entityr"zDuplicate func found in yaml: z alreadyr) r*r+ isinstancer1rr-rparser/namelenr;r )r=indicesrr2funcsr5opnameindexs r:parse_et_yaml_structrLJs GIG $ GGIf %%<UG'<<%*AA#q  %%&6&B&BCHHW$W(Fvhh&WW$ **u +q 0#GFO $  !!r<cVtt}|D]}|jd}t|ts Jd|t j |d}tj|jj}tD]}|j|x}||||< |S)zGiven a loaded yaml representing a list of operators, extract the kernel key related fields indexed by the operator name. r?r@rArB) rdictr,rDr1rr-rrEr/rF ET_FIELDS)r=fieldsrrIr5rJfieldvalues r:extract_kernel_fieldsrSbs2=T1BF .v%%<UG'<<%*AA#q  %%&6&B&BCHH .E&3(-vu% . . Mr<pathtags_yaml_path ignore_keysskip_native_fns_genct|5}tj|t}dddt }t |t |||||}|j|fS#1swY=xYw)zParse native_functions.yaml into NativeFunctions and an Operator Indexed Dict of fields to persist from native_functions.yaml to functions.yaml )LoaderN)rW loaded_yaml)openyamlloadr rSstrip_et_fieldsrr)rTrUrVrWfr= et_kernel native_yamls r: parse_et_yamlrbvst d-q YYq ,-&b)IB# / K  ' ' 22--s A$$A-cL|D]}tD]}|j|d!y)zGiven a loaded yaml representing a list of operators, remove ET specific fields from every entries for BC compatibility N)rOr+)r=r4rQs r:r^r^s0# #E IIeT " ##r<)NF)# collectionsrrtypingrrrrr r r\torchgen.executorch.modelr r torchgen.genr rtorchgen.modelrrrrrtorchgen.utilsrrrOr1objectr;rLrSboolrbr^r(r<r:rls/88 @6+.+=?R*ST  9 .S&[).d;3O.P.b"V" "0flDcN.J)K./3 % 3 33#k*+3 3  4 lDcN&B!C CD 36##4#r<