sh?/SrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKJ r SSK J r J r SSKJr SSKJrJr "SS\R&5r\"/S Q5rS r"S S 5r\"\5rS rSrSr"SS\R<5r"SS\5r "SS\5r!\ S5r"\ S5r#\ S5r$SSjr%S Sjr&\ SSj5r'S\!4Sjr(Sr)\RT(a\)"5 gg)!ag The ``numba.core.event`` module provides a simple event system for applications to register callbacks to listen to specific compiler events. The following events are built in: - ``"numba:compile"`` is broadcast when a dispatcher is compiling. Events of this kind have ``data`` defined to be a ``dict`` with the following key-values: - ``"dispatcher"``: the dispatcher object that is compiling. - ``"args"``: the argument types. - ``"return_type"``: the return type. - ``"numba:compiler_lock"`` is broadcast when the internal compiler-lock is acquired. This is mostly used internally to measure time spent with the lock acquired. - ``"numba:llvm_lock"`` is broadcast when the internal LLVM-lock is acquired. This is used internally to measure time spent with the lock acquired. - ``"numba:run_pass"`` is broadcast when a compiler pass is running. - ``"name"``: pass name. - ``"qualname"``: qualified name of the function being compiled. - ``"module"``: module name of the function being compiled. - ``"flags"``: compilation flags. - ``"args"``: argument types. - ``"return_type"`` return type. Applications can register callbacks that are listening for specific events using ``register(kind: str, listener: Listener)``, where ``listener`` is an instance of ``Listener`` that defines custom actions on occurrence of the specific event. N) default_timer)contextmanager ExitStack) defaultdict)configutilsc\\rSrSrSr\R "5r\R "5rSr g) EventStatus2zStatus of an event. N) __name__ __module__ __qualname____firstlineno____doc__enumautoSTARTEND__static_attributes__r d/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/core/event.pyr r 2s IIKE ))+Crr )znumba:compiler_lockz numba:compileznumba:llvm_locknumba:run_passcfURS5(aU[;aUS3n[U5eU$)aGuard to ensure that an event kind is valid. All event kinds with a "numba:" prefix must be defined in the pre-defined ``numba.core.event._builtin_kinds``. Custom event kinds are allowed by not using the above prefix. Parameters ---------- kind : str Return ------ res : str znumba:zG is not a valid event kind, it starts with the reserved prefix 'numba:') startswith_builtin_kinds ValueError)kindmsgs r _guard_kindr Bs: x  T%?==o Krc\rSrSrSrS Sjr\S5r\S5r\S5r \S5r \S 5r \S 5r S r \ rS rg)EventXzAn event. Parameters ---------- kind : str status : EventStatus data : any; optional Additional data for the event. exc_details : 3-tuple; optional Same 3-tuple for ``__exit__``. Ncl[U5UlX lX0lUbUScSUlgUUlgNr)r _kind_status_data _exc_details)selfrstatusdata exc_detailss r__init__Event.__init__ds< &   + 3{1~7M""- rcUR$)z&Event kind Returns ------- res : str )r&r*s rr Event.kindlzzrcUR$)z0Event status Returns ------- res : EventStatus )r'r1s rr+ Event.statusvs||rcUR$)z)Event data Returns ------- res : object )r(r1s rr, Event.datar3rc<UR[R:H$)z3Is it a *START* event? Returns ------- res : bool )r'r rr1s ris_startEvent.is_starts||{0000rc<UR[R:H$)z2Is it an *END* event? Returns ------- res : bool )r'r rr1s ris_end Event.is_ends||{..rcURSL$)zIs the event carrying an exception? This is used for *END* event. This method will never return ``True`` in a *START* event. Returns ------- res : bool N)r)r1s r is_failedEvent.is_faileds  D((rcURb [UR5ROSnSURSURSUS3$)NNonezEvent(z, z, data: ))r,typerr&r')r*r,s r__str__ Event.__str__sM99( ?//0.4  |2dll^8D6CCr)r(r)r&r'NN)r rrrrr.propertyrr+r,r9r<r?rE__repr__rr rrr"r"Xs /11// ) )D Hrr"cx[U[5(de[U5n[UR U5 g)zbRegister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N) isinstanceListenerr _registeredappendrlisteners rregisterrQs4 h ) )) ) t DX&rc|[U[5(de[U5n[UnUR U5 g)zdUnregister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N)rKrLr rMremove)rrPlsts r unregisterrUs7 h ) )) ) t D d CJJxrcZ[URHnURU5 M g)zUBroadcast an event to all registered listeners. Parameters ---------- event : Event N)rMrnotify)eventrPs r broadcastrYs#  +,rcf\rSrSrSr\R S5r\R S5rSr Sr g)rLz(Base class for all event listeners. cg)zKCalled when there is a *START* event. Parameters ---------- event : Event Nr r*rXs ron_startListener.on_start rcg)zICalled when there is a *END* event. Parameters ---------- event : Event Nr r]s ron_endListener.on_endr`rcUR(aURU5 gUR(aURU5 g[ S5e)zPNotify this Listener with the given Event. Parameters ---------- event : Event unreachableN)r9r^r<rbAssertionErrorr]s rrWListener.notifys5 >> MM% \\ KK  / /rr N) r rrrrabcabstractmethodr^rbrWrr rrrLrLs>      0rrLcJ\rSrSrSrSrSrSr\S5r \S5r Sr g ) TimingListenerizxA listener that measures the total time spent between *START* and *END* events during the time this listener is active. cSUlgr%)_depthr1s rr.TimingListener.__init__s  rclURS:Xa[5UlU=RS- slg)Nr)rmtimer_tsr]s rr^TimingListener.on_start s% ;;! wDH q rcU=RS-slURS:Xa-[USS5n[5UR- U-Ulgg)Nrpr _duration)rmgetattrrqrrru)r*rXlasts rrbTimingListener.on_endsF q ;;! 4a0D#g0D8DN rc[US5$)zReturns a ``bool`` indicating whether a measurement has been made. When this returns ``False``, the matching event has never fired. If and only if this returns ``True``, ``.duration`` can be read without error. ru)hasattrr1s rdoneTimingListener.donest[))rcUR$)zReturns the measured duration. This may raise ``AttributeError``. Users can use ``.done`` to check that a measurement has been made. )rur1s rdurationTimingListener.durations~~r)rmrurrN) r rrrrr.r^rbrHr{r~rr rrrkrks> 9 **rrkc*\rSrSrSrSrSrSrSrg)RecordingListeneri(zA listener that records all events and stores them in the ``.buffer`` attribute as a list of 2-tuple ``(float, Event)``, where the first element is the time the event occurred as returned by ``time.time()`` and the second element is the event. c/UlgNbufferr1s rr.RecordingListener.__init__.s  rcdURR[R"5U45 grrrNtimer]s rr^RecordingListener.on_start1 DIIK/0rcdURR[R"5U45 grrr]s rrbRecordingListener.on_end4rrrN) r rrrrr.r^rbrr rrrr(s 11rrc#b# [X5 Uv [X5 g![X5 f=f7f)aUInstall a listener for event "kind" temporarily within the duration of the context. Returns ------- res : Listener The *listener* provided. Examples -------- >>> with install_listener("numba:compile", listener): >>> some_code() # listener will be active here. >>> other_code() # listener will be unregistered by this point. N)rQrUrOs rinstall_listenerr8s)$ T#4" 4"s / / ,/c## [5n[X5 Uv SSS5 UR(aU"UR5 gg!,(df  N3=f7f)aInstall a TimingListener temporarily to measure the duration of an event. If the context completes successfully, the *callback* function is executed. The *callback* function is expected to take a float argument for the duration in seconds. Returns ------- res : TimingListener Examples -------- This is equivalent to: >>> with install_listener(kind, TimingListener()) as res: >>> ... N)rkrr{r~)rcallbacktls r install_timerrQsC*  B $ # $ ww $ #sAA -A AAc#r# [5n[X5 Uv SSS5 g!,(df  g=f7f)aBInstall a RecordingListener temporarily to record all events. Once the context is closed, users can use ``RecordingListener.buffer`` to access the recorded events. Returns ------- res : RecordingListener Examples -------- This is equivalent to: >>> with install_listener(kind, RecordingListener()) as res: >>> ... N)rr)rrls rinstall_recorderrns'&  B $ # $ # #s7& 7 47cL[U[RUS9n[U5 g)zTrigger the start of an event of *kind* with *data*. Parameters ---------- kind : str Event kind. data : any; optional Extra event data. )rr+r,N)r"r rrY)rr,evts r start_eventrs T+"3"3$ ?C cNrcL[U[RXS9n[U5 g)zTrigger the end of an event of *kind*, *exc_details*. Parameters ---------- kind : str Event kind. data : any; optional Extra event data. exc_details : 3-tuple; optional Same 3-tuple for ``__exit__``. Or, ``None`` if no error. )rr+r,r-N)r"r rrY)rr,r-rs r end_eventrs"  +// CcNrc#^^# [5nURUU4Sj5n[TTS9 Sv SSS5 g!,(df  g=f7f)aA context manager to trigger the start and end events of *kind* with *data*. The start event is triggered when entering the context. The end event is triggered when exiting the context. Parameters ---------- kind : str Event kind. data : any; optional Extra event data. c>[TTUS9 g)N)r,r-)r)r-r,rs ron_exittrigger_event..on_exits d; ?r)r,N)rpushr)rr,scopers`` r trigger_eventrs?   @  @ Dt$ s A%= A A ArPc F[R"5n[R"5n/nURHdupEUR n[ UR5nUS-nUR(aSOSn USn Un [XqX(XU S9n URU 5 Mf U$)zGPrepare events in `listener` for serializing as chrome trace data. i@BBEname)catpidtidtsphrargs) osgetpid threading get_native_idrr,strrr9dictrN) rPrrevsrrecr,r ts_scaledrrrevs r_prepare_chrome_trace_datars ))+C  ! ! #C C??xx#((mN LLScF| #  2# Jrc^^[5m[ST5 [Rm[RUU4Sj5ng)zTSetup a RecordingListener and an exit handler to write the captured events to file. rc>[T5n[TS5n[R"X[R S9 SSS5 g!,(df  g=f)Nw)cls)ropenjsondumpr_LazyJSONEncoder)routfilenamerPs r_write_chrome_trace=_setup_chrome_trace_exit_handler.._write_chrome_traces;)2 (C C IIcE$:$: ;! s $A AN)rrQr CHROME_TRACEatexit)rrrPs @@r _setup_chrome_trace_exit_handlerrs9!"H x(""H __<rs(!F )0#$$)),XXv$  ' #0sww#0L#X#L 1 1 ##08. $*):0 <  $&r