+ j]>PlR-R.l4t7] !]PpPrP|Pv4]-!R/RaR 7R0R1l4t?]?PlR2R3l4t7] !]PpPrP~Pv4]-!R4RaR 7R5R6l4t@]@PlR7R8l4t7] !]PpPrPPv4]-!R9RaR 7R:R;l4tA]APlR<R=l4t7] !]PpPrPPv4]-!R>RaR 7R?R@l4tB]BPlRARBl4t7] !]PpPrPPv4]-!RCRaR 7RDREl4tC]CPlRFRGl4t7] !]PpPrPPv4]-!RHRaR 7RIRJl4tD]DPlRKRLl4t7] !]PpPrPPv4]-!RMRaR 7RNROl4tE]EPlRPRQl4t7] !]PpPrPPv4!RRRS4tF!RTRU]%4tG!RVRW]G4tH!RXRY]H4tI!RZR[]H4tJ]PP]I]PP]J/tN]^kR\R]ltO!R^R_]!4tPR#)bN)Callable)AnyOptional)ConstDictVariable) TupleVariable)has_side_effectProxy)graph_break_hints)create_call_function) TYPE_CHECKING unimplemented)CURRENT_STREAM_INDEXget_external_object_by_indexregister_graph_created_objectregister_user_objectreset_user_object_trackingCurrentStreamSource)GetSetMethodVariableTrackerConstantVariable)FxTracebackAnnotateVariable)LazyVariableTracker)InstructionTranslatorBase) PyCodegen) custom_opc<V^8dQhR\R\R\/#argskwargsreturn)rint)formats"`/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_dynamo/variables/streams.py __annotate__r(&s!SCCc \P!V/VBp\V\P \ .4\ /444#N)torchEventr EventVariable make_construct_in_graph_event_fnrr)r"r#events*, r' new_eventr1&sC KK ( (E ( 66 " 04  r)cRV^8dQhR\\,R\R\/#r )tuplerr%)r&s"r'r(r(0s%eCjCCr)c \P!V/VBp\V\P \ .4\ /444#r+)r,StreamrStreamVariable!make_construct_in_graph_stream_fnrr)r"r#streams*, r' new_streamr90sC \\4 *6 *F (88 " 04  r)c@V^8dQhR\PRRRR/#)r!devicecgrr$Nr,r;)r&s"r'r(r(:s%55ELL5k5d5r)caSPV3Rl4S!\V44SP\^R44R#)cl<SP\PPPR4#stash_graph_created_objectload_import_fromr,_dynamograph_bytecode_inputs__name__)r<sr')_codegen_current_stream..<s&## MM / / 8 8 ( r)FN) add_push_nullr extend_outputr )r;r<s&fr'_codegen_current_streamrK:s<  6"#)!U34r)cDV^8dQhR\PR\/#)r!r;r$)r,r;r%)r&s"r'r(r(Esu||r)cba\PPS4p\VV3Rl4#)c<\SV4#r+)rK)_r<r;s&&r'rG$get_current_stream..Hs 5fbAr))r, acceleratorcurrent_streamr)r;r8sf r'get_current_streamrSEs,    - -f 5F (A r)cDV^8dQhR\R\P/#r!indexr$)r%r,r5)r&s"r'r(r(Ls r)cx\V4p\V\P4'g\ RV 24hV#)z3Fork/join stream expected a stream object at index )r isinstancer,r5AssertionError)rVr8s& r'_get_stream_by_indexrZLs; )% 0F fell + +A% I   Mr)cDV^8dQhR\R\P/#rU)r%r,r-)r&s"r'r(r(Ussu{{r)cx\V4p\V\P4'g\ RV 24hV#)z4Record/wait event expected an event object at index )rrXr,r-rY)rVr0s& r'_get_event_by_indexr]Us; ( /E eU[[ ) )B5' J   Lr)z streams::fork) mutates_argsc4V^8dQhR\R\RR/#r! from_indexto_indexr$Nr%)r&s"r'r(r(_s+AAAA Ar)cV\PP\V44R#r+r,rQ set_streamrZrarbs&&r' fork_streamrh^s    !5h!?@r)c4V^8dQhR\R\RR/#r`rc)r&s"r'r(r(g&      r)cR#r+rgs&&r'rOrOf  r)z streams::joinc4V^8dQhR\R\RR/#r`rc)r&s"r'r(r(rs&AACA3A4Ar)cV\PP\V44R#r+rergs&&r' join_streamrpqs   !5h!?@r)c4V^8dQhR\R\RR/#r`rc)r&s"r'r(r(wrjr)cR#r+rlrgs&&r'rOrOvrmr)zstreams::record_eventc4V^8dQhR\R\RR/#r! event_index stream_indexr$Nrc)r&s"r'r(r(s!cr)cT\V4p\V4pVPV4R#r+)r]rZrecordrurvr0r8s&& r' record_eventrzs"  ,E !, /F LLr)c4V^8dQhR\R\RR/#rtrc)r&s"r'r(r(&      r)cR#r+rlrurvs&&r'rOrOrmr)zstreams::wait_eventc4V^8dQhR\R\RR/#rtrc)r&s"r'r(r(s!Cstr)cT\V4p\V4pVPV4R#r+)r]rZwaitrys&& r' wait_eventrs"  ,E !, /F JJvr)c4V^8dQhR\R\RR/#rtrc)r&s"r'r(r(r|r)cR#r+rlr~s&&r'rOrOrmr)zstreams::synchronize_eventc(V^8dQhR\RR/#r!rur$Nrc)r&s"r'r(r(s34r)c<\V4pVP4R#r+)r] synchronize)rur0s& r'synchronize_eventrs  ,E r)c(V^8dQhR\RR/#rrc)r&s"r'r(r(s  3 4 r)cR#r+rl)rus&r'rOrOr)zstreams::synchronize_devicec4V^8dQhR\R\RR/#r! device_type device_indexr$Nstrr%)r&s"r'r(r(s&KKCKsKtKr)cl\PP\P!W44R#r+)r,rQrr;rrs&&r'synchronize_devicers !!%,,{"IJr)c4V^8dQhR\R\RR/#rr)r&s"r'r(r(s!  3 c d r)cR#r+rlrs&&r'rOrOrr)zstreams::synchronize_streamc(V^8dQhR\RR/#r!rvr$Nrc)r&s"r'r(r(sSTr)c<\V4pVP4R#r+)rZr)rvr8s& r'synchronize_streamrs !, /F r)c(V^8dQhR\RR/#rrc)r&s"r'r(r(s  C D r)cR#r+rl)rvs&r'rOrOrr)zstreams::wait_streamc4V^8dQhR\R\RR/#)r!waiting_stream_indexwaited_on_stream_indexr$Nrc)r&s"r'r(r(s!##c#3#4#r)cT\V4p\V4pVPV4R#r+)rZ wait_stream)rrwaiting waited_ons&& r'rrs&"#78G$%;Cll M  Mr)cl\PPPP W4R#)auAn op which waits on an event and moves the last usage of to_dealloc after the wait, so that after the sync occurs, the deallocation or subsequent reuse of the tensor's memory will be guaranteed to happen after a side stream is finished using it. See https://docs.pytorch.org/docs/stable/generated/torch.Tensor.record_stream.html#torch.Tensor.record_stream for more detailsN)r,opsstreamsrdefaultrrrs&&&r' sync_deallocrs# II  (()9Lr)cTV^8dQhR\R\R\PRR/#rr)r&s"r'r(r(s4        r)cR#r+rlrs&&&r'rOrOs  r)zstreams::record_streamcHV^8dQhR\PR\RR/#r!tensorrvr$Nr,rr%)r&s"r'r(r(s%==%,,=c=d=r)c:VP\V44R#r+) record_streamrZrrvs&&r'rrs -l;r?_device_handle_attrrYr make_guard EQUALS_MATCHhasattrrrrgetattrr@)rrr>r?names&& r'_stream_device_handle_get(StreamVariable._stream_device_handle_gets 9'' < I  ;;; $++001J1JK L 4::t $ $#**74::t+DE E 4:: / /#**4::+C+CD Dr)c <V^8dQhRS[/#robject)r&rs"r'r(r.fr)cVP#r+rrs&r'get_real_python_backed_value+StreamVariable.get_real_python_backed_value zzr)cV<V^8dQhRRRS[S[,RS[S[S[3,RS[/#r!rrr"r#r$listrrr)r&rs"r'r(r.B-- '-?#-S/)* -  -r)cFV^,p\V\4'g\R\V4 24hVPP R\ PPPVPVP3/4\P!R4#)rzExpected EventVariable, got call_functionN) rXr.rYroutput create_proxyr,rrrrrr)rrr"r# event_args&&&& r'rStreamVariable.wait_events G )]33 #?Y?P!QR R   II   ( (  ( ($*@*@ A    &&t,,r)cV<V^8dQhRRRS[S[,RS[S[S[3,RS[/#rTrU)r&rs"r'r(r.rWr)cFV^,p\V\4'g\R\V4 24hVPP R\ PPPVPVP3/4\P!R4#)rzExpected StreamVariable, got rYN) rXr6rYrrZr[r,rrrrrr)rrr"r# other_streams&&&& r'rStreamVariable.wait_streams Aw ,77 #@lAS@T!UV V   II   ) )  # #\%C%C D    &&t,,r)cV<V^8dQhRRRS[S[,RS[S[S[3,RS[/#rTrU)r&rs"r'r(r.B - - ' -?# -S/)* -  -r)cVPPR\PPP VP 3/4\P!R4#rYN) rZr[r,rrrrrrrrr"r#s&&&&r'rStreamVariable.synchronizesN   II   0 0  # # %    &&t,,r)cV<V^8dQhRRRS[S[,RS[S[S[3,RS[/#rTrU)r&rs"r'r(r.B  ' ?# S/)*    r)c ^RIHp^RIHpV!\VVP P !RR.V!V.V,V4O5!R7#r!)proxy_args_kwargs)wrap_fx_proxy_cls call_methodquery) target_clsrr-utilsrlbuilderrmrrZr[rrr"r#rlrms&&&& r'roStreamVariable.queryI .. '))((w):D6D=&)Q  r)cV<V^8dQhRRRS[S[,RS[S[S[3,RS[/#rTrU)r&rs"r'r(r.sB# # '# ?## S/)* #  # r)c |^RIHpVPP\ VP 44V'd@\ V^,\4'd#V^,pVP pVPpMLVP P4p\V\P\.4\/444pVPPR\P P"PWpP3/4V!VVPPR\$V3/4R7#)) wrap_fx_proxyrY)rr-)rsrzrZ'check_event_record_after_input_mutationrrrXr.rrzrr/rrr[r,rrr)rrr"r#rz event_varr0rus&&&& r'rzStreamVariable.record_events + 99"TZZ.I JtAw 66QIOOE#55KJJ++-E7>>!"%'8'<K   II   * * 00 1   ))((,   r)rrrrorzc^RIHpHp^RIHp^RIHp\V\4'gVP!VR8H4#VP'd,V!VPPVP44VP'd,V!VPPVP44Wc,p\P!VV!VPVP44#)r!r=)cmp_name_to_op_mappingr__ne__)rAr>r?rrrconstantrrXr6rrrCrDrbuildr) rrotheropr>r?rrop_fns &&&& r'tp_richcompare_impl"StreamVariable.tp_richcompare_impls82.%00$**2>: : ;;; $++001J1JK L <<< %,,11,2K2KL M&*$$  $**ekk *  r)c <V^8dQhRS[/#rr )r&rs"r'r(r.+%r)cVP#r+r-rs&r'as_proxyStreamVariable.as_proxy+rRr)c <V^8dQhRS[/#rr)r&rs"r'r(r..sSr)cR#)ztorch._Crlrs&r' module_nameStreamVariable.module_name.sr)c <V^8dQhRS[/#rr)r&rs"r'r(r.1sr)cR#)r5rlrs&r'fn_nameStreamVariable.fn_name1sr)c"<V^8dQhRRRR/#r!codegenrr$Nrl)r&rs"r'r(r.4sNN;N4Nr)caVP'd \R4hVPe]SPV3Rl4SP SP VP44SP \^R44R#RVP 2pSPPPW P4pSP SPVRR74R#)z2Stream should not have a source during reconstructNcl<SP\PPPR4#rrBrsr'rG,StreamVariable.reconstruct..;&00MM77@@2r)F_stream_Tadd)rrYrrI append_outputcreate_load_constrJr r;rrZinstall_global_by_idrcreate_load_globalrrprefixrGs&f r' reconstructStreamVariable.reconstruct4s ;;; !UV V  ! ! -  ! !   ! !'";";D.__annotate__O& B Bc BK BD Br)c<aSPV3Rl4SPV3Rl4S!S4S!S4SP\^R44SP\^R44R#)cl<SP\PPPR4#r@rBrsr'rGNStreamVariable.make_construct_in_graph_stream_fn..fn..Q&00MM77@@0r)cl<SP\PPPR4#) build_streamrCr,rDrrrFrsr'rGrWs%00MM''00.r)FNrIrJr rVrr"r#s&fr'fn.fnOg  ! !   ! !  DM FO  ! !"6q%"@ A  ! !"6q%"@ Ar)rlr"r#rsff r'r70StreamVariable.make_construct_in_graph_stream_fnK B B" r)c@<V^8dQh/S[;R&S[R,;R&#)r!r9NrB)rr)r&rs"r'r(r.ts#& t* r))r;r-rrr+) rFrrrrr,r5r9rBrrrHrPrrrrorzr tp_methodsrrrrrrr'r7__annotate_func__rrr(r)s@@r'r6r6ts;,,M'+)),""$--"--" - -  "# # L fZ(vk*vk*|, J &NN(qr)r6ct]tRtRtRt]P PtRt R] !] PR4/t RtR#)CudaStreamVariableiczMRepresents torch.cuda.Stream, preserving device-specific type and attributes. cuda_streamNrl)rFrrrrr,rr5r9rBrr6rH tp_getsetrrlr)r'rrcs5WJJ%%M' vnFFMIr)rct]tRtRtRt]P PtRt R] !] PR4/t RtR#)XpuStreamVariableinzLRepresents torch.xpu.Stream, preserving device-specific type and attributes. sycl_queueNrl)rFrrrrr,xpur5r9rBrr6rHrrrlr)r'rrns5VII$$M& f^EEtLIr)rcTV^8dQhR\R\\,R,/#)r! stream_fnr$N)rLrr6)r&s"r'r(r(s$55543G$3N5r)c,\PV4#r+)_stream_fn_to_variable_clsget)rs&r'_get_stream_variable_clsrs % ) )) 44r)c taa]tRtRtoV3RlV3RlltV3RlRltV3RlRltV3RlR ltV3R lR ltV3R lR lt V3RlRlt V3RlRlt R] !]4R] !] 4R] !] 4R] !] 4/t V3RlV3RlltV3RlRlt]V3RlRl4t]V3RlRl4tV3RlRltR tVtV;t#)!r.ic X<V^8dQhRS[RS[PRS[R,RS[RR/#r,)r r,r-r%r)r&rs"r'r(EventVariable.__annotate__sE333{{3: 3  3  3r)c &<VejRVPP9dOVPPR,V8wd-\RVPPR, RV 24h\SV`!R/VBWnW nW0nR#)Nr0r1r2rl)r3r4rYrrr-rrr5s&&&&,r'rEventVariable.__init__s~  EJJOO!Czz/58$*5::???+K*LDQVPWX "6"  !2r)c0<V^8dQhRRRS[RS[RS[/#)r!rrrrr$)rr)r&rs"r'r(rs-77-76E7KN7 7r)c ^RIHpV!WW#4#)ry)object_richcompare)object_protocolr)rrrrrs&&&& r'r!EventVariable.tp_richcompare_impls 8!$E66r)c <V^8dQhRS[/#rr)r&rs"r'r(rs  T r)c,\VP4#r+)rrrs&r'rEventVariable.python_typesDJJr)c <V^8dQhRS[/#rrK)r&rs"r'r(rrMr)cVP#r+rOrs&r'rP*EventVariable.get_real_python_backed_valuerRr)cV<V^8dQhRRRS[S[,RS[S[S[3,RS[/#rTrU)r&rs"r'r(rsB-- '-?#-S/)* -  -r)c\PWV4wrEVPPR\P P PVPV3/4\P!R4#re) r._get_stream_argrZr[r,rrrrrr)rrr"r#rOrvs&&&& r'rEventVariable.waitsi (77&I   II   ( (&&    &&t,,r)cV<V^8dQhRRRS[S[,RS[S[S[3,RS[/#rTrU)r&rs"r'r(rsB-- '-?#-S/)* -  -r)cR\PWV4wrEVPP\ VP 44VPP R\PPPVPV3/4\P!R4#re)r.rrZr{rrr[r,rrrzrrr)rrr"r# stream_argrvs&&&& r'rxEventVariable.records $1#@#@6#R  99"Z=M=M:NO   II   * *&&    &&t,,r)cV<V^8dQhRRRS[S[,RS[S[S[3,RS[/#rTrU)r&rs"r'r(rrcr)cVPPR\PPP VP 3/4\P!R4#re) rZr[r,rrrrrrrfs&&&&r'rEventVariable.synchronizesN   II   / /  # # %    &&t,,r)cV<V^8dQhRRRS[S[,RS[S[S[3,RS[/#rTrU)r&rs"r'r(rrir)c ^RIHp^RIHpV!\VVP P !RR.V!V.V,V4O5!R7#rkrqrts&&&& r'roEventVariable.queryrvr)rrxrroc \<V^8dQhRRRS[RS[S[,RS[S[S[3,RS[/#)r!rrrGr"r#r$)rrVrr)r&rs"r'r(rsL  '  ?#  S/)*    r)c(<W P9d\SV` WW44#\VP4P R\VP4P RV 2p\R\V4RV R2.\POR7R#).zUnsupported event methodz#Dynamo doesn't support tracing the zC method. We currently support wait, record, synchronize, and query.)gb_typecontext explanationhintsN) rrrnrrrrr rr SUPPORTABLE)rrrGr"r# method_namer s&&&&& r'rnEventVariable.call_methods ?? "7&r> >DJJ**+1T$**-=-J-J,K1TF S  .I=k]KIJ"..  r)c <V^8dQhRS[/#rr)r&rs"r'r(rrr)cVP#r+rrs&r'rEventVariable.as_proxyrRr)c j<V^8dQhRRRS[S[,RS[S[S[3,RS[RS[3,/#)r!rrr"r#r$r6)rVrrrr3r%)r&rs"r'r(r sK88 '8?#8S/)*8 $ % 8r)c RpV'd V^,pMV'dVPR4pV'g(VPP4pWDP3#W3P3#)aReturns (stream_variable, stream_index_for_op). The ambient current stream is registered at index 0 in the external object registry. The inductor wrapper updates index 0 at runtime so that cudagraph capture sees the capture stream, not the stale trace-time default stream. Nr8)rrrr)rr"r#rrs&&& r'rEventVariable._get_stream_arg s[ aJ H-J11<<>J;;; ;7777r)cD<V^8dQhRS[RS[RS[S[R.R3,/#rr)r&rs"r'r(r#rr)caaRVV3RllpV#)c,V^8dQhR\RRRR/#rrc)r&s"r'r(DEventVariable.make_construct_in_graph_event_fn..__annotate__&rr)c<aSPV3Rl4SPV3Rl4S!S4S!S4SP\^R44SP\^R44R#)cl<SP\PPPR4#r@rBrsr'rGLEventVariable.make_construct_in_graph_event_fn..fn..(rr)cl<SP\PPPR4#) build_eventrrsr'rGr .s%00MM''00-r)FNrrs&fr'r:EventVariable.make_construct_in_graph_event_fn..fn&rr)rlrsff r'r/.EventVariable.make_construct_in_graph_event_fn"rr)c"<V^8dQhRRRR/#rrl)r&rs"r'r(r9sNN;N4Nr)caVP'd \R4hVPe]SPV3Rl4SP SP VP44SP \^R44R#RpSPPPW P4pSP SPVRR74R#)z1Event should not have a source during reconstructNcl<SP\PPPR4#rrBrsr'rG+EventVariable.reconstruct..@rr)F_eventTr) rrYrrIrrrJr rrZrrrrs&f r'rEventVariable.reconstruct9s ;;; !TU U  ! ! -  ! !   ! !'";";Dqr)rl)Rrrcollections.abcrtypingrrr,torch._dynamo.variables.dictsrtorch._dynamo.variables.listsrtorch.fxrr r bytecode_transformationr excr r rErrrrrrrbaserrrrr ctx_managerrlazyrtorch._dynamo.symbolic_convertrrrtorch._library.custom_opsrrr1r9rKrSrZr]rh register_fakerOrrforkrrpjoinrzrrrrrrrrrr6rrrrRrrrr.r()rs@r'r)sd$ ;7+ :.)11&4%H#/ 5 ?,A-A    !!&&../ ?,A-A    !!&&../ "45    !!..667 r23     !!,,445 'b9:    !  !!33;;< (r:K;K!! "  !!44<<= (r:; !! "  !!44<<= !3#4#     !!--556 "4 M5 M   !!..667 #"5=6=   !!//7786 6 r/7/dl*l^ JJ1 II/B 5GNOGNr)