ȅi_5SSKrSSKrSSKrSSKJr 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 ce[ U5[La;X:wa [CS5eU RES5(aSUSU [GS5S3n O;X;a6U RES5(d T"SUS 35 SUSU [GS5S3n URIU5 U RJ[R"R*LaUR'U S!U35 GMSnU RJb*[MU RJ[N[P[45(a[U RJ5nO[MU RJ[[4RR45(aS"U RJS"3nO[MU RJ[4RT5(a>[U RJ5nS#nUREU5(deU[GU5SnO"T"SUS$[ U RJ5S%35 UR'U S!US&U35 GM X:wa([U5U- n[GU5S':a T"US(35 T"TRV5unn [YUT5nUbUS)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 U/home/james-whalen/.local/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: 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 signaturer1r'rtypeobjectr2Anyboolset enumerate parametersitemssupported_paramkind Parameter KEYWORD_ONLYr& annotationemptySUPPORTED_PARAM_TYPESr r class_nametorch_C ScriptObjecthasattrr: tuple_to_listkeysr 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_seenraretr"rr r!rr,r-s @@@@@@@r infer_schemarus3T 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 / $ !,,Y7777)#i.*:;  &MdSXS`S`NaMbcPQ MM[M4&,@ Ao@p& #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_typrws 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)rvrwrxryrzresultrs ` 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$) NrTFSymIntr`rHr1Scalar ScalarTypeDevicer) GroupName)rr_r`rHr1r 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[]rr`rHrcUcgU[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_TYPESr4rcr\)rQrr5r6arg output_tys rrbrb!sW&&,,,JK   z *F U 3 3 .zl;((>'?qB  &j11 ??: &D , , .zl;((>'?qB   $G$31#6$GHI 4yA~)Oc) ?S  Hs<C5rkcUR[RR[RR4;$r)rNrCrOPOSITIONAL_OR_KEYWORDrP)rks rrMrMDs6 :://&& 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 rrYrYKs J5I fll"   ?   Y1 IaL!! Y1 1!9IaL!!FLLy!1233r)%rrCr2r rrrrUrrrtorch.utils._exposed_inr opaque_objectr r _TestTensorCallabler1rurErrHrrrrr_r`rrrbrSrOrMr'rYrrrrsB " ...8 ll  O " -- c] -  --D$T6.../$$$ $  $N2 H KKL* 7& LL(!@237,,4d5k4d4j4r