+ "ݳj2Rt^RIHtHt^RIHt^RIHt^RIH t ]!RR7!RR44t ]!RR7!R R 44t !R R ] 4t ]!RR7!R R44t !RR4t!RR] 4t!RR4tR#)aSSynchronous execution-boundary extensions, independent of persistence. Boundary objects are borrowed for the duration of on_boundary only. An extension must quiesce before treating results as complete. Retaining a boundary or using it from a worker thread is an error. No tensor serialization or checkpoint format belongs in this module. )Futurewait) dataclass) get_ident)ProtocolT)frozenc0a]tRt^toRtV3RltRtVtR#)LoopStepz?An execution identity, separate from transformer-layer indices.c><V^8dQh/S[;R&S[;R&S[;R&#)kindindexname)strint)format __classdict__s"[/home/luca/screen-checking/.venv/lib/python3.14/site-packages/gptqmodel/looper/extension.py __annotate__LoopStep.__annotate__s) I  J  I N)__name__ __module__ __qualname____firstlineno____doc____annotate_func____static_attributes____classdictcell__rs@rr r sIrr c8a]tRt^toRtRtV3RltRtVtR#)LoopPlanc\VP4\Jg7\\ VP44\VP48wd \ R4hR#)z=execution plan requires an immutable sequence of unique stepsN)typestepstuplelenset ValueErrorselfs&r __post_init__LoopPlan.__post_init__sE  5 (CDJJ,@C O,SO -TrcFVPPV4^,#))r%r )r+steps&&rafterLoopPlan.after#szz%))rc:<V^8dQh/S[S[R3,;R&#)r .r%)r&r )rrs"rrLoopPlan.__annotate__s 3 rrN) rrrrr,r1rrrr s@rr"r"s *rr"cZa]tRt^'toRtV3RlRltV3RlRltV3RlRltRtVt R #) LoopExecutionzBPlacement identity and scheduler continuation owned by the looper.cF<V^8dQhRS[S[S[S[,3,/#r return)dictrlist)rrs"rrLoopExecution.__annotate__*sAAS$s)^(<ArcR#Nrr*s&rexecution_device_pools$LoopExecution.execution_device_pools*crc <V^8dQhRS[/#r8r:)rrs"rrr<,s//d/rcR#r>rr*s&rexecution_state_dict"LoopExecution.execution_state_dict,sCrc$<V^8dQhRS[RR/#)r stater9NrC)rrs"rrr<.sAAtAArcR#r>r)r+rHs&&rload_execution_state_dict'LoopExecution.load_execution_state_dict.rArrN) rrrrrr?rErJrrr s@rr6r6'sLAA//AArr6c0a]tRt^1toRtV3RltRtVtR#) LoopContextNcd<V^8dQh/S[;R&S[;R&S[;R&S[;R&S[R,;R&#)r planmodel processors shared_stateN execution)r"objectr&r:r6)rrs"rrLoopContext.__annotate__1sG N M    t#* rr)rrrrrSrrrr s@rrMrM1s '+I rrMcDa]tRt^:toV3RlRltV3RlRltRtVtR#) LoopBoundaryc:<V^8dQhRS[RS[S[R3,/#)r r0futures.)r r&r)rrs"rrLoopBoundary.__annotate__;s$Xfck0BrcJWnW n\4VnRVnR#)TN)r0_futuresr_owner_active)r+r0rYs&&&r__init__LoopBoundary.__init__;s  k  rc4<V^8dQhRS[S[R3,/#)r r9.)r&rT)rrs"rrrZAs B Bvs{+ Brc R\4VP8wd \R4hVP'g \R4hVP'd\ VP4\ ;QJd.RVP4FNK 5#!RVP44#)zDrain all preceding finalizers, then return results or raise. Waiting for every future before raising prevents one failed worker from leaving other workers mutating state during recovery. z2Loop boundaries belong to the orchestration threadz!Loop boundary is no longer activec3@"TFqP4xK R#5ir>)result).0futures& r 'LoopBoundary.quiesce..MsA=]]__=s)rr] RuntimeErrorr^r\rr&r*s&rquiesceLoopBoundary.quiesceAsr ;$++ %ST T|||BC C ===  uA4==AuAuA4==AAAr)r^r\r]r0N)rrrrr_rjrrr s@rrWrW:s B BrrWc2a]tRt^PtoV3RlRltRtVtR#) LoopExtensionc$<V^8dQhRS[RR/#)r boundaryr9N)rW)rrs"rrLoopExtension.__annotate__Qs  L T rc R#)zBObserve a step before the next step can mutate continuation state.Nr)r+ros&&r on_boundaryLoopExtension.on_boundaryQs rrN)rrrrrrrrr s@rrmrmPs  rrmc\a]tRt^VtoRtR RltRtV3RlRltR V3RlRlltRt Vt R #) LoopExtensionszFOwn extension dispatch and outstanding work on the coordinator thread.c4\V4Vn.VnR#r>)r& _extensions_pending)r+ extensionss&&rr_LoopExtensions.__init__Ys ,&( rc,\VP4#r>)boolrwr*s&r__bool__LoopExtensions.__bool__]sD$$%%rc&<V^8dQhRS[RS[/#)r contextr9)rMr)rrs"rrLoopExtensions.__annotate__`s,,[,S,rc .pVPFp\VRR4pVfKV!V4pVfK$\V4\Jg/^Tu;8:d%\ VP P 48:gM\R4hVPV4K \ \V44^8d \R4hV'd V^,#^#)zALet extensions install continuation before any planned step runs.on_startNz.extension returned an invalid execution cursorz1extensions returned conflicting execution cursors) rwgetattrr$rr'rOr%r)appendr()r+rcursors extensioncallbackcursors&& rstartLoopExtensions.start`s))Iy*d;H#!'*%F|3.a67S **F7)LNN6** s7| q PQ Q$wqz+!+rc$<V^8dQhRS[RR/#)r r0r9N)r )rrs"rrrss  H T rcV'gR#VPPV4\V\VP44pVPFpVP V4K RVnVPUu.uFEpVP4'd+VP4'gVP4fKCVNKG upVnR# RTni;iuupi)NF) rxextendrWr&rwrrr^done cancelled exception)r+r0rYrorrfs&&& rpublishLoopExtensions.publishss  W%eDMM&:; %!-- %%h/. %H  -- ';;==F$4$4$6$6&:J:J:L F'   %H  s$$C?C C 0C C  C)rwrxN)r) rrrrrr_r}rrrrr s@rruruVs+P)&,,&   rruN)rconcurrent.futuresrr dataclassesr threadingrtypingrr r"r6rMrWrmrurrrrs,! $ $ * * *BHB $+++BB, H - - r