3j%SSKJr SSKrSSKr\R"\5r0rS\S'SSjr SSjr SSjr SSjr SS jr SS jrSS jrSS jrSS jrSSjrg)) annotationsNz2dict[torch._ops.OpOverload, torch._ops.OpOverload]_manual_out_variant_registrycU[U'g)z0Register a functional op -> out variant mapping.N)r) functional_opout_ops U/home/wildlama/miniconda3/lib/python3.13/site-packages/torch/_library/_out_variant.pyregister_out_variantr s 39 /c,[RU5$)z;Return the manually registered out variant for op, or None.)rget)ops rlookup_manual_out_variantrs ( + +B //r cURRS5SnURS5(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 r !_is_functional..'s<+;C<<+;s)namesplitendswithany arguments)schemaop_names r_is_functionalr!sK kk%b)G<6+;+;<< <> % A#..*A*AAr cURVs/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)rr$lenziprstrtype default_value)schema_aschema_brnon_mutable_args_anon_mutable_args_babs r_signatures_matchr1.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)r&returnsr'r# before_set)r mutable_argsretrs r_has_valid_out_variant_returnsr7As  6>>a 6>>c,// 5 >> !S^^%; >> $ $(A(A A 6 r cURnURVs/sH!n[U5(dMURPM# sn$s snf)z5Get the names of out arguments for an out variant op.)_schemarr$r)rrrs rget_out_arg_namesr:Us8 ^^F & 0 0 I 0OC4HHCHH 0 II Is AAcURn[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)r9r! RuntimeError namespacerrgetattrtorchops overloadsTag out_varianttagsr1rr$r&r3r7) r rr>r torch_packet overload_name candidatecandidate_schemarr5s rto_out_variantrJ[sSZZF & ! !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.)rJ_get_out_variants_infoAssertionError)rexpected_out_opr tagged_infos rcheck_out_variantrPsM *F ~,]; 1-A3 (j)  +   m_Kx@+,-E F  !r cURnURRRS5Sn[ [ [ R U5U5n/nUR5HYn[ X55n[ RRUR;dM8URSUSUR35 M[ SRU5$)zJCollect information about overloads tagged with out_variant for debugging.rr<z - z:  ) r>r9rrr?r@rArBrCrDrEappendjoin)rr>r rFoverloads_inforGrHs rrLrLs''I##((..t4Q7G7599i8'BL "N%//1 L8 99 INN 2  ! !Dr):K:K9L"M N 2 99^ $$r )rtorch._ops.OpOverloadrrVreturnNone)r rVrWztorch._ops.OpOverload | None)rtorch._C.FunctionSchemarWbool)rztorch._C.ArgumentrWrZ)r+rYr,rYrWrZ)rrYr5zlist[torch._C.Argument]rWrZ)rrVrWz list[str])rrVrNrVrWrX)rWr() __future__rloggingr@ getLogger__name__logr__annotations__r rr!r$r1r7r:rJrPrLr rrbs" ! TVPU9(9 !9 900!0=B%% & #) (J .b ( ;P   6 %r