Χg.ddlmZddlmZmZddlmZddlZddlm Z m Z ddl m Z m Z ddlmZmZmZmZmZddlmZed d d gZgd Zd$dZd%dZd&dZ d'd(d!Zd)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]c|}|ddx}iS|di}|di}|ddi}|D]}|d}||d|d}tj|d } | d } t | jd | d z} |tdd gntj |||} | D]8} | |vs-Jdt| zdzt|z| || <9|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_keys U/var/www/html/ai-engine/env/lib/python3.11/site-packages/torchgen/executorch/parse.pyparse_from_yamlr9s  A55D)))2 '(uu\2'>'>J,-EE2CR,H,HO D)))9;N::99Z((   LL $ $ $ii .. *A 1   ,==d=KK*#/+j8   T * * * + +*8ZQQ  & : :J^333(3z??:S@3q66I433*:N: & &  : esobjectrci}|D]}|}|d}t|ts Jd|t j|d}t j|jj }||vsJd|dtt|x}dkr|||<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) isinstancer/rr+r parser-namelenr9r)r;indicesrr0funcsr3opnameindexs r8parse_et_yaml_structrKLs GIG $ $ GGIIf %%%<<'r?r@rA) rdictr*rCr/rr+r rDr-rE ET_FIELDS)r;fieldsrrHr3rIfieldvalues r8extract_kernel_fieldsrSds2=T1B1BF . .v%%%<<'rosw""""""//////// @@@@@@@@66666666+*****z.+=?R*STT  9 8 8 ....b""""0.,0 % 333336######r: