xKg/dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddlmZddlmZGddej$Zegd Zd ZGd d ZeeZd ZdZdZGddej:ZGddeZGddeZ e dZ!e dZ"e dZ#ddZ$d dZ%e ddZ&de fdZ'dZ(ejRre(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)configcTeZdZdZej Zej Zy) EventStatuszStatus of an event. N)__name__ __module__ __qualname____doc__enumautoSTARTENDT/home/alanp/www/video.onchill/myenv/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 r r)propertyrr&r'r0r2r4r9__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 rregisterrDs1 h )) ) t DX&rcrt|tsJt|}t|}|j |y)zxUnregister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N)r>r?rr@remove)rrClsts r unregisterrHs4 h )) ) t D d CJJxrcVt|jD]}|j|y)zeBroadcast an event to all registered listeners. Parameters ---------- event : Event N)r@rnotify)eventrCs r broadcastrLs#  +,rc^eZdZdZej dZej dZdZy)r?z(Base class for all event listeners. cy)zkCalled when there is a *START* event. Parameters ---------- event : Event Nrr%rKs ron_startzListener.on_start rcy)ziCalled when there is a *END* event. Parameters ---------- event : Event NrrOs ron_endzListener.on_endrQrc|jr|j|y|jr|j|yt d)zpNotify this Listener with the given Event. Parameters ---------- event : Event unreachableN)r0rPr2rSAssertionErrorrOs rrJzListener.notifys5 >> MM% \\ KK  / /rN) r r r r abcabstractmethodrPrSrJrrrr?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_tsrOs rrPzTimingListener.on_start s% ;;! wDH q rc|xjdzc_|jdk(r-t|dd}t|jz |z|_yy)Nr^r _duration)r\getattrr_r`rb)r%rKlasts rrSzTimingListener.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. rb)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. )rbr+s rdurationzTimingListener.durations~~rN) r r r r r)rPrSr;rgrirrrrZrZs> 9 **rrZc"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|fyrmrnrAtimerOs rrPzRecordingListener.on_start1 DIIK/0rcb|jjtj|fyrmrprOs rrSzRecordingListener.on_end4rrrN)r r r r r)rPrSrrrrkrk(s 11rrkc#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)rDrHrBs rinstall_listenerru8s0$ 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)rZrurgri)rcallbacktls r install_timerryQsF*  B $ # $ ww $ #sAA(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)rkru)rrls rinstall_recorderr|ns*&  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 rrL)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 rrL)rr'r(r~s 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_eventrs@   @  @ Dt$ s A '? A AA rCc 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_idrnr'strrr0dictrA) rCrrevsrrecr'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||dddy#1swYyxYw)Nw)ropenjsondump)routfilenamerCs r_write_chrome_tracez=_setup_chrome_trace_exit_handler.._write_chrome_traces5)2 (C C IIc3 ! s 9AN)rkrDr CHROME_TRACEatexit)rrrCs @@r _setup_chrome_trace_exit_handlerrs9!"H x(""H __  rrmr:)*r rrrrWrrqrtimeitrr_ contextlibrr collectionsr numba.corerEnumr frozensetrrrlistr@rDrHrLABCr?rZrkruryr|rrrrrrrrrrs(!F )0#$)),XXv$  ' #0sww#0L#X#L 1 1 ##08. $*):0   $&r