NjeSSKJr SSKrSSKr\R"\5rS SjrS SjrS Sjr S Sjr SSjr SSjr SS jr g)) annotationsNcURRS5SnURS5(ag[SUR55(+$)zX A schema is functional if no argument is written to and the name doesn't end with '_'. ::_Fc38# UHoRv M g7fN)is_write).0args l/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/torch/_library/_out_variant.py !_is_functional..s<+;C<<+;s)namesplitendswithany arguments)schemaop_names r _is_functionalr sK kk%b)G<6+;+;<< <<cXURSL=(a URR$r ) alias_infor )r s r _is_mutable_argrs >> % A#..*A*AArcURVs/sHn[U5(aMUPM nnURVs/sHn[U5(aMUPM nn[U5[U5:wag[X45HkupVURUR:wa g[ UR 5[ UR 5:wa gURUR:wdMk g gs snfs snf)zOCompare two schemas by their non-mutable arguments (name, type, default value).FT)rrlenziprstrtype default_value)schema_aschema_br non_mutable_args_anon_mutable_args_babs r _signatures_matchr(s *2););X);#?SVCW#);X)1););X);#?SVCW#);X #&8"99&; 66QVV  qvv;#aff+ % ??aoo - < YXsC1C1C6C6cT[UR5S:Xag[UR5[U5:wag[URU5HQup#URb URc gURRURR:wdMQ g g)zFOut variant must return either nothing or the mutable args themselves.rTF)rreturnsrr before_set)r mutable_argsretr s r _has_valid_out_variant_returnsr.-s  6>>a 6>>c,// 5 >> !S^^%; >> $ $(A(A A 6 rcURn[U5(d[SUSU35eURnURR S5Sn[ [ [RU5U5nUR5Hn[ XE5n[RRUR;aM8URn[X5(dMVURVs/sHn[U5(dMUPM n n[!U 5[!UR"5:waM[%Xy5(d[SUSU35eUs $ gs snf)zM Given a functional operator overload, return its corresponding out variant. z#Failed to find out variant for op 'z&' as its schema is not functional. rz Out variant z^ has invalid returns. Expected either no returns or returns that alias the mutable args, got: N)_schemar RuntimeError namespacerrgetattrtorchops overloadsTag out_varianttagsr(rrrr*r.) oprr3r torch_packet overload_name candidatecandidate_schemar r,s r to_out_variantr@AsSZZF & ! !1"6    Ikk%a(G7599i8'BL&//1 L8  99  6 $,, :: ,55 5C9MC5   | FNN 3 3 -.>MMyk*()+  528  s 0EEc [U5nUc%[U5n[SUSU=(d S35eX!:wa[SUSUSUS35eg) z Checks that to_out_variant returns the expected out variant for a functional op. Raises AssertionError if the out variant is not valid. Nz#We did not find an out variant for a. Some common mistakes include: 1. The out variant is missing the torch.Tag.out_variant tag. 2. The out variant is not an overload of the original op (e.g., 'op.out' or 'op.overload_out') 3. The out variant's input arguments does not match the functional op's signature (excluding the mutable args). 4. The original operator is not functional. Overloads tagged with out_variant: z (none)zto_out_variant(z ) returned z, but expected z8. The out variant name does not match the functional op.)r@_get_out_variants_infoAssertionError) functional_opexpected_out_opout_op tagged_infos r check_out_variantrHrsM *F ~,]; 1-A3 (j)  +   m_Kx@+,-E F  !rcURnURRRS5Sn[ [ [ R U5U5n/nUR5HYn[ X55n[ RRUR;dM8URSUSUR35 M[ SRU5$)zJCollect information about overloads tagged with out_variant for debugging.rr0z - z:  ) r3r1rrr4r5r6r7r8r9r:appendjoin)rDr3rr<overloads_infor=r>s r rBrBs''I##((..t4Q7G7599i8'BL "N%//1 L8 99 INN 2  ! !Dr):K:K9L"M N 2 99^ $$r)rtorch._C.FunctionSchemareturnbool)r ztorch._C.ArgumentrOrP)r"rNr#rNrOrP)rrNr,zlist[torch._C.Argument]rOrP)r;torch._ops.OpOverloadrOztorch._ops.OpOverload | None)rDrQrErQrONone)rOr) __future__rloggingr5 getLogger__name__logrrr(r.r@rHrBrr rYs" !=B%% & #) (.b ( ;P   6 %r