qh?/dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddlmZddlmZmZGddej&Zegd Zd ZGd d ZeeZd ZdZdZGddej<ZGddeZ GddeZ!e dZ"e dZ#e dZ$ddZ%d dZ&e ddZ'de!fdZ(dZ)ejTre)yy)!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)configutilscTeZdZdZej Zej Zy) EventStatuszStatus of an event. N)__name__ __module__ __qualname____doc__enumautoSTARTENDJ/var/www/html/engine/venv/lib/python3.12/site-packages/numba/core/event.pyr r 2s! DIIKE $))+Crr )znumba:compiler_lockz numba:compileznumba:llvm_locknumba:run_passcX|jdr|tvr|d}t||S)a>Guard 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_kindrBs7 x T%?==o KrceZdZdZd dZedZedZedZedZ edZ ed Z d Z e Z y) EventzAn event. Parameters ---------- kind : str status : EventStatus data : any; optional Additional data for the event. exc_details : 3-tuple; optional Same 3-tuple for ``__exit__``. Nclt||_||_||_||dd|_y||_yNr)r_kind_status_data _exc_details)selfrstatusdata exc_detailss r__init__zEvent.__init__ds@ &   + 3{1~7M""- rc|jS)zFEvent kind Returns ------- res : str )r"r&s rrz Event.kindlzzrc|jS)zPEvent status Returns ------- res : EventStatus )r#r,s rr'z Event.statusvs||rc|jS)zIEvent data Returns ------- res : object )r$r,s rr(z Event.datar-rc<|jtjk(S)zSIs it a *START* event? Returns ------- res : bool )r#r rr,s ris_startzEvent.is_starts||{0000rc<|jtjk(S)zRIs it an *END* event? Returns ------- res : bool )r#r rr,s ris_endz Event.is_ends||{..rc|jduS)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%r,s r is_failedzEvent.is_faileds  D((rc|j t|jjnd}d|jd|jd|dS)NNonezEvent(z, z, data: ))r(typer r"r#)r&r(s r__str__z Event.__str__sM99( ?//0.4  |2dll^8D6CCrNN)r r r rr*propertyrr'r(r1r3r5r:__repr__rrrrrXs /11// ) )D Hrrcnt|tsJt|}t|j |y)zvRegister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N) isinstanceListenerr _registeredappendrlisteners rregisterrEs1 h )) ) t DX&rcrt|tsJt|}t|}|j |y)zxUnregister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N)r?r@rrAremove)rrDlsts r unregisterrIs4 h )) ) t D d CJJxrcVt|jD]}|j|y)zeBroadcast an event to all registered listeners. Parameters ---------- event : Event N)rArnotify)eventrDs r broadcastrMs(  +rc^eZdZdZej dZej dZdZy)r@z(Base class for all event listeners. cy)zkCalled when there is a *START* event. Parameters ---------- event : Event Nrr&rLs ron_startzListener.on_start rcy)ziCalled when there is a *END* event. Parameters ---------- event : Event NrrPs ron_endzListener.on_endrRrc|jr|j|y|jr|j|yt d)zpNotify this Listener with the given Event. Parameters ---------- event : Event unreachableN)r1rQr3rTAssertionErrorrPs rrKzListener.notifys5 >> MM% \\ KK  / /rN) r r r rabcabstractmethodrQrTrKrrrr@r@s>      0rr@cBeZdZdZdZdZdZedZedZ y)TimingListenerzA listener that measures the total time spent between *START* and *END* events during the time this listener is active. cd|_yr!)_depthr,s rr*zTimingListener.__init__s  rcj|jdk(rt|_|xjdz c_y)Nr)r]timer_tsrPs rrQzTimingListener.on_start s% ;;! wDH q rc|xjdzc_|jdk(r-t|dd}t|jz |z|_yy)Nr_r _duration)r]getattrr`rarc)r&rLlasts rrTzTimingListener.on_endsF q ;;! 4a0D#g0D8DN rct|dS)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. rc)hasattrr,s rdonezTimingListener.donest[))rc|jS)zReturns the measured duration. This may raise ``AttributeError``. Users can use ``.done`` to check that a measurement has been made. )rcr,s rdurationzTimingListener.durations~~rN) r r r rr*rQrTr<rhrjrrrr[r[s> 9 **rr[c"eZdZdZdZdZdZy)RecordingListeneraA 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. cg|_yN)bufferr,s rr*zRecordingListener.__init__.s  rcb|jjtj|fyrnrorBtimerPs rrQzRecordingListener.on_start1 DIIK/0rcb|jjtj|fyrnrqrPs rrTzRecordingListener.on_end4rsrN)r r r rr*rQrTrrrrlrl(s 11rrlc#hKt|| |t||y#t||wxYww)aInstall 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)rErIrCs rinstall_listenerrv8s0$ T8#4" 4"s 2! 2/2c#Kt}t||5|ddd|jr||jyy#1swY)xYww)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)r[rvrhrj)rcallbacktls r install_timerrzQsM*  B $ # wwsAA(AA Ac#jKt}t||5|dddy#1swYyxYww)anInstall 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)rlrv)rrls rinstall_recorderr}ns3&  B $ #s3' 303cRt|tj|}t|y)zTrigger the start of an event of *kind* with *data*. Parameters ---------- kind : str Event kind. data : any; optional Extra event data. )rr'r(N)rr rrM)rr(evts r start_eventrs T+"3"3$ ?C cNrcTt|tj||}t|y)aTrigger 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)rr rrM)rr(r)rs r end_eventrs$  +//+ CcNrc#Kt5}|jfd}tddddy#1swYyxYww)a;A 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"t|y)N)r(r))r)r)r(rs ron_exitztrigger_event..on_exits d; ?r)r(N)rpushr)rr(scopers`` r trigger_eventrsK   @  @ Dt$ s A '? A AA rDc Btj}tj}g}|jD]e\}}|j }t |j}|dz}|jrdnd} |d} |} t||||| | | } |j| g|S)zGPrepare events in `listener` for serializing as chrome trace data. i@BBEname)catpidtidtsphrargs) osgetpid threading get_native_idror(strrr1dictrB) rDrrevsrrecr(r ts_scaledrrrevs r_prepare_chrome_trace_datars ))+C  ! ! #C C?? Cxx#((mN LLScF| #)  2  Jrcttdtjtjfd}y)z\Setup a RecordingListener and an exit handler to write the captured events to file. rct}td5}tj||tj dddy#1swYyxYw)Nw)cls)ropenjsondumpr_LazyJSONEncoder)routfilenamerDs r_write_chrome_tracez=_setup_chrome_trace_exit_handler.._write_chrome_tracesF)2 (C  rs(!F )0#$$)),XXv$  ' #0sww#0L#X#L 1 1 ##08. $*):0 <  $&r