Ei16SSKrSSKrSSKrSSKJr SSKJrJr SSKrSSKJrJ r J r Jr SSK J r SSK JrJr \Rr\ "S5SS .S \R"S \\S \4S jj5rS\\\R*4S\S\S\S\4 SjrSr\ S\R2\ S\\ S\S\S\S\R:S0rSr\"5r S\RBS \4Sjr"S\\#S \\4Sjr$g) N) GenericAlias)OptionalUnion)devicedtypeTensortypes) exposed_in) _OPAQUE_TYPESis_opaque_typez torch.library)op_nameprototype_functionrreturnc  ^^^^^^^SnURmSm[R"U5mU4SjmS[4UUU4SjjmS[[ [ [[4S4S[[[RS4[44U4S jjmS [ [ [[4S[[R[44UU4S jjm/n[5nS n[TRR55GHunup[U 5(dT"S 5 U R [R"R$:XaU(dUR'S5 SnU R([R"R*La T"SUS35 T"U R(5upSn U [,;a[/U 5(a[0U R2n OU [4R6R8:Xa T"SUS35 O[;U S5(abU R<[LaO[?U 5n SnU [,;aSU S3nT"SUSU R(SUS[,RA5S3 5 O8T"SUSU R(S[,RA5S35 O [,U n U c[CSU35e[ U5[La;X:wa [ES5eU RGS5(aSUS U [IS5S3n O;X;a6U RGS5(d T"SUS!35 SUS U [IS5S3n URKU5 U RL[R"R*LaUR'U S"U35 GMSnU RLb*[OU RL[P[R[45(a[U RL5nO[OU RL[[4RT45(aS#U RLS#3nO[OU RL[4RV5(aL[U RL5nS$nURGU5(d[CS%U<S&35eU[IU5SnO"T"SUS'[ U RL5S(35 UR'U S"US)U35 GM X:wa([U5U- n[IU5S*:a T"US+35 T"TRX5unn [[UT5nUbUS,S-R]U5S.U3$S,S-R]U5S.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 X/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/_library/infer_schema.pyerror_fninfer_schema..error_fnFs /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_stringIs< Y? ?  ..??RS  s ))tys.rc>/nSnUH"nT"U5up4URU5 X$-nM$ U(a [U5S4$US4$)NFT)appendtuple)r$reschangedty ty_changedunstringify_types runstringify_types'infer_schema..unstringify_typesQsVB-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_changedr"r-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's type cannot be inferred from the schema as it is a ScriptObject. Please manually specify the schema using the `schema=` kwarg with the actual type of the ScriptObject. __origin__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: zschema_type is None for param 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 dtype repr z must start with 'torch.'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 signaturer1r'rtypeobjectr2Anyboolset enumerate parametersitemssupported_paramkind Parameter KEYWORD_ONLYr& annotationemptySUPPORTED_PARAM_TYPESr r class_nametorch_C ScriptObjecthasattrr: tuple_to_listkeysAssertionErrorr startswithlenadddefaultr0intfloatrrreturn_annotation parse_returnjoin)r mutates_argsrUNKNOWN_MUTATESparams seen_argssaw_kwarg_only_argidxnameparamr_ schema_type list_typeexample_type_str default_repr dtype_repr torch_dotmutates_args_not_seenrbretr"rr r!rr,r-s @@@@@@@r infer_schemarvsVT 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 7o..+O<GG  EHH$9$99 'ZZ 66#..%7)/: #)  55)6yk'G$ &??!"5==/3 EMM5;;77 / $ !,,Y77(%j^3LM *#i.*:;  &MdSXS`S`NaMbcPQ MM[M4&,@ Aw@x& #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)r2SequenceListr collectionsabclist)r~s rderived_seq_types(derived_types..derived_seq_typessF OOC KK  11C6 : v &   rc30># UH oTS34v M g7f)z[]N.0seq_typrxs r derived_types..s 6R7 "o &6Rsc32># UH nUTS34v M g7f)z?[]Nrrs rrrs' I #& 'Hsc3T># UHn[RUTS34v M g7f)z[]?N)r2rrs rrrs- 7__W %(3'7 87s%()r2rrrE _SpecialFormextend)rwrxryrzr{resultrs ` 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[RR5(a5S S K J n UR[R "["U5SSSS45 /nUHnUR%['U65 M [)U5$) NrTFSymIntrarHr1Scalar ScalarTypeDevicer) GroupName)rr`rarHr1r NumberrrrU distributed is_available"torch.distributed.distributed_c10drr&r2castrErrdict)datarrlines rget_supported_param_typesrs 4u- heT* ud+ vtUD) eUE5) xue4  eUE2 5%/ RD %%''@ V[[y15%NO F mT*+ <rrzTensor[]rrarHrcUcgU[RRLaU"S5 [R"U5nU[ La&U[ ;aU"SUS[ S35 [ U$[R"U5nUH"nU[ ;dMU"SUS[ S35 M$ SRUVs/sH n[ UPM sn5n[U5S:XaSU-S -nSU-S -$s snf) Nz()z7No return type annotation was provided. Please add one.zReturn has unsupported type r<r;rAr r@r) rCrOrRr2r3r'SUPPORTED_RETURN_TYPESr4rdr])rQrr5r6arg output_tys rrcrc%sW&&,,,JK   z *F U 3 3 .zl;((>'?qB  &j11 ??: &D , , .zl;((>'?qB   $G$31#6$GHI 4yA~)Oc) ?S  Hs<C5rlcUR[RR[RR4;$r)rNrCrOPOSITIONAL_OR_KEYWORDrP)rls rrMrMHs6 :://&& 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 r)getattrr2Tupler'rr]Ellipsisr)r type_argss rrYrYOs J5I fll"   ?   Y1 IaL!! Y1 1!9IaL!!FLLy!1233r)%rrCr2r rrrrUrrrtorch.utils._exposed_inr opaque_objectr r _TestTensorCallabler1rvrErrHrrrrr`rarrrcrSrOrMr'rYrrrrsB " ...8 ll  O " C-C- c] C-  C-C-L$T6.../$$$ $  $N2 H KKL* 7& LL(!@237,,4d5k4d4j4r