iSSKrSSKrSSKrSSKJr SSKJrJrJrJ r SSK r SSK J r "SS\ R5rS\S \ \S \4S jrS \\S S4Sjr\S:Xa\"\R*SS5 gg!\ a SSK Jr NZf=f)N) defaultdict)AnyDictListOptional) CSafeLoader) SafeLoaderc,^\rSrSrSU4SjjrSrU=r$)BlankLineDumperct>[TU]U5 [UR5S:Xa[TU]5 gg)N)superwrite_line_breaklenindents)selfdata __class__s b/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/codegen/tools/merge_yaml.pyr BlankLineDumper.write_line_breaks1  & t||  ! G $ & ")N)__name__ __module__ __qualname____firstlineno__r__static_attributes__ __classcell__)rs@rr r s ''rr functions_yaml_pathfallback_yaml_path output_dirc [RRUS5nS[S[4Sjn[ U5n[ R"U[S9n[[5nUH nXU"U5'M SSS5 Ubn[RRU5(aJ[ U5n[ R"U[S9n U HnU"U5n U W;dMXU 'M SSS5 [ US5n[ R"[WR55U[SSSS 9 SSS5 g!,(df  N=f!,(df  Ni=f!,(df  g=f) Nz merged.yamlfuncreturnc`SU;aUSOUSRS5SnSU;aSU-nU$)aget the canonical name of an operator "op" and "func" are two keywords we are supporting for yaml files. To give an example: - op: add.Tensor # mostly used for binding ATen ops to kernels - func: add.Tensor(Tensor self, Tensor other, Scalar alpha) # mostly used for defining custom ops. These two will be supported Args: func (object): yaml object Returns: str: canonical name of the operator opr$(rz::zaten::)split)r$opnames rget_canonical_opname#merge..get_canonical_opname s@ $t|df1C1CC1H1K v &F r)LoaderwFi)Dumperdefault_flow_style sort_keyswidth)ospathjoinobjectstropenyamlloadr-rexistsdumplistvaluesr ) r r!r" output_filer+f functions_objfunctions_dictr$ fallback_objr*s rmergerDs)'',,z=9K6c( ! "a !F3 ,7,?!D9=/5 6" # %"''..9K*L*L $ %99Qv6L$-d3/-16*%& k3 1 &&( ) "$    # " & % s)5E< E  E E,argvr%c[R"SS9nURSSSSS9 URSS S S S9 URS S SS9 URU5nURb.[ R RUR5(de[URURUR5 g)aMerge functions.yaml and fallback yaml. The output yaml will be a union of all entries in functions.yaml and fallback yaml, with operator entries in functions.yaml overriding entries with the same op name in fallback yaml. E.g., functions.yaml: - op: add.Tensor - kernel: add_impl fallback yaml: - op: add.Tensor - kernel: add_fallback - op: relu - kernel: relu_fallback Merged: - op: add.Tensor - kernel: add_impl - op: relu - kernel: relu_fallback zQMerge functions.yaml, custom_ops.yaml with fallback yaml, for codegen to consume.) descriptionz--functions-yaml-pathz--functions_yaml_pathz'path to the functions.yaml file to use.T)helprequiredz--fallback-yaml-pathz--fallback_yaml_pathzpath to fallback yaml file.Fz --output_dirz+The directory to store the output yaml fileN) argparseArgumentParser add_argument parse_argsr r3r4r;rDr!r")rEparseroptionss rmainrPLs( $ $gF  6    *   ; %G  & & 2rww~~##88  ' % %w'A'A7CUCUVr__main__r)rJr3sys collectionsrtypingrrrrr9rr- ImportErrorr SafeDumperr r7rDrPrrErrrrWs #,, ** 'doo', s,  , SV, ^-WtCy-WT-W` z!"W*)*sA66 BB