+ j=!^RIt^RIHt^RIHt^RIHtHtRR3RRllt]P!]R R R 7t ]P!]R RR 7t RR lt RRlt ]!RR44t]!RR44tRtR#)N)deque) dataclass)AnyLiteralcVP#N)childrenxs&V/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/profiler/_utils.pyr s1::Fc$V^8dQhR\/#)reverse)bool)formats"r __annotate__rs***rc#"V'd\MRp\V!V44pV'd5V!V4pVxV!V!V44FpVPV4K K<R#5i)cV#rr s&r r _traverse.. sqrN)reversedrappend)treenext_fn children_fnrorder remaining curr_event child_events&&&& r _traverser"sVH[EeDk"I Y'  Z!89K   [ ): s *A#6A#c"VP4#r)popr s&r r r s aeegrT)rrc"VP4#r)popleftr s&r r r s rcVeV\V48d \V4p\W#4FpV!W,4'gKVu# R#r)lenrange)seq predicatestartendis&&&& r index_of_first_matchr/s@ {cSXo#h 5  SV  H rcV^8dQhRR/#)rreturnNr)rs"r rr%s  t rcl^RIHpV!4;_uu_4RRR4R# +'giR#;i)rprofileN)torch.autograd.profilerr4r3s r _init_for_cuda_graphsr6%s/   s" 3 c0a]tRt^,toRtV3RltRtVtR#) TimelineEventz-Represents an event in the profiler timeline.c<V^8dQh/S[;R&S[R,;R&S[R,R,;R&S[S[,R,;R&S[S[S[3,;R&#) r timestamp event_typeN marker_type identifierevent)r,r-regularfilenamenode)intrstrdictr)r __classdict__s"r rTimelineEvent.__annotate__,sgN  122  +,t33 c D   S>rrN)__name__ __module__ __qualname____firstlineno____doc____annotate_func____static_attributes____classdictcell__rFs@r r8r8,s7rr8c4a]tRt^7toRtRtV3RltRtVtR#)ContextStackEntryz5Represents a context (filename or node) in the stack.Nc<V^8dQh/S[R,;R&S[S[,;R&S[R,;R&S[R,;R&#)r context_typer=Nmetadatatidr@)rrDrCrE)rrFs"r rContextStackEntry.__annotate__7sK,--  c   Tk  trr) rHrIrJrKrLrVrMrNrOrPs@r rRrR7s? CrrRc ha^RIHpVPR.4p.oRpV3RlpVFpRV9gRV9dKV!V4'dGVR,^RpVPR4'd V!R We4KO\ V4pV!R XV4KgVR,pSP \VR R R V44K SPR R7.p SEFKp V P;R8XEdV Pf \R4hV PR 8Xd\V P\4'g,\R\V P4P 24hVPV P4p V P"PR4p V P \%R V PW44KV PR 8XdR p V P"PR4p \'V 4F5pVP(R 8XgKVP*V 8XgK)VP,p M V 'ddV PR/4pV PV9d>VV P,pV P \%R V PVV 44EKEKEKEK;R8Xd\/\1V 4^, RR4FYpV V,pV PVP(8XgK)V PVP8XgKFV P3V4EKN EKSR 8XgEK\R pR pV P"PR4p\'V 4FupVP*V8XgKVP(R 8XgK)VP,'gK=VP,PRR4pVP,PRR4pM V'g V'd=V P"P5R/4pV'dVVR&V'd VVR&EKHEKKEKN R # \ dELi;i)aF Maps recorded profiler events to their corresponding fx nodes and adds stack traces. Builds a timeline of all events (regular ops and FX markers for filenames/nodes), sorts by timestamp, then processes chronologically while maintaining a context stack of active filename/node scopes. Regular events are augmented with stack traces and node names from the innermost active context. Runtime is O(n log n) for n events. Args: traced_data: Json of profiler events from Chrome trace Returns: Dict mapping recorded event names to their aten operations with added stack traces )_FX_METADATA_REGISTRY traceEventscVPR4R8H;'dKVPRR4PR4;'d"VPRR4PR4#)catcpu_opnamez## z ##)get startswithendswith)r>s&r is_fx_marker_eventLmap_recorded_events_to_aten_ops_with_stack_trace..is_fx_marker_eventWs^ IIe  ( 6 6 &"%007 6 6 &"%..u5 rc <VR,pW2R,,pSP\VRWV44SP\VRWV44R#)tsdurr,r-N)rr8)r;r=r>start_tsend_tsevent_timelines&&& r append_fx_marker_eventPmap_recorded_events_to_aten_ops_with_stack_trace..append_fx_marker_event^sR;%L( (GZU K   &% G rrfrgr^z.pyrArBr?NcVP#r)r:r s&r r Bmap_recorded_events_to_aten_ops_with_stack_trace..~sakkr)keyr,z+identifier must not be None for start eventz0identifier must be str for filename marker, got rV node_metadatar- stack_tracezNo model stack trace availabler_args node_name)torch.fx.tracebackrYr`rbrC ValueErrorrr8sortr;r=AssertionErrorr< isinstancerDtyperHr>rRrrTrVrUr)r(r$ setdefault) traced_datarY trace_eventsrcrkr>content node_indexrh context_stacktimeline_eventrUrVcurrent_file_metadata ctx_entryrp node_metar.current_stack_tracecurrent_node_name event_tidrrrjs& @r 0map_recorded_events_to_aten_ops_with_stack_tracerAs9??="5L+-N   u U 2  e $ $FmAb)G&&&z7B!$WJ'vz5AT{H  ! !-)T4QV"W X',12.0M)''!,,4()VWW!--;%n&?&?EE,##'(A(A#B#K#K"LN 5889R9RSH(..2259C!(()&(A(A8 $//69,0)(..2259C%-m%< %22j@ )  44=4F4F1! &=-(=(A(A/SU(V )44 E5B . 9 96I*00 1$*N,E,EyRU!" F-:0s=1A5r2>A -a 0I&22i6L6LL*559M9MM%))!,?'+#$(!*0044U; !)-!8I }} 1$11V; @R@R@R2;2D2D2H2H -/O3/1:0B0B0F0Fvr0R-""9'*;)//::62FD*.A]+(,=[)) +rs!*>u*  4EtT    ,e      T>r