Foj)&bSSKrSSKrSSKJr SSKJr SSKJr SSKJ r J r J r SSK J r SSKJr SS KJrJr S rS \4S jrS \S\\\\4S \4SjrS \S S4SjrS"SSSSSS.S\\S\\S\\S \\S\\\\4S\\\\ 4S \\4SjjjrS"S\\S S4SjjrS#S\\S\S S4SjjrS$S\\S\\S S4SjjrS%S\\S\\S\\S \\S\\S\\\\ 44 SjjrS"SSSSSSS .S\\S\\S\\S \\S\\S\\\\ 44 S!jjjrg)&N)proton)getenv)flags) HookManager LaunchHookInstrumentationHook)Hook)BaseMode)OptionalUnionrreturnc[RRRR 5R nUS:XagUS:Xag[ S5e)Ncudacuptihip roctracerz/No backend is available for the current target.)tritonruntimedriveractiveget_current_targetbackend ValueError)rs Q/mnt/workspace/venv-train/lib/python3.13/site-packages/triton/profiler/profile.py_select_backendrsHnn##**==?GGG& E JKKrmodecUS:XaL[RRRR 5R nU(aUSU3$U$U(a [ U5$S$)Ninstrumentation:)rrrrrrstr)rrprefixs r _get_mode_strr%sX##&&--@@BJJ%)&4&!5v53t9$"$rcUS:Xa)SS/nUHn[US5cM[SUS35e [RRR R 5Hup4URn[US5bM [[RRU5nUcMI[RRU5=n(dMq[RRXWS5 M g)NrHIP_VISIBLE_DEVICESCUDA_VISIBLE_DEVICESz3Proton does not work when the environment variable zK is set on AMD GPUs. Please unset it and use `ROCR_VISIBLE_DEVICES` insteadr) rrrknobsrknob_descriptorsitemskeygetattrtoenvsetenv)rhip_device_envsenvattrdescr,valenv_vals r _check_envr6 s+02HI"Cc4 , I#OZ[#ll))::@@B hh #t  $&,,--t4C$ll005575LL''QZ8 Crshadowtreecontextdatarrhooknamer:r;r<c>[R(d)[RRR (agS[lUc[OUnUc [5OUn[X45n[U5 [R"XX#U5n[U[5(a[R "XW5 O8US:Xa [R "[#5U5 OUb[%SU<35eUS:Xa [R "['U5U5 U$)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_NAMErr%r6 libprotonstart isinstancer rregisterrrr )r=r:r;rrr<mode_strsessions rrDrD3sf V\\0088E#'< TD#*?oGW+HwoodTHEG$T+  Z\73 -dX677##06@ NrrHc[R(aUS:wa [S5e[R"U5 Uc[ R "5 g[ R"U5 g)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 rzEOnly one session can be activated when running from the command line.N)rr?rractivaterC activate_all)rHs rrJrJsL gl`aa! 7#rflushingc[R(aUS:wa [S5e[R"U5 Uc[ R "U5 g[ R"X5 g)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 rzGOnly one session can be deactivated when running from the command line.N)rr?rr deactivaterCdeactivate_all)rHrLs rrNrNsN glbcc7#  *W/r output_formatc[R"U5 Uc"S[l[R "U5 g[R (aUS:wa [S5e[R"X5 g)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 NFrzEOnly one session can be finalized when running from the command line.) r unregisterrrArC finalize_allr?rfinalize)rHrPs rrTrTsV7#"}-   'Q,de e72rc^^^^^^^^[R"T5UUUUUUU4Sj5nU$)z Context manager for profiling. Internally use only. Args: See start() for the arguments. Returns: wrapper (function): The wrapped function. c J>[T TTTT TS9nT"U0UD6n[U5 U$)Nr9)rDrN) argskwargsrHretrr:r;funcr<rr=s rwrapper_profiling..wrappers5gD'PT[_`D#F#7 r) functoolswraps)rZr=r:r;rrr<r[s``````` r _profilingr_s,&__T Nrr=r:r;rrr<c L^^^^^^Uc UUUUUU4SjnU$[UTTTTTTS9$)z Decorator for profiling. Usage: ```python @proton.profile def foo(): pass ``` Args: See start() for the arguments. Returns: decorator (function): The decorator function. c ">[UTTTTTTS9$)Nr`r_)frr:r;r<rr=s r decoratorprofile..decoratorsadG$PW^bimn nrr`rc)rZr=r:r;rrr<res `````` rprofilergs96 | o o$T7w]ahlmmr)N)NF)Nr")Nr7r8NNN) r]rtriton._C.libprotonrrCtriton._C.libtritonrrhooksrrr hooks.hookr rr typingr r rBr#rr%r6intrDrJboolrNrTr_rgrrrpsf 3&??"LL%3%huS(]/C&D%%999(K& !+/'+K 3-Kc]K 3- K c] K 5h' ( K 5d# $Kc]K\$hsm$t$,0 000.3hsm38C=3RV34% !'+ 3-c] 3-  c]  3-  5d# $< #n% !'+#n 3-#nc] #n 3- #n c] #n 3-#n 5d# $#nr