+ jr( 2^RIt^RIt^RIHt^RIHt^RIHt^RIH t H t H t ^RI H t ^RIHt^RIHtHtR tR R ltR R ltRRltR$RRRRRRRRRR/RRllltR$RRlltR%RRlltR&RRlltR'RR lltR$R!RRRRRRRRRRR/R"R#llltR#)(N)proton)getenv)flags) HookManager LaunchHookInstrumentationHook)Hook)BaseMode)OptionalUnionrc$V^8dQhR\/#)returnstr)formats"X/home/luca/screen-checking/.venv/lib/python3.14/site-packages/triton/profiler/profile.py __annotate__rsBBBc\PPPP 4P p\ P!V4#N)tritonruntimedriveractiveget_current_targetbackend libproton#select_profiler_from_triton_backend)rs r_select_backendr s7nn##**==?GGG  8 8 AArctV^8dQhR\R\\\\3,,R\/#)rrmoder)rr r r )rs"rrrs/%%3%huS(]/C&D%%rcVR8XdN\PPPP 4P pV'dV RV 2#T#V'd \ V4#R#)instrumentation:)rrrrrrr)rr"prefixs&& r _get_mode_strr(sX##&&--@@BJJ%)&4&!5v53t9$"$rc(V^8dQhR\RR/#)rrrNr)rs"rrrs9999rcFVR9d+RR.pVF p\VR4fK\RV R24h VR8H;'d@\PPP P 4P^d8p\PPPP4FwrEVPp\VR4eK!VR8XdMV'dE\PPV\PPP4Kt\\PPV4pVfK\PP!V4;p'gK\PPWh^,4K R#) rocprofilerHIP_VISIBLE_DEVICESCUDA_VISIBLE_DEVICESNz3Proton does not work when the environment variable zK is set on AMD GPUs. Please unset it and use `ROCR_VISIBLE_DEVICES` insteadcuptiTRITON_CUPTI_LIB_PATH)r+ roctracer)r ValueErrorrrrrrarchknobsrknob_descriptorsitemskeysetenvcupti_lib_blackwell_dirgetattrtoenv) rhip_device_envsenvuse_blackwell_cuptiattrdescr6valenv_vals & r _check_envrBsG..02HI"Cc4 , I#OZ[# "W,nn1F1F1M1M1`1`1b1g1gkn1nll))::@@B hh #t  $--2E ##C)<)<)T)TU&,,--t4C$ll0055755LL''QZ8Crcontextshadowdatatreerr"hookcJV^8dQhR\\,R\\,R\\,R\\,R\\\\3,,R\\\\3,,R\\ ,/#)rnamerCrErr"rGr)r rr r r int)rs"rrr4sNN 3-Nc]N 3- N c] N 5h' ( N 5d# $Nc]NrcP\P'g+\PPP 'dR#R\nVf\MTpVf \4MTp\W44p\V4\P!WW#V4p\V\4'd\P !WW4M:VR8Xd!\P !\#4V4MVe\%RV: 24hVR8Xd!\P !\'V4V4V#)a Start profiling with the given name and backend. Usage: ```python proton.start("my_profile") # do something proton.finalize() ``` Args: name (str, optional): The name (with path) of the profiling session. If not provided, the default name is "~/proton.", where suffix is the default format according to the data type. For example, if data is "tree", the default name is "~/proton.hatchet". context (str, optional): The context to use for profiling. Available options are ["shadow", "python"]. Defaults to "shadow". data (str, optional): The data structure to use for profiling. Available options are ["tree", "trace"]. Defaults to "tree". backend (str, optional): The backend to use for profiling. Available options are [None, "cupti", "rocprofiler", "roctracer", "instrumentation"]. Defaults to None, which automatically selects the backend matching the current active runtime. On AMD GPUs, "rocprofiler" is preferred and will fall back to "roctracer" if rocprofiler-sdk is not available. mode (Union[str, BaseMode], optional): The "mode" to use for profiling, which is specific to the backend. Can be a string or an instance of BaseMode (or any subclass thereof). Defaults to None. For "cupti", available options are [None, "pcsampling", "periodic_flushing"]. For "rocprofiler", available options are [None, "periodic_flushing"]. For "roctracer", available options are [None, "periodic_flushing"]. For "instrumentation", available options are [None]. Each mode has a set of control knobs following with the mode name. For example, "periodic_flushing" mode has a knob: - format: The output format of the profiling results. Available options are ["hatchet", "hatchet_msgpack", "chrome_trace"]. Default is "hatchet". The can be set via `mode="periodic_flushing:format=chrome_trace"`. hook (Union[str, Hook], optional): The hook to use for profiling. You may pass either: - a string hook name, e.g. "triton" (kernel launch metadata), or - a custom Hook instance. Defaults to None. Returns: session (Optional[int]): The session ID of the profiling session, or None if profiling is disabled. NTrzUnsupported hook: r$)r command_linerr3rdisable profiling_onDEFAULT_PROFILE_NAMEr r(rBrstart isinstancer rregisterrr1r)rIrCrErr"rGmode_strsessions&$$$$$ rrPrP4sl V\\00888E#'< TD#*?oGW+HwoodTHEG$T+  Z\73 -dX677##06@ Nrc>V^8dQhR\\,RR/#)rrTrN)r rJ)rs"rrrs$$hsm$t$rc\P'dV^8wd \R4h\P!V4Vf\ P !4R#\ P!V4R#)z Activate the specified session. The profiling session will be active and data will be recorded. Args: session (int): The session ID of the profiling session. Defaults to None (all sessions) Returns: None zEOnly one session can be activated when running from the command line.N)rrLr1ractivater activate_all)rTs&rrWrWsO gl`aa! 7#rcJV^8dQhR\\,R\RR/#)rrTflushingrN)r rJbool)rs"rrrs%00 000rc\P'dV^8wd \R4h\P!V4Vf\ P !V4R#\ P!W4R#)aP Stop the specified session. The profiling session's data will still be in the memory, but no more data will be recorded. Args: session (int): The session ID of the profiling session. Defaults to None (all sessions) flushing (bool): Whether to flush the profiling data before deactivating. Defaults to True. Returns: None zGOnly one session can be deactivated when running from the command line.N)rrLr1r deactivaterdeactivate_all)rTrZs&&rr]r]sQ glbcc7#  *W/rc`V^8dQhR\\,R\\,RR/#)rrT output_formatrN)r rJr)rs"rrrs*33hsm38C=3RV3rc\P!V4Vf$R\n\P !V4R#\P 'dV^8wd \R4h\P!W4R#)a Finalizes a profiling session. Flush and write the profiling data to the file specified by the session name. Args: session (int, optional): The session ID to finalize. If None, all sessions are finalized. Defaults to None. output_format (str, optional): The output format for the profiling results. Available options are ["hatchet", "hatchet_msgpack", "chrome_trace"]. Returns: None NFzEOnly one session can be finalized when running from the command line.) r unregisterrrNr finalize_allrLr1finalize)rTr`s&&rrdrdsY7#"}-    'Q,de e72rcV^8dQhR\\,R\\,R\\,R\\,R\\,R\\\\3,,/#rrIrCrErr"rGr rr r )rs"rrrsh 3-c] 3-  c]  3-  5d# $rc^aaaaaaa\P!S4VVVVVVV3Rl4pV#)z Context manager for profiling. Internally use only. Args: See start() for the arguments. Returns: wrapper (function): The wrapped function. c P<\S SSSS SR7pS!V/VBp\V4V#))rCrErr"rG)rPr]) argskwargsrTretrrCrEfuncrGr"rIs *, rwrapper_profiling..wrappers5gD'PT[_`D#F#7 r) functoolswraps)rmrIrCrErr"rGrnsfffffff r _profilingrrs,&__T NrrIcV^8dQhR\\,R\\,R\\,R\\,R\\,R\\\\3,,/#rfrg)rs"rrrsr#n#n 3-#nc] #n 3- #n c] #n 3-#n 5d# $#nrc TaaaaaaVfVVVVVV3RlpV#\VSSSSSSR7#)z Decorator for profiling. Usage: ```python @proton.profile def foo(): pass ``` Args: See start() for the arguments. Returns: decorator (function): The decorator function. c (<\VSSSSSSR7#)rIrCrErr"rGrr)frrCrErGr"rIs&r decoratorprofile..decoratorsadG$PW^bimn nrrvrw)rmrIrCrErr"rGrys&dddddd rprofiler{s96 | o o$T7w]ahlmmrr)NF)Nr&)NrDrFNNN)rprtriton._C.libprotonrrtriton._C.libtritonrrhooksrrr hooks.hookr r"r typingr r rOr r(rBrPrWr]rdrrr{rrrs 3&??"B %92N&N! N " N ,0 N(,NNb$,0.30:#n#n& #n ! #n " #n#n(,#n#nr