Eig#SSKrSSKJr SSKJr SSKJrJrJr SSKJ r J r J r J r SSK Jr SSKJr "S S \5r\R"S S 55rS \ R&S\S\4SjrS\S\4SjrSrSrg)N)Callable) dataclass)AnyOptionalProtocol)_C_opsautogradTensor)_pytree)utilsc6\rSrSr%\\\S'\\\S'Srg) InfoProtocol _backward_fn_setup_context_fnN__name__ __module__ __qualname____firstlineno__rr__annotations____static_attributes__rT/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/_library/autograd.pyrr s8$$))rrc6\rSrSr%\\\S'\\\S'Srg)InforrrNrrrrrrs8$$))rropinforeturnc^^^^^ ^ STRSTRSTR3n[R"TR 5m [ "SS55mU4Sjm U UU4SjnUU4Sjn[U[R4[U5[U5S.5mTR n[S /URQURQ755(a [T5mUUU 4S jnU$) NGeneratedBackwardFor__cH\rSrSr%\R \S'\\\ 4\S'Sr g)$make_autograd_impl..Metadatakeysetkeyword_only_argsrN) rrrrrDispatchKeySetrdictstrrrrrrMetadatar(s!!!S>)rr/c>USnUSSn[R"5 URnURnTR"U[R -/UQ70UD6nUsSSS5 $!,(df  g=f)N)r_AutoDispatchBelowAutogradr*r+ redispatch_after_autograd_keyset)argsmetadatar*kwargsresultr!s rforward_no_grad+make_autograd_impl..forward_no_grad"sj8CRy  * * ,__F//F]]6B,E,E#EWWPVWF - , ,s AA,, A:c>USnUSSn[R"5 URnURnTR"U[R -/UQ70UD6nTR (aK[R"TRX5upT(aTR XXES9 OTR XUS9 UsSSS5 $!,(df  g=f)Nr1)ctxinputskeyword_only_inputsoutput)r<r=r?) rr2r*r+r3r4rr fill_defaults_schema) r<r5r6r*r7r8has_kwarg_only_argsr"r!s rforward#make_autograd_impl..forward,s8CRy  * * ,__F//F]]6B,E,E#EWWPVWF%% %222::tL &**&+**s*O/- , ,s BC Cc>TR(aZURnURSSUlTR"U/UQ76nX l[U[5(a/UQSP7$US4$[ STS35e!WUlf=f)Nr1zTrying to backward through zQ but no autograd formula was registered. Please use register_autograd to add one.)rneeds_input_grad isinstancetuple RuntimeError)r<gradsprev_needs_input_gradr8r"r!s rbackward$make_autograd_impl..backwardIs    =(+(<(<%'*';';CR'@$**377'<$&%((&&4< )".7 8  (=$s 3A<< B)rCrLc3b# UH%n[R"UR5v M' g7fN)ris_tensorlist_like_typetype).0as r %make_autograd_impl..ds) 5A %%aff--5s-/c>[R"5(a4[R"U6(aTR"/UQT"X5P76nU$T"/UQT"X5P76nU$rO)ris_grad_enabled_any_requires_gradapply)r*r5r+r8 Generatedr/r9s r autograd_impl)make_autograd_impl..autograd_implls_    B$9$94$@__PdPHV,OPF %PdPHV,OPF r) _namespace_opname _overloadnamerrBrArrQr Function staticmethodany argumentsreturnssupports_tensorlist) r!r"namerCrLschemar[rZr/r9rBs `` @@@@rmake_autograd_implrhs' a |1REUEUDVWD33BJJ?***: "   #G,$X. IZZF  56##5fnn5 ( 2  rclsc^^^^URmURmURm["SS55mUU4SjnU4SjnUU4SjnXlX lX0lU$)a$Allows a given autograd.Function class to support List[Tensor] inputs/outputs. Regular autograd.Function has a constraint that it only directly supports autograd for Tensors. Applying @supports_tensorlist enables an autograd.Function to support autograd for List[Tensor] inputs and outputs. cp\rSrSr%\R \S'Sr\\R \S'Sr \\ \S'Sr g)%supports_tensorlist..Metadata input_specN output_specresult_is_tupler) rrrrr TreeSpecrrorrpboolrrrrr/rls1$$$26 Xg../6*.$.rr/c>USnUSSn[UT5(d [S5e[R"[ U5UR 5nT"U/UQ76n[U[ 5UlUR(dU4n[R"U[5upEXRl [US5(a [S5eX l [ U5$)Nr1zNYI: calling supports_tensorlist autograd.Function.forward directly. You should probably be calling .apply instead. Please file an issue if not. _pt_metadataz@Please don't set ctx._pt_metadata; PyTorch uses it to store info)rGNotImplementedErrorr tree_unflattenlistrnrHrp tree_flattennot_list_of_tensorrohasattrrIrt)r<r5r6r8 flat_resultror/ orig_forwards r new_forward(supports_tensorlist..new_forwards8CRy(H--%/  %%d4j(2E2EFc)D)#-fe#< ''YF#*#7#7@R#S * 3 ' 'R $[!!rcL>[US5(d [S5eURn[R"[ U5UR 5nURn[R"[ URSS5UR5UlT"U/UQ76nX0l[U[5(dU4n[R"U[5upVXbR:wa[SUSURS35e[US/-5$!X0lf=f)NrtzNYI: calling supports_tensorlist autograd.Function.backward directly. This will automatically get called by PyTorch autograd. Please file an issue if you need this.r1zRExpected the return from backward to be of the same structure as the inputs. Got: z (return from backward), z (inputs))rzrurtr rvrwrorFrnrGrHrxnot_list_of_optional_tensorrI)r<rJr6rK grad_inputsflat_grad_inputsgrad_inputs_spec orig_backwards r new_backward)supports_tensorlist..new_backwards0sN++%9  ##&&tE{H4H4HI!$ 4 4 9#*#9#9S))#2./1D1D$C (4e4K#8 +u--&.K .5-A-A 4. * 22 2''7&88Q&&'y2  %.//%$9 s !ADD#c>[R"U[S9upT"U5nT"/UQUP76nURc [ S5e[R "[ U5UR5nUR(dW[U[5(d[ S[U535e[U5S:wa[ S[U535eUS$U$)N)is_leafz%metadata.output_spec must not be Nonezresult must be tuple, got r z%result tuple must have length 1, got r) r rxryroAssertionErrorrvrwrprGrHrQlen)r5 flat_argsrnr6r8r/ orig_applys r new_apply&supports_tensorlist..new_applys ' 4 4TCU V J'1Y11    ' !HI I''V h6J6JK''fe,,$'A$v,%PQQ6{a$;CK=I!9  r)rCrLrYr)rir}rrr/rrr|s @@@@rrerevs];;LLLMJ/// "2*0X"KLI Jrc~[U[5(ag[U[5(a[SU55$g)NFc3L# UHn[U[5(+v M g7frOrGr rRls rrT%not_list_of_tensor..s;dz!V,,,ds"$TrGrHrwrbtrees rryrys3$$;d;;; rc~[U[5(ag[U[5(a[SU55$g)NFc3`# UH$oSL=(a [U[5(+v M& g7frOrrs rrT.not_list_of_optional_tensor..s$M1D=>Av)>%>>s,.Trrs rrrs3$$MMMM r) dataclassescollections.abcrrtypingrrrtorchrr r r torch.utilsr rrr OpOverloadrhreryrrrrrs$!**,,*8*  *** [4??[,[8[|jSjSjZr