U ,dy/@spdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddlmZddlmZGdddejZed d d d gZd dZGdddZeeZddZddZddZGdddejZGdddeZGdddeZ e ddZ!e dd Z"e d!d"Z#d0d#d$Z$d1d%d&Z%e d2d'd(Z&d)d*Z'e d+d,d-Z(d.d/Z)ej*rle)dS)3ag 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)configc@s eZdZdZeZeZdS) EventStatuszStatus of an event. N)__name__ __module__ __qualname____doc__enumautoSTARTENDrr4/tmp/pip-unpacked-wheel-eu7e0c37/numba/core/event.pyr2srznumba:compiler_lockz numba:compileznumba:llvm_locknumba:run_passcCs(|dr$|tkr$|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)kindmsgrrr _guard_kindBs rc@sneZdZdZdddZeddZeddZed d Zed d Z ed dZ eddZ ddZ e Z dS)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__``. NcCs8t||_||_||_|dks*|ddkr.dn||_dSNr)r_kind_status_data _exc_details)selfrstatusdata exc_detailsrrr__init__ds  zEvent.__init__cCs|jS)zFEvent kind Returns ------- res : str )rrrrrrlsz Event.kindcCs|jS)zPEvent status Returns ------- res : EventStatus )rr$rrrr vsz Event.statuscCs|jS)zIEvent data Returns ------- res : object )rr$rrrr!sz Event.datacCs |jtjkS)zSIs it a *START* event? Returns ------- res : bool )rrrr$rrris_startszEvent.is_startcCs |jtjkS)zRIs it an *END* event? Returns ------- res : bool )rrrr$rrris_endsz Event.is_endcCs |jdkS)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)rr$rrr is_faileds zEvent.is_failedcCs8|jdk rt|jjnd}d|jd|jd|dS)NNonezEvent(z, z, data: ))r!typer rr)rr!rrr__str__s z Event.__str__)NN)rr r r r#propertyrr r!r%r&r'r+__repr__rrrrrXs        rcCs(t|tstt|}t||dS)zvRegister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N) isinstanceListenerAssertionErrorr _registeredappendrlistenerrrrregistersr5cCs,t|tstt|}t|}||dS)zxUnregister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N)r.r/r0rr1remove)rr4lstrrr unregistersr8cCst|jD]}||q dS)zeBroadcast an event to all registered listeners. Parameters ---------- event : Event N)r1rnotify)eventr4rrr broadcastsr;c@s4eZdZdZejddZejddZddZdS) r/z(Base class for all event listeners. cCsdS)zkCalled when there is a *START* event. Parameters ---------- event : Event Nrrr:rrron_startszListener.on_startcCsdS)ziCalled when there is a *END* event. Parameters ---------- event : Event Nrr<rrron_endszListener.on_endcCs0|jr||n|jr$||ntddS)zpNotify this Listener with the given Event. Parameters ---------- event : Event Z unreachableN)r%r=r&r>r0r<rrrr9s   zListener.notifyN) rr r r abcabstractmethodr=r>r9rrrrr/s   r/c@s@eZdZdZddZddZddZedd Zed d Z d S) TimingListenerzA listener that measures the total time spent between *START* and *END* events during the time this listener is active. cCs d|_dSr)_depthr$rrrr#szTimingListener.__init__cCs$|jdkrt|_|jd7_dS)Nr)rBtimer_tsr<rrrr= s zTimingListener.on_startcCs:|jd8_|jdkr6t|dd}t|j||_dS)NrCr _duration)rBgetattrrDrErF)rr:lastrrrr>s  zTimingListener.on_endcCs t|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. rF)hasattrr$rrrdoneszTimingListener.donecCs|jS)zReturns the measured duration. This may raise ``AttributeError``. Users can use ``.done`` to check that a measurement has been made. )rFr$rrrdurationszTimingListener.durationN) rr r r r#r=r>r,rJrKrrrrrAs rAc@s(eZdZdZddZddZddZdS) 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. cCs g|_dSN)bufferr$rrrr#.szRecordingListener.__init__cCs|jt|fdSrMrNr2timer<rrrr=1szRecordingListener.on_startcCs|jt|fdSrMrOr<rrrr>4szRecordingListener.on_endN)rr r r r#r=r>rrrrrL(srLc cs&t||z |VW5t||XdS)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)r5r8r3rrrinstall_listener8s  rQc cs6t}t|| |VW5QRX|jr2||jdS)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)rArQrJrK)rcallbacktlrrr install_timerQs  rTc cs&t}t|| |VW5QRXdS)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)rLrQ)rZrlrrrinstall_recorderns rUcCst|tj|d}t|dS)zTrigger the start of an event of *kind* with *data*. Parameters ---------- kind : str Event kind. data : any; optional Extra event data. )rr r!N)rrrr;)rr!evtrrr start_events rWcCst|tj||d}t|dS)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)rrrr;)rr!r"rVrrr end_events rXc #s<t,}|jfdd}tddVW5QRXdS)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. cst|ddS)N)r!r")rX)r"r!rrron_exitsztrigger_event..on_exit)r!N)rpushrW)rr!ZscoperZrrYr trigger_events  r\cCs,z tWStk r&tYSXdSrM) threadingZget_native_identAttributeError get_identrrrr_get_native_idents r`)r4c Cspt}t}g}|jD]R\}}|j}t|j}|jr:dnd}|d} |} t|||||| | d} | | q|S)zGPrepare events in `listener` for serializing as chrome trace data. BEname)catpidtidtsphrcargs) osgetpidr`rNr!strrr%dictr2) r4rerfevsrgZrecr!rdrhrcriZevrrr_prepare_chrome_trace_datas(  rocs.ttdtjtjfdd}dS)z\Setup a RecordingListener and an exit handler to write the captured events to file. rc s.t}td}t||W5QRXdS)Nw)roopenjsondump)rnoutfilenamer4rr_write_chrome_traces z=_setup_chrome_trace_exit_handler.._write_chrome_traceN)rLr5r CHROME_TRACEatexit)rwrrur _setup_chrome_trace_exit_handlers  rz)N)NN)N)+r rjrrryr?r rPr]ZtimeitrrD contextlibrr collectionsrZ numba.corerEnumr frozensetrrrlistr1r5r8r;ABCr/rArLrQrTrUrWrXr\r`rorzrxrrrrsT#   [  &&