i2&%SrSSKrSSKrSSKJrJrJr SSKrSSKJ r SSK J r J r 0r \\ \ 4\S'0r\\ \\ 4\S'0r\\ \\ 4\S'0r\\ \\ 4\S '0r\\ \\ 4\S 'S \R&RS \\ 4S jrS\ S \\ 4SjrS\ S \\4SjrS\ S \\4SjrS\S \\\44SjrS\S\S \ 4SjrS\ S \ 4SjrS\ S S4SjrS\ S \\ \\44SjrS\ S \\ 4SjrS\S\S \4Sjr S\S\S \4Sjr!g)a Handle the following op convertions: - convert a functional op to an out variant op - convert an out variant op to a scratch op. We assume there is already a functionalization pass being done that removes aliases and inplace variants. For the to_out_variant convertion, The functional variant will be represented as qualified op name plus the overload name. The returned out variant constains the following information - the OpOverload for the out variant - the list of keyward arguments names that are out variables. There should be at least one out variables. Some ops may also have multiple out variables, e.g. aten::topk.values returns both values and indices for the topk elements. N)DictOptionalTuple) OpOverload)FunctionSchema SchemaKind_op_overload_to_schema_cache_func_to_out_variant_map_out_variant_to_scratch_map_mutable_to_out_variant_map_schema_mismatch_map pybind_schemareturncSn[R"[U55nU$![[[ 4a& [ R"S[U535 U$f=f)a9 We have 2 FunctionSchema definitions in python. One is defined in torchgen (call it native FunctionSchema), another is a pybind of c10::FunctionSchema (call it pybind FunctionSchema). Because we want to leverage torchgen to handle out variant, we will convert any pybind FunctionSchema to native FunctionSchema. NzFail to parse function schema: )rparsestr RuntimeErrorAssertionError ValueErrorloggingdebug)r native_schemas _/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/exir/operator/convert.py_pybind_schema_to_native_schemar4sjMN&,,S-?@ $ # .* 5 N  7M8J7KLM # Ns%7A A  op_overloadcz[RU5nU(d[UR5nU[U'U$N)r getr_schema)rrs r_get_overload_schemar Us6044[AM 7 8K8KL 4A$[1 c*[[U55$r)get_out_args_from_schemar )rs rget_out_args_from_opoverloadr$]s #$8$E FFr!out_var_schemacUR5(d SU35e[SURR55$)z_ Assume the input is the schema for an out variant. Return the name list of the out arguments. z Expect an out variant, but get: c38# UHoRv M g7fr)name).0args r +get_out_args_from_schema..isB%Ac%As) is_out_fntuple argumentsout)r%s rr#r#asH   ""; ).)9:;" B^%=%=%A%AB BBr!qualified_opnamectURS5n[U5S:wa[SU35e[U5$)zj Given a qualified opname like aten::add, return a tuple for namespace (aten here) and op name (add here) z::zInvalid qualified_opname )splitlenrr.)r1 ns_and_opnames rparse_qualified_opnamer7lsA %**40M =Q67G6HIJJ  r!overloadc[U5up#U(dSn[[[[RU5U5U5$)zq Arguments: qualified_opname: string like {namespace}::{op name} overload: the overload string of the op default)r7getattrtorchops)r1r8nsopnames rget_op_overloadr@ws8 ((89JB  77599b16:H EEr!schemac[URR5nURRn[X5$r)rr( overload_namer@)rAqualified_namer8s rschema_to_opoverloadrEs0))*N{{((H > 44r!opc[UR5nUR5[R[R [R 4;deUR5[R:Xa U[;aeUR5[R :Xa U[;aeUR5[R :Xa U[;ae[URR5n[RRU5Vs/sHn[U5PM nnUVs/sH oUcMUPM nn0nUH`nUR5n[ R""USS9nUR5nUR%U05n X;d SUSU35eXYU'Mb UR'5GH,n U R)[R5n U R)[R 5n U R)[R 5n U R)[R*5n U (aLU (a [-U 5OS[[-U 5'U c%[/SU5S5nU[0[-U 5'U (a$U (a [-U 5OS[[-U 5'U (dGMU (a [-U 5OS[[-U 5'GM/ gs snfs snf)a op can either be a functional op, mutable op, or out variant op. This method is only called if 1. either op is a functional op and it's missing in the _func_to_out_variant_map cache. 2. or op is a out variant op and it's missing in the _out_variant_to_scratch_map cache. Setup entries in _func_to_out_variant_map and _out_variant_to_scratch_map for all ops sharing the same op name as the passed in OpOverload. N)returnszSchema of kind z already exist for c3l# UH*oR5[R:XdM&Uv M, g7fr)kindrr0)r)ss rr+%set_mapping_for_op..s!J 1vvx:>>/IQQ s%4 4)rrrKr functionalr0mutabler r r rr(r<_C_jit_get_schemas_for_operator signature dataclassesreplace setdefaultvaluesrscratchrEnextr )rFrr1r all_schemasrAgroup_by_signaturerRrK group_by_kindfunc_op_schemar%mutable_op_schemascratch_schemamismatched_out_schemas rset_mapping_for_opr`s4BJJ?M    $    5 55"@X:X   .29T3T   2 22r=X7X 2::??+#XXCCDTUUM ( 6U )4J f6 KJFH$$&  '' 2> {{}*55iD  % ? TF"5fX > ? %$d',,224 &**:+@+@A&**:>>:)--j.@.@A&**:+=+=> 8F$^4D %%9.%I J%BFJ JDC%*%%9.%IJ 8F$^4D ((<^(L M  8F$^4D ((<=N(O P;5A Ks#K:>K?K?cj[U5nUR5(a U[U54$UR5[R :Xd=UR5[R :XdSUR5SU35eU[;aU[;a [U5 U[;a [UnO[RU5nU(dLSUS3nU[;a1[U(a#USUR5SUS[U3- n[U5eU[U54$)z Convert the passed in OpOverload to its out variant. Raise an exception if on return the op_overload is not guaranteed to be an out variant. If a conversion is found, return the out variant OpOverload alongwith the name of out arguments. z#Expect a functionalish op, but get  z'Missing out variant for functional op: zf . Make sure you have loaded your custom operator library for compiler. E.g., custom_ops_generated_libz( Found an out variant for operator name zE but its schema mismatched with functional op. functional op schema: z out variant op schema: )r r-r#rKrrNrOr r r`rr r(rr$)rrAout_varmsgs rto_out_variantresD"+ .F 4V<<<  ... ;;=J.. .F -V[[]O1VHEF / 33 : : ;'11-k:*..{; 7x@fg . .#K0? @P@P@R?ST006x01Ek1R0SU 3 09 99r!c[U5nUR5[R:wa([R "SUR535 gU[ ;a [U5 [ RU5nU$)Nz(Expect an out variant op as input, got: ) r rKrr0rrr r`r)rrA scratch_ops r to_scratch_oprhsf !+ .F {{} & @PQ55;',00=J r!cV[X5n[U5nUcgUR5$NF)r@r r-r1r8rrAs ris_out_variantrl&s.!"2=K !+ .F ~    r!cx[X5n[U5nUcgUR5[R:H$rj)r@r rKrinplacerks ris_inplace_variantro.s6!"2=K !+ .F ~ ;;=J.. ..r!)"__doc__rSrtypingrrrr< torch._opsrtorchgen.modelrrr __annotations__r r r r rPrr rr$r#r7r@rEr`rerhboolrlrorHr!rrvs"(( !5 BDd:~#=>C DF$z8J+??@EFHT*hz.B"BCHFHT*hz.B"BCH DFd:x'??@E88** nBjXn5MGjGU3ZGC^Cc C S U38_  Fc FS FZ F55J5 [:[$[|): ):uZs5K/L):Xzhz.B&SCD////r!