[Th0SSKrSSKrSSKrSSKJr SSKJrJr SSKrSSKJrJ r J r Jr SSK J r \Rr \ "S5SS.S\RS \\S \4S jj5rS \\\R$4S \S\S\S\4 SjrSr\ S\R,\ S\\ S\S\S\S\R4S0rSr\"5rS\R<S \4SjrS\\ S \\4Sjr!g)N) GenericAlias)OptionalUnion)devicedtypeTensortypes) exposed_inz torch.library)op_nameprototype_functionr returnc  ^^^^^^^SnURmSm[R"U5mU4SjmS[4UUU4SjjmS[[ [ [[4S4S[[[RS4[44U4S jjmS [ [ [[4S[[R[44UU4S jjm/n[5nS n[TRR55GHfunup[U 5(dT"S 5 U R [R"R$:XaU(dUR'S5 SnU R([R"R*La T"SUS35 T"U R(5upU [,;aU R.[La][1U 5n Sn U [,R35;aSU S3n T"SUSU R(SU S[,R35S3 5 O.T"SUSU R(S[,R35S35 [,U n[ U5[:Xa;X:wa [5S5eUR7S5(aSUSU[9S5S3nO;X;a6UR7S5(d T"SUS35 SUSU[9S5S3nUR;U5 U R<[R"R*LaUR'USU35 GM5SnU R<b*[?U R<[@[B[45(a[U R<5nO[?U R<[[DRF45(aS U R<S 3nO[?U R<[DRH5(a>[U R<5nS!nUR7U5(deU[9U5SnO"T"SUS"[ U R<5S#35 UR'USUS$U35 GMi X:wa([U5U- n[9U5S%:a T"US&35 T"TRJ5unn [MUT5nUbUS'S(ROU5S)U3$S'S(ROU5S)U3$)*aParses the schema of a given function with type hints. The schema is inferred from the function's type hints, and can be used to define a new operator. We make the following assumptions: * None of the outputs alias any of the inputs or each other. * | String type annotations "device, dtype, Tensor, types" without library specification are | assumed to be torch.*. Similarly, string type annotations "Optional, List, Sequence, Union" | without library specification are assumed to be typing.*. * | Only the args listed in ``mutates_args`` are being mutated. If ``mutates_args`` is "unknown", | it assumes that all inputs to the operator are being mutates. Callers (e.g. the custom ops API) are responsible for checking these assumptions. Args: prototype_function: The function from which to infer a schema for from its type annotations. op_name (Optional[str]): The name of the operator in the schema. If ``name`` is None, then the name is not included in the inferred schema. Note that the input schema to ``torch.library.Library.define`` requires a operator name. mutates_args ("unknown" | Iterable[str]): The arguments that are mutated in the function. Returns: The inferred schema. Example: >>> def foo_impl(x: torch.Tensor) -> torch.Tensor: >>> return x.sin() >>> >>> infer_schema(foo_impl, op_name="foo", mutates_args={}) foo(Tensor x) -> Tensor >>> >>> infer_schema(foo_impl, mutates_args={}) (Tensor x) -> Tensor unknownNc(>[SUSTS35e)Nzinfer_schema(func): z Got func with signature )) ValueError)whatsigs S/var/www/auris/envauris/lib/python3.13/site-packages/torch/_library/infer_schema.pyerror_fninfer_schema..error_fnDs /v5NseSTUVVannotation_typecX>[UTT5$![a T"SUS35 gf=f)NzUnsupported type annotation z. It is not a type.)eval Exception)rr pf_globals pf_localss rconvert_type_string)infer_schema..convert_type_stringGs< Y? ?  ..??RS  s ))tys.r c>/nSnUH"nT"U5up4URU5 X$-nM$ U(a [U5S4$US4$)NFT)appendtuple)r!reschangedty ty_changedunstringify_types runstringify_types'infer_schema..unstringify_typesOsVB-b1NB JJrN  !G :t# #: rr'c>[U[5(a T"U5S4$[R"U5=n(a2T"[R"U55up#U(a [ X5S4$US4$)NTF) isinstancestrtyping get_originget_argsr)r'originargs args_changedrr*s rr)&infer_schema..unstringify_type]sk b#  &r*D0 0((, ,V ,!26??23F!G D#F14775yrFz>We do not support positional-only args, varargs, or varkwargs.*Tz Parameter z must have a type annotation.z z For example, z. z has unsupported type z]. We do not support Tuple inputs in schema. As a workaround, please try to use List instead. zThe valid types are: .. The valid types are: zomutates_args must either be a sequence of the names of the arguments that are mutated or the string 'unknown'. rzTensor(az!)zM is in mutable_args but only Tensors or collections of Tensors can be mutated "ztorch.z' has an unsupported default value type z;. Please file an issue on GitHub so we can prioritize this.=rz in mutates_args were not found in the custom op's signature. mutates_args should contain the names of all args that the custom op mutates, or just the string 'unknown' if you don't know.(, z) -> )( __globals__inspect signaturer.r$rtypeobjectr/Anyboolset enumerate parametersitemssupported_paramkind Parameter KEYWORD_ONLYr# annotationemptySUPPORTED_PARAM_TYPES __origin__ tuple_to_listkeysr startswithlenadddefaultr-intfloattorchrrreturn_annotation parse_returnjoin)r mutates_argsr UNKNOWN_MUTATESparams seen_argssaw_kwarg_only_argidxnameparamr_ list_typeexample_type_str schema_type default_repr dtype_repr torch_dotmutates_args_not_seenrZretrrrrrr)r*s @@@@@@@r infer_schemarnsT O#//JI   . /CWS  5fs*+S0 1  uVZZ_%t+ ,  U4<#45 % D@P:Q  FI'(<(<(>?]du%% U V ::**77 7% c"%)"   w0066 6 z$'DE F.e.>.>? "7 7))U2)/: #)  5 : : <<)6yk'G$ &??!"5==/3 EMM5;;77 / $ !,,Y7777)#i.*:;  &MdSXS`S`NaMbcPQ MM[M4&,@ AM@N& #L 1I = $ % ) ()*UV  ,C,A,ABq (( 3C!DIIf-.eC599 tyy !se ,,r base_typecpp_type list_baseoptional_base_listoptional_list_basec^UT4[RUTS34/nS[[[R44SjnU(a!UR U4SjU"U555 U(a2UR U4SjU"[RU555 U(a!UR U4SjU"U555 U$)N?typc[RU[RU[[R RU45[[ U454$N)r/SequenceListr collectionsabclist)rvs rderived_seq_types(derived_types..derived_seq_typessF OOC KK  11C6 : v &   rc30># UH oTS34v M g7f)z[]N.0seq_typrps r derived_types..s 6R7 "o &6Rsc32># UH nUTS34v M g7f)z?[]Nrrs rrrs$ H #& 'Hsc3T># UHn[RUTS34v M g7f)z[]?N)r/rrs rrrs- 7__W %(3'7 87s%()r/rrrA _SpecialFormextend)rorprqrrrsresultr~s ` r derived_typesrs H  #z^4PF  uT6+>+>%>?   6G 6R    ,V__Y-GH    ,Y7   Mrc [SSSS4[SSSS4[SSSS4[SSSS4[SSSS4[ R SSSS4[S SSS4[S SSS4/n/nUHnUR[U65 M [U5$) NrTFSymIntrXrDr.Scalar ScalarTypeDevice) rrWrXrDr.r Numberrrrrdict)datarlines rget_supported_param_typesrs 4u- heT* ud+ vtUD) eUE5) xue4  eUE2 5%/ RDF mT*+ <rrzTensor[]rrXrDrcUcgU[RRLaU"S5 [R"U5nU[ La4U[ R5;aU"SUS[ S35 [ U$[R"U5nUH"nU[ ;dMU"SUS[ S35 M$ SSRUVs/sH n[ UPM sn5-S-$s snf) Nz()z7No return type annotation was provided. Please add one.zReturn has unsupported type r8r7r<r=r) r?rKrNr/r0r$SUPPORTED_RETURN_TYPESrRr1r\)rMrr2r3args rr[r[sW&&,,,JK   z *F U 388: : .zl;((>'?qB &j11 ??: &D , , .zl;((>'?qB  4H4C2374HI IC OOHs C* rdcUR[RR[RR4;$rx)rJr?rKPOSITIONAL_OR_KEYWORDrL)rds rrIrI$s6 :://&& r tuple_typec<[USS5nU[RLdU[Ld US:XdUc[$[ U5S:Xa [US$[ U5S:XaUS[ La [US$[[R[U5$)zu Convert `tuple_type` into a list type with the same type arguments. Assumes that `tuple_type` is typing.Tuple type. __args__Nrr)getattrr/Tupler$r}rTEllipsisr)r type_argss rrQrQ+s J5I fll"   ?   Y1 IaL!! Y1 1!9IaL!!FLLy!1233r)"r{r?r/r rrrrYrrrtorch.utils._exposed_inr _TestTensorCallabler.rnrArrDrrrzr}rWrXrrr[rOrKrIr$rQrrrrs? " ... ll  O " n-n- c] n-  n-n-b"T6.../""" "  "J& H KKL* 7& LL(P6237,,4d5k4d4j4r