+ ړLj)& 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"C/app/.local/lib/python3.14/site-packages/triton/profiler/profile.py __annotate__rsLLLc\PPPP 4P pVR8XdR#VR8XdR#\ R4h)cudacuptihip roctracerz/No backend is available for the current target.)tritonruntimedriveractiveget_current_targetbackend ValueError)r s r_select_backendr"sHnn##**==?GGG& E JKKrctV^8dQhR\R\\\\3,,R\/#)rr moder)rr r r )rs"rrrs/%%3%huS(]/C&D%%rcVR8XdN\PPPP 4P pV'dV RV 2#T#V'd \ V4#R#)instrumentation:)rrrrrr r)r r$prefixs&& r _get_mode_strr*sX##&&--@@BJJ%)&4&!5v53t9$"$rc(V^8dQhR\RR/#)rr rNr)rs"rrr s9999rcVR8Xd+RR.pVF p\VR4fK\RV R24h \PPP P 4Fwr4VPp\VR4eK!\\PPV4pVfKK\PPV4;p'gKt\PPWW^,4K R#)rHIP_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` instead) rr!rknobsrknob_descriptorsitemskeygetattrtoenvsetenv)r hip_device_envsenvattrdescr2valenv_vals& r _check_envr< s+02HI"Cc4 , I#OZ[#ll))::@@B hh #t  $&,,--t4C$ll0055755LL''QZ8 Crcontextshadowdatatreer r$hookcJV^8dQhR\\,R\\,R\\,R\\,R\\\\3,,R\\\\3,,R\\ ,/#)rnamer=r?r r$rAr)r rr r r int)rs"rrr3sKK 3-Kc]K 3- K c] K 5h' ( K 5d# $Kc]KrcP\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", "roctracer", "instrumentation"]. Defaults to None, which automatically selects the backend matching the current active runtime. 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 "roctracer", available options are ["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_linerr/rdisable profiling_onDEFAULT_PROFILE_NAMEr"r*r< libprotonstart isinstancer rregisterrr!r)rCr=r?r r$rAmode_strsessions&$$$$$ rrKrK3sf V\\00888E#'< TD#*?oGW+HwoodTHEG$T+  Z\73 -dX677##06@ Nrc>V^8dQhR\\,RR/#)rrOrN)r rD)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)rrFr!ractivaterJ activate_all)rOs&rrRrRsO gl`aa! 7#rcJV^8dQhR\\,R\RR/#)rrOflushingrN)r rDbool)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)rrFr!r deactivaterJdeactivate_all)rOrUs&&rrXrXsQ glbcc7#  *W/rc`V^8dQhR\\,R\\,RR/#)rrO output_formatrN)r rDr)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 unregisterrrHrJ finalize_allrFr!finalize)rOr[s&&rr_r_sY7#"}-    'Q,de e72rcV^8dQhR\\,R\\,R\\,R\\,R\\,R\\\\3,,/#rrCr=r?r r$rAr 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#))r=r?r r$rA)rKrX) argskwargsrOretr r=r?funcrAr$rCs *, rwrapper_profiling..wrappers5gD'PT[_`D#F#7 r) functoolswraps)rhrCr=r?r r$rArisfffffff r _profilingrms,&__T NrrCcV^8dQhR\\,R\\,R\\,R\\,R\\,R\\\\3,,/#rarb)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#)rCr=r?r r$rArm)fr r=r?rAr$rCs&r decoratorprofile..decoratorsadG$PW^bimn nrrqrr)rhrCr=r?r r$rArts&dddddd rprofilervs96 | o o$T7w]ahlmmr)N)NF)Nr()Nr>r@NNN)rkrtriton._C.libprotonrrJtriton._C.libtritonrrhooksrrr hooks.hookr r$r typingr r rIr"r*r<rKrRrXr_rmrvrrr}s 3&??"L%9&K&K! K " K ,0 K(,KK\$,0.30:#n#n& #n ! #n " #n#n(,#n#nr