+ ja0t$^RIHt^RIt^RIt]P !]4t/tR] R&RRlt RRlt R R lt R R lt R RltRRltRRltRRltRRltRRltR#)) annotationsNz2dict[torch._ops.OpOverload, torch._ops.OpOverload]_manual_out_variant_registryc$V^8dQhRRRRRR/#) functional_optorch._ops.OpOverloadout_opreturnNone)formats"\/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_library/_out_variant.py __annotate__rs&99(9 !9 9cV\V&R#)z0Register a functional op -> out variant mapping.N)r)rr s&&rregister_out_variantrs 39 /rc V^8dQhRRRR/#roprr ztorch._ops.OpOverload | Noner )r s"rrrs000!0rc,\PV4#)z;Return the manually registered out variant for op, or None.)rget)rs&rlookup_manual_out_variantrs ( + +B //rc V^8dQhRRRR/#)rschematorch._C.FunctionSchemar boolr )r s"rrrs==2=t=rc&VPPR4R,pVPR4'dR#\;QJd0RVP4F'gK R'*# R'*#!RVP44'*#)zX A schema is functional if no argument is written to and the name doesn't end with '_'. ::_Fc38"TFqPxK R#5iN)is_write).0args& r !_is_functional..'s<+;C<<+;sT)namesplitendswithany arguments)rop_names& r_is_functionalr.so kk%b)Gs<6+;+;> % A A#..*A*AArc$V^8dQhRRRRRR/#)rschema_arschema_br rr )r s"rrr.s&%% rcVPUu.uFp\V4'dKVNK ppVPUu.uFp\V4'dKVNK pp\V4\V48wdR#\W44FqwrVVPVP8wdR#\ VP 4\ VP 48wdR#VPVP8wgKpR# R#uupiuupi)zOCompare two schemas by their non-mutable arguments (name, type, default value).FT)r,r2lenzipr(strtype default_value)r4r5r$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 - < YXsC:C:C?C?c$V^8dQhRRRRRR/#)rrr mutable_argszlist[torch._C.Argument]r rr )r s"rrrAs& #) rch\VP4^8XdR#\VP4\V48wdR#\VPV4FVwr#VPeVPfR#VPPVPP8wgKUR# R#)zFOut variant must return either nothing or the mutable args themselves.TF)r7returnsr8r1 before_set)rrBretr$s&& r_has_valid_out_variant_returnsrGAs  6>>a 6>>c,// 5 >> !S^^%; >> $ $(A(A A 6 rc V^8dQhRRRR/#)rr rr z list[str]r )r s"rrrUsJJ3J JrcVPpVPUu.uF"p\V4'gKVPNK$ up#uupi)z5Get the names of out arguments for an out variant op.)_schemar,r2r()r rr$s& rget_out_arg_namesrKUs8 ^^F & 0 0 I 0OC4HHCHH 0 II Is AAc V^8dQhRRRR/#rr )r s"rrr[s,,,,1M,rcVPp\V4'g\RV RV 24hVPpVPP R4^,p\ \ \PV4V4pVP4Fp\PPVP9dK.VPp\W4'gKMVPUu.uFp\V4'gKVNK pp\!V4\!VP"48wdK\%Wh4'g\RV RV 24hVu# R#uupi)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)rJr. RuntimeError namespacer(r)getattrtorchops op_overloadsTagouttagsr@r,r2r7rDrG) rrrOr- torch_packet candidatecandidate_schemar$rBs & rto_out_variantrZ[sGZZF & ! !1"6    Ikk%a(G7599i8'BL"..0 99==  . $,, :: ,55 5C9MCC5   | FNN 3 3 -.>MMyk*()+  114  s ,EEc$V^8dQhRRRRRR/#)rrrexpected_out_opr r r )r s"rrrs$  ( ;P   rc \V4pVf'\V4p\RT RT;'gR 24hW!8wd\RV RV RV R24hR#) 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 at. Some common mistakes include: 1. The out variant is missing the torch.Tag.out 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: z (none)zto_out_variant(z ) returned z, but expected z8. The out variant name does not match the functional op.)rZ_get_out_variants_infoAssertionError)rr\r tagged_infos&& rcheck_out_variantrasM *F ~,]; 1-A+ ((j)  +   m_Kx@+,-E F  !rcV^8dQhRR/#)rr r9r )r s"rrrs % %S %rcVPpVPPPR4^,p\ \ \ P V4V4p.pVP4FZp\ W54p\ PPVP9gK9VPRV RVP 24K\ RPV4#)zBCollect information about overloads tagged with out for debugging.rz - z:  ) rOrJr(r)rPrQrR overloadsrTrUrVappendjoin)rrOr-rWoverloads_info overload_namerXs& rr^r^s''I##((..t4Q7G7599i8'BL "N%//1 L8 99==INN *  ! !Dr):K:K9L"M N 2 99^ $$r)__conditional_annotations__ __future__rloggingrQ getLogger__name__logr__annotations__rrr.r2r@rGrKrZrar^)rjs@rrqsi"" ! TVPU90=B&(J ,^ 6 %r