g/dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddlmZddlmZGddejZegd Zd ZGd d ZeeZd ZdZdZGddejZGddeZGddeZ e dZ!e dZ"e dZ#d dZ$d!dZ%e d dZ&dZ'de fdZ(dZ)ej*r e)dSdS)"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)configcNeZdZdZejZejZdS) EventStatuszStatus of an event. N)__name__ __module__ __qualname____doc__enumautoSTARTENDF/granary/edev/quantum/lib/python3.11/site-packages/numba/core/event.pyr r 2s2 DIKKE $)++CCCrr )znumba:compiler_lockz numba:compileznumba:llvm_locknumba:run_passcj|dr|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_kindrBsG x  T%?%?===oo KrceZdZdZd dZedZedZedZedZ edZ ed Z d Z 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__``. Ncpt||_||_||_||ddn||_dSNr)r_kind_status_data _exc_details)selfrstatusdata exc_detailss r__init__zEvent.__init__dsF &&   + 3{1~7M"T"- rc|jS)zFEvent kind Returns ------- res : str )r!r%s rrz Event.kindl zrc|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,|jtjkS)zSIs it a *START* event? Returns ------- res : bool )r"r rr+s ris_startzEvent.is_starts|{000rc,|jtjkS)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((rct|jt|jjnd}d|jd|jd|dS)NNonezEvent(z, z, data: ))r'typer r!r")r%r's r__str__z Event.__str__sM9( ??/11.4 C CCdlCCDCCCCrNN)r r r r r)propertyrr&r'r0r2r4r9__repr__rrrrrXs  ////XXX11X1//X/ ) )X )DDD HHHrrct|tsJt|}t||dS)zvRegister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N) isinstanceListenerr _registeredappendrlisteners rregisterrDsF h ) )))) t  DX&&&&&rct|tsJt|}t|}||dS)zxUnregister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N)r>r?rr@remove)rrClsts r unregisterrHsI h ) )))) t  D d CJJxrcZt|jD]}||dS)zeBroadcast an event to all registered listeners. Parameters ---------- event : Event N)r@rnotify)eventrCs r broadcastrLs9  +rcXeZdZdZejdZejdZdZdS)r?z(Base class for all event listeners. cdS)zkCalled when there is a *START* event. Parameters ---------- event : Event Nrr%rKs ron_startzListener.on_start  rcdS)ziCalled when there is a *END* event. Parameters ---------- event : Event NrrOs ron_endzListener.on_endrQrc|jr||dS|jr||dSt d)zpNotify this Listener with the given Event. Parameters ---------- event : Event unreachableN)r0rPr2rSAssertionErrorrOs rrJzListener.notifysV > 0 MM% \ 0 KK      // /rN) r r r r abcabstractmethodrPrSrJrrrr?r?si        0 0 0 0 0rr?cPeZdZdZdZdZdZedZedZ dS)TimingListenerzA listener that measures the total time spent between *START* and *END* events during the time this listener is active. cd|_dSr )_depthr+s rr)zTimingListener.__init__s  rcb|jdkrt|_|xjdz c_dS)Nr)r\timer_tsrOs rrPzTimingListener.on_start s. ;!  wwDH q rc|xjdzc_|jdkr1t|dd}t|jz |z|_dSdS)Nr^r _duration)r\getattrr_r`rb)r%rKlasts rrSzTimingListener.on_endsS q ;!  4a00D#gg0D8DNNN  rc"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. 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 999 **X*XrrZc$eZdZdZdZdZdZdS)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|_dSN)bufferr+s rr)zRecordingListener.__init__.s  rc`|jtj|fdSrmrnrAtimerOs rrPzRecordingListener.on_start1) DIKK/00000rc`|jtj|fdSrmrprOs rrSzRecordingListener.on_end4rrrN)r r r r r)rPrSrrrrkrk(sK 11111111rrkc#~Kt|| |Vt||dS#t||wxYw)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_listenerru8sR$ T8#4""""" 4""""s*<c#Kt}t||5|Vdddn #1swxYwY|jr||jdSdS)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_timerryQs*   B $ # # ws 266c#Kt}t||5|VddddS#1swxYwYdS)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 $ # #s 377c\t|tj|}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)rr rrL)rr'evts r start_eventrs* T+"3$ ? ? ?C cNNNNNrc^t|tj||}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)rr rrL)rr'r(r~s r end_eventrs4  +/+   CcNNNNNrc#Kt5}|jfd}tdVddddS#1swxYwYdS)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|dS)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 A cp tjS#t$rtjcYSwxYwrm) threadingget_native_identAttributeError get_identrrr_get_native_identrsF%)+++ %%%"$$$$$%s 55rCc &tj}t}g}|jD]e\}}|j}t |j}|dz}|jrdnd} |d} |} t||||| | | } | | f|S)zGPrepare events in `listener` for serializing as chrome trace data. i@BBEname)catpidtidtsphrargs) osgetpidrrnr'strrr0dictrA) rCrrevsrrecr'r ts_scaledrrrevs r_prepare_chrome_trace_datars )++C   C C?  Cx#(mmN L )SScF| #)    2 Jrcttdtjtjfd}dS)z\Setup a RecordingListener and an exit handler to write the captured events to file. rct}td5}tj||ddddS#1swxYwYdS)Nw)ropenjsondump)routfilenamerCs r_write_chrome_tracez=_setup_chrome_trace_exit_handler.._write_chrome_traces)22 (C  C Ic3                     sAA AN)rkrDr CHROME_TRACEatexit)rrrCs @@r _setup_chrome_trace_exit_handlerrs^!""H x((("H _     _   rrmr:)+r rrrrWrrqrtimeitrr_ contextlibrr collectionsr numba.corerEnumr frozensetrrrlistr@rDrHrLABCr?rZrkruryr|rrrrrrrrrrrs!!F ))))))00000000######$),XXXXXXXXvk$  ' ' '   #0#0#0#0#0sw#0#0#0L#####X###L 1 1 1 1 1 1 1 1 ###08.    $*%%%):0     '$$&&&&&''r