K ipddlmZddlmZddlmZmZmZddlm Z ddl m Z GddZ Gdd Z y) )LazyDict)abstractmethod)DictAnyOptional) defaultdictNc eZdZUdZeed<edededede eefdedd f d Z ed e dd fd Z ed e dd fd Z eddZeddZy )Hookrprioritymodulefunctionnamemetadata_grouphashreturnNctNNotImplementedError)selfr r rrrs `/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/triton/profiler/hooks/hook.py init_handlezHook.init_handle s "!metadatactrrrrs renterz Hook.enter!!rctrrrs rexitz Hook.exitrrctrrrs ractivatez Hook.activaterrctrrr"s r deactivatezHook.deactivaterr)rN)__name__ __module__ __qualname__r int__annotations__rrstrrrrrr r#r%rrr r sHc"#""C"QUVY[^V^Q_"""&"""h"4"""X"$""""""rr c >eZdZUgZeeed<edZe e e ee ffed<e de de dede eefded d f d Ze d ed d fd Ze d ed d fdZe ddee d d fdZe ddee d d fdZe dede d d fdZe ddee d d fdZy ) HookManager active_hooksc ttSr)rboolr,rrzHookManager.%s [QUEVr session_hooksr r rrrrNcXtjD]}|j|||||yr)r.r/r)r r rrrhooks rrzHookManager.init_handle's/,, KD   VXt^T J KrrcPtjD]}|j|yr)r.r/rrr5s rrzHookManager.enter,s#,, !D JJx  !rcbttjD]}|j|yr)reversedr.r/r r7s rr zHookManager.exit1s*[556 D IIh  rsessionc||tjj}n|g}|D]q}tj|D]Y}|tjvr/|j tjj |dtj||<[stjj ddy)NTc|jSrr xs rr2z&HookManager.activate..E AJJrkeyreverse)r.r3keysr/r#appendsort)r:sessionsr5s rr#zHookManager.activate7s ?"00557HyH @G#11': @{777MMO,,33D9;? ))'248  @ @   %%*>%Mrc|tjj}n|g}t}|D]S}tj|D];tjvr|j dtj|<=U|D]bt fdtjjDr4jtjjdy)NFc3(K|] }| ywrr,.0r3r5s r z)HookManager.deactivate..Wsc}}T*c) r.r3rDsetr/addanyvaluesr%remove)r:rGdeactivated_hooksr5s @rr%zHookManager.deactivateGs ?"00557HyHE AG#11': A;333%))$/;@ ))'248 A A& 6Dc @Y@Y@`@`@bcc!((//5 6rr5c@dtj||<|tjvr/|jtjj |tjj ddt jjjtjt jjjtjt jjjtjy)NTc|jSrr=r>s rr2z&HookManager.register..br@rrA)r.r3r/r#rErFknobsruntimekernel_load_end_hookrOrlaunch_enter_hookrlaunch_exit_hookr )r5r:s rregisterzHookManager.register[s37 !!'*40 {// / MMO  $ $ + +D 1  %%*>%M  **..{/F/FG ''++K,=,=> &&**;+;+;.ysg==.grM)r.r3r/r%clearpopitemsrPrQrRrVrWrXrrYrrZr )r: popped_hooksactiver5s @r unregisterzHookManager.unregisteris>  7+2K2K#K  ?#00 "! "  $ $ * * ,  % % + + -&4488AL , 2 2 4 : fgKD]D]DdDdDfggOO%,,33D9  :'' MM . . 5 5k6M6M N MM + + 2 2;3D3D E MM * * 1 1+2B2B C(rr)r&r'r(r/listr r*rr3rr)r1 staticmethodrr+rrrr rr#r%r[rcr,rrr.r.!sk!L$t*!1<=V1WM4T$*--.WKCK3KcK4PSUXPX>KadKimKK!!T!! x D   N(3- N4 N N6HSM6T66& =t =c =d = =DHSMDTDDrr.)triton.compilerrabcrtypingrrr collectionsr triton.knobsrVr r.r,rrrks/$&&#""2_D_Dr