EiZ%SSKrSSKrSSKrSSKrSSKrSSKrSSKJr SSKJrJ r J r J r J r SSK r SSKJr SSKJrJr SSKJr \ (aSSKJr /S Qr\"S S 9"S S \ 55r0r\\\\R84\S'Sr\"5\\ R>'"SS5r \ RB\ RD\ RF\RH\ RJRL\RN\ RP\ "5\)"S5\ RT\RV"S5\S. r,\-"\5Hr.\/"\\.5\,\.'M SSS\4Sjr0S\ RbRdS\R84Sjr30r4\\5\6\64\R84\S'S\ RbRdS\R84Sjr7\"S S 9S\S\5SS\\6S 44S!j5r8\"S S 9S3S"\S#\94S$jj5r:\"S S 9S%5r;\"S S 9S&\S'\4S(j5r<\"S S 9S4S\S\5\S)4S\ \\6\4S*\ \5\S+\ \\6\4S,\9S\4S-jj5r=\"S S 9S4S\S\5\S)4S\ \\6\4S*\ \5\S+\ \\6\4S,\9S\ \4S.jj5r>\"S S 9S5S/\ R~RS\6S\5\S\ \\6\4S,\9S\ \4 S0jj5rAS1\R8S\5\S)4S\\6\4S,\9S\ \4 S2jrBg)6N)Callable)Anycast NamedTupleOptional TYPE_CHECKING)boolean_dispatched) OpOverloadOpOverloadPacket) compatibility)Argument)ArgsKwargsPaircheck_for_mutable_operationget_signature_for_torch_opcreate_type_hint type_matchesnormalize_functionnormalize_moduleF)is_backward_compatiblecB\rSrSr%Sr\\S4\S'\\ \4\S'Sr g)r z4 Simple named tuple for wrapping args/kwargs pairs. .argskwargsN) __name__ __module__ __qualname____firstlineno____doc__tupler__annotations__dictstr__static_attributes__rV/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/fx/operator_schemas.pyrr s% S/ cNr&r_manual_overridesc/nSnUR[R"U55 S[4SjnUR[R"U55 U$)NcgNr)selfs r'nonzero!_nonzero_schemas..nonzero0 r&as_tuplecgr+r)r,r0s r'r-r.5r/r&)appendinspect signaturebool) signaturesr-s r'_nonzero_schemasr7-sTJ g''01 4 g''01 r&c\rSrSrSrSrg)_FakeGlobalNamespace@c0US:Xa[$[S5e)Ntorchz!Expected a torch namespace lookup)r< RuntimeError)r,names r' __getattr__ _FakeGlobalNamespace.__getattr__As 7?L>??r&rN)rrrrr?r%rr&r'r9r9@s@r&r9t) TensorDeviceLayoutnumberFuture AnyEnumTypeQScheme __torch__NoneTypeStoragerAPyObjectts_typeztorch._C.JitTypereturnc6[UR[5$)z Convert a TorchScript type to a Python type (including subtypes) via eval'ing the annotation_str. _type_eval_globals sets up expressions like "List" and "Future" to map to actual types (typing.List and jit.Future) )evalannotation_str_type_eval_globals)rMs r' _torchscript_type_to_python_typerSYs &&(: ;;r& ts_schemac SSKJn /nURGHXn[UR5nUR 5(a UR O URnURS:wa UROSnUR(a URO URnUS:XaXqR:wa[SU35eURn[U5HwupU RUR:wa%[SU RSU R35eU"U RURU R U R"S 9X('My UR%U"XgXTS 95 GM[ UR&V s/sHn [U R5PM n n [)U 5S:XaSn O [)U 5S :XaU Sn O [+U 5n [R,"X,S 9$s sn f) Nr) Parameterr,inputfromz$Expected POSITIONAL_OR_KEYWORD, got z)Expected POSITIONAL_OR_KEYWORD for param z, got )r>kinddefault annotationr )return_annotation)r3rV argumentsrStypehas_default_value default_valueemptyr> kwarg_only KEYWORD_ONLYPOSITIONAL_OR_KEYWORDAssertionErrorPOSITIONAL_ONLY enumeraterYrZr[r2returnslenr! Signature) rTrV parametersargarg_typerZr>rYidxpret return_types return_types r'%_torchscript_schema_to_signature_implrsbs""$J""3CHH='*'<'<'>'>###IOO 88v-sxx7~~  " "00 6>666$'KD6%RSS,,D$J/66Y<<<(CAFF86RSRXRXQYZ#,"22II || #  0  4G Q G#N?H>O>O>Os(2>O <A \ a "1o L)   Z GGsG'_SCHEMA_TO_SIGNATURE_CACHEcURUR4n[RU5nUbU$[ U5nU[U'U$r+)r> overload_namertgetrs)rT cache_key cache_valress r' _torchscript_schema_to_signaturer{sN 7 77I*..y9I / :C,/y) Jr&targetr)r.rrc8[USS9up4U(avU(an/n[X45H*upgUR"U0UD6 URXg45 M, Sn[ U5S:Xag[ U5S:XaUSupU"U 5 gggg![a Mpf=f)NT)return_schemascDUR(a[SUS35eg)Nz!Tried to trace mutable operation z|. FX only supports functional code, so operations that mutate operands in-place (e.g. via `out` arguments) are not supported) is_mutabler=)schemas r'throw_if_mutable5check_for_mutable_operation..throw_if_mutables/  "7x@()!r&rr )rzipbindr2 TypeErrorri) r|rrr6schemasmatched_schemascandidate_signaturerr_schema_to_checks r'rrs5VDQJg ,/z+C '  #(($9&9&&(;'DE,D   1 $   !Q &!0!3 A _ - ?z  s$B  BBopr~c:[U[5(aUR/nO[U[5(a4UR 5Vs/sHn[ X5RPM nnO[ RU5nU(a U(aUS4$S$[RRRU5nUc U(aS$S$[RRU5nUVs/sHn[U5PM nnU(aXr4$U$s snfs snf)aX Given an operator on the `torch` namespace, return a list of `inspect.Signature` objects corresponding to the overloads of that op.. May return `None` if a signature could not be retrieved. Args: op (Callable): An operator on the `torch` namespace to look up a signature for Returns: Optional[List[inspect.Signature]]: A list of signatures for the overloads of this operator, or None if the operator signatures could not be retrieved. If return_schemas=True, returns a tuple containing the optional Python signatures and the optional TorchScript Function signature N)NN) isinstancer _schemar overloadsgetattrr(rwr<jit _builtins _find_builtin_C_jit_get_schemas_for_operatorr{)rr~roverloadoverrideaten_fnrr6s r'rrs "j!!::, B( ) )ACPX72(00P$((, '5Hd# ?4 ?))%%33B7 ?#1< ;t ;((88AIPQv26:JQ$2J B BQRs D0Dc[U[[45(a[U[5(aSnOSn[U5S:Xa U"[5$USnUH5n[ X25(aM[ X#5(aUnM)U"[5s $ U"U5$U$![ a [R"SU35 U$f=f)a Produces a type hint for the given argument. The :func:`create_type_hint` looks for a type hint compatible with the input argument `x`. If `x` is a `list` or `tuple`, it looks for an object in the list whose type is a superclass of the rest, and uses that as `base_type` for the `List` or `Tuple` to be returned. If no such object is found, it defaults to `List[Any]`. If `x` is neither a `list` nor a `tuple`, it returns `x`. c[U$r+)listxs r'ret_type"create_type_hint..ret_type s 7Nr&c[US4$)N.)r!rs r'rrs C=(r&rz@We were not able to successfully create type hint from the type ) rrr!rir issubclass Exceptionwarningswarn)rr base_typerAs r'rrs a$ ' '!T""# )1v{}$!Ia++ -- !I#C=( I& &- (8 H   Nqc R  H  sAB=B B#CCsignature_type argument_typec^^[USU5nUTLagU[RLa'UT:wa!URn[ U4SjU55$[USS5[ LaURSmT[ La T[ Lag[R"T5(d[R"SUS35 g[TSS5[ La[TRST5$U4SjnU"T5$U[ LaT[RLagU[RLaT[ [ 1;ag[R"T5(a'[R"U5(a [TU5$g) N __origin__Tc3<># UHn[UT5v M g7fr+)r).0crs r' type_matches../sI=a<=11=srz.Does not support nested parametric types, got z. Please file a bug.Fc>[USS5[LagURnURS:Xag[U4SjU55$)NrF)rTc3X># UHo[L=(d [UT5v M! g7fr+)Ellipsisr)rr sig_el_types r'r=type_matches..is_homogeneous_tuple..Fs"XiX D*Q *DDis'*)rr!__args__all)rA containedrs r'is_homogeneous_tuple*type_matches..is_homogeneous_tuple@s@q,-U: IzzU"XiXX Xr&)rtypingUnionranyrintr3isclassrrrr<dtypenumbersNumberfloat)rrsig_origin_type sig_containedrrs ` @r'rr$sKnlNKO&&,,&>]+J&// I=III~|T2d:$--a0  C K3$6{++ MM@@PPde  =, 5 =m44Q7E E Y$M22%++!='Mc5\,I}%%'//.*I*I-88 r&. arg_types kwarg_typesnormalize_to_only_use_kwargscN[XX#XE5nUc[SUSUSU35eU$)zV Wrapper around normalize_function that never returns None, but loudly errors instead zFailed to normalize function z with args z and kwargs )rr=)r|rrrrrrzs r'_normalize_function_or_errorrWsH f C {+F8;tfLQWPX Y   r&clUc0nSn[U[R5(d8[U[[45(d[ US5(a UR n[U[R5(d[U[[45(dUnU[;aw[U[5(a [S5e[UnUSUSp[R"U 5R[R"U 5R:wagU n[U5(d[S[U535e[R"[R"U55n [!XX%5nU$[U5(d[S[U535e[#U5n /n U (GaYU H'nUR$"U0UD6 U R'U5 M) [+U 5S:XaU$[+U 5S :Xa[!U SXU5nU$UcUbU(aUO[-[.[0S 5nU(aUO0nU HnS nUR$"U0UD6nUR2R55H4unnURUnU=(a [7UR8U5nM6 U(dMs[!UUUU5n U$ U$S R;SU 55n[=S[>R@"U5SU35eU$![(a GMgf=f![(a S nN~f=f)a Returns normalized arguments to PyTorch functions. This means that `args/kwargs` will be matched up to the functional's signature and return exclusively kwargs in positional order if `normalize_to_only_use_kwargs` is True. Also populates default values. Does not support positional-only parameters or varargs parameters (*args, **kwargs). Does not support modules. May require `arg_types` and `kwarg_types` in order to disambiguate overloads. Args: target (Callable): Function that we are normalizing args (Tuple[Any]): Tuple of args to the function kwargs (Optional[Dict[str, Any]]): Dict of kwargs to the function arg_types (Optional[Tuple[Any]]): Tuple of arg types for the args kwarg_types (Optional[Dict[str, Any]]): Dict of arg types for the kwargs normalize_to_only_use_kwargs (bool): Whether to normalize to only use kwargs. Returns: Returns normalized_args_and_kwargs, or `None` if not successful. N_opz"target should not be a string hereif_trueif_falsez*target_for_analysis must be callable, got ztarget must be callable, got rr rTF c38# UHn[U5v M g7fr+)r$)rrs r'r%normalize_function..s12AF /sz Tried to normalize arguments to zy but the schema match was ambiguous! Please provide argument types to the normalize_arguments() call. Available schemas: )!rtypesBuiltinFunctionTyper r hasattrrr r$rer3r4rkcallabler^unwrap&_args_kwargs_to_normalized_args_kwargsrrr2rrirr!rr]itemsrr[joinr=r<typename)r|rrrrrnew_args_and_kwargstarget_for_analysis dispatchedrrsigtorch_op_schemasrr sig_matches bound_typesarg_namermparamschema_printoutss r'rrosr>~ vu88 9 9F%5z$BCC FE " "  fe77 8 86,j9::$ ' ' &#&&$%IJJ+F3J *9 5z*7MX!!'*55$$X.99:") +,, ?(8?#q(Z Y_%*&L#A&6R'#T M(K,C-6 DsR>&;Q:RSOO_N`b e!4 )0*/K0s%=#LA$L$ L! L!$ L32L3rootcURU5n[URS5(aURR n[ [RUS5UR:XaG[R"[R"UR55nUc0n[XX45n U $g![an[SUS35UeSnAff=f)a Returns normalized arguments to PyTorch modules. This means that `args/kwargs` will be matched up to the functional's signature and return exclusively kwargs in positional order if `normalize_to_only_use_kwargs` is True. Also populates default values. Does not support positional-only parameters or varargs parameters (*args, **kwargs). Args: root (nn.Module): root module upon which we query modules target (Callable): Function that we are normalizing args (Tuple[Any]): Tuple of args to the function kwargs (Optional[Dict[str, Any]]): Dict of kwargs to the function normalize_to_only_use_kwargs (bool): Whether to normalize to only use kwargs. Returns: Returns normalized_args_and_kwargs, or `None` if not successful. z$Tried to normalize node with target z# but root did not have that target!Nr) get_submoduleAttributeErrorr=r __class__rrr<nnr3r4rforwardr) rr|rrrsubmode classnamerrs r'rrs6##F+ v,,$$-- 588Y -1A1A A##GNN6>>$BCC~"H6# ' &  26(;  !  sB55 C?CCrcd^ [RR[RR1m [ U 4SjUR R 555(a*[UR R55/SQ:wagUR"U0UD6nUR5 0n/n[UR 5HLupxU(d/U[U5:a URURU5 M;URUXX'MN [[!U5U5$)aU Given a call target, args, and kwargs, return the arguments normalized into an ArgsKwargsPair, or None if the type signature is not supported by this normalization. Args: sig (inspect.Signature): Signature object for the target args (Tuple): Arguments that appear at the callsite for `target` kwargs (Dict): Keyword arguments that appear at the callsite for `target` normalize_to_only_use_kwargs (bool): Whether to normalize to only use kwargs. Returns: Optional[ArgsKwargsPair]: Normalized args and kwargs for `target`, or `None` if this target is not supported. c3@># UHoRT;v M g7fr+)rY)rrosupported_parameter_typess r'r9_args_kwargs_to_normalized_args_kwargs..Hs Tr s_  $AA 23) e,Z->@4$w'8'8"99:?  $4#5%-- @@llllllnnii99}}%'T ##    VA#FA.q <.@