K iddlmZmZddlmZddlmZddlmZ ddl m Z dZ e d d Z e d d ZGd deZy )) enter_state exit_state)LazyDict)Hook)proton) ContextVar__proton_launch_metadataop_nameN)defaultidceZdZdZdZgdZeDcgc]}d| c}}dgzdgzZdZfdZde d e d e d dfd Z d Z dZ ded dfdZded dfdZxZScc}}w) LaunchHookdN) @flopsbytescyNselfs b/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/triton/profiler/hooks/launch.py__init__zLaunchHook.__init__ cd|jtt|||_|jSr) _instancesuperr__new__)cls __class__s rr$zLaunchHook.__new__s* == !*c:3?CM}}r namemetadata_grouphashreturncyrr)rmodulefunctionr'r(r)s r init_handlezLaunchHook.init_handlerr cyrrrs ractivatezLaunchHook.activate rr cyrrrs r deactivatezLaunchHook.deactivate#rr metadatactt|j}ttj Dcic] }||vs|||}}t j|dtjtjtjtj|dtjtj|ycc}w)Nr') rCOMPUTE_METADATA_SCOPE_NAMEgetrrmetricsr setr libproton record_scopeenter_op add_metrics)rr3 lazy_metadatak fn_metricss renterzLaunchHook.enter&s/0   3=3E3E\amI[aq))\ \ M&)* y%%'(2668]6%:;bffh 3 ]s C#C#cxtjtjtjyr)r9exit_oprr6r )rr3s rexitzLaunchHook.exit0s"&&(GKKM2r )__name__ __module__ __qualname__priorityr" flops_widthr7rr$strdictr.r0r2rr@rC __classcell__).0widthr&s00@rrr sHI!K,785ug8G9DyPG   # t SV [_   4h4443X3$3;9s Ar)staterrtriton.compilerrhookrtriton._C.libprotonr r9 contextvarsr r5r rrrr rrSs>+$3"8 Y -d#$3$3r