9i*#TSSKrSSKrSSKrSSKrSSKJr SSKJr SSKJ r J r J r SSK J r JrJr SSKJr SSKJrJr S rS \4S jrS \S \4S jrS \S\\\\4S \4SjrS \S S4SjrS!SSSSSS.S\\S\\S\\S \\S\\\\4S\\4 SjjjrS!S\\S S4SjjrS!S\\S S4SjjrS"S\\S\\S S4SjjrS#S\\S\\S\\S \\S\\S\\4 Sjjr S!SSSSSSS.S\\S\\S\\S \\S\\S\\4 S jjjr!g)$N)knobs)proton)set_profiling_offset_profiling_onis_command_line) HookManager LaunchHookInstrumentationHook)BaseMode)OptionalUnionrreturnc[RRRR 5R nUS:XagUS:Xag[ S5e)Ncudacuptihip roctracerz/No backend is available for the current target.)tritonruntimedriveractiveget_current_targetbackend ValueError)rs W/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/triton/profiler/profile.py_select_backendrsHnn##**==?GGG& E JKKrcSnUS:Xan[RRnUcQ[[R "[ 5RRR5S- S- S- S- 5nU$)Nrbackendsnvidialib) rr cupti_dirstrpathlibPath__file__parentabsolute)rlib_paths r_get_backend_default_pathr,sjH'<<))  7<<188??HHJZWZbbejjmttuH OrmodecUS:XaL[RRRR 5R nU(aUSU3$U$U(a [ U5$S$)Ninstrumentation:r )rrrrrrr%)rr-prefixs r _get_mode_strr2&sX##&&--@@BJJ%)&4&!5v53t9$"$rczUS:Xa5SS/nUH*n[R"US5cM[SUS35e gg)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` instead)osgetenvr)rhip_device_envsenvs r _check_envr:-sV+02HI"Cyyd#/ I#OZ[#rshadowtreecontextdatarr-hooknamer>r?r@cv[5(ag[5 Uc[OUnUc [5OUn[ U5n[ X45n[ U5 [R"XX#Xv5nUS:Xa[R"[5U5 US: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"]. For "roctracer", available options are [None]. For "instrumentation", available options are [None]. Each mode has a set of control knobs following with the mode name. For example, "pcsampling" has an "interval" control knob, expressed as "pcsampling:interval=1000". hook (str, optional): The hook to use for profiling. Available options are [None, "launch"]. Defaults to None. Returns: session (int): The session ID of the profiling session. Nrr/) rrDEFAULT_PROFILE_NAMErr,r2r: libprotonstartr registerr r ) rAr>r?rr-r@ backend_pathmode_strsessions rrErE7s^#'< TD#*?oG,W5LW+HwoodTHSG xZ\73##06@ NrrIc[5(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)rrr activaterD activate_allrIs rrKrK~sLW\`aa! 7#rc[5(aUS:wa [S5e[R"U5 Uc[R "5 g[R"U5 g)z 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) Returns: None rzGOnly one session can be deactivated when running from the command line.N)rrr deactivaterDdeactivate_allrMs rrOrOsLW\bcc7#  "W%r output_formatc[R"U5 Uc![5 [R"U5 g[ 5(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", "chrome_trace"]. Returns: None NrzEOnly one session can be finalized when running from the command line.)r unregisterrrD finalize_allrrfinalize)rIrQs rrUrUsT7#}-   Ade 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$)Nr=)rErO) argskwargsrIretrr>r?funcr@r-rAs rwrapper_profiling..wrappers5gD'PT[_`D#F#7 r) functoolswraps)r[rAr>r?rr-r@r\s``````` r _profilingr`s,&__T NrrAr>r?rr-r@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$)Nrar`)frr>r?r@r-rAs r decoratorprofile..decoratorsadG$PW^bimn nrrard)r[rAr>r?rr-r@rfs `````` rprofilerhs96 | o o$T7w]ahlmmr)N)Nr )Nr;r<NNN)"r^rr6r&rtriton._C.libprotonrrDflagsrrrhooksr r r r-r typingr rrCr%rr,r2r:rEintrKrOrUr`rhrrrosK 3GG??"LL s s %3%huS(]/C&D%%D& !+/D 3-Dc]D 3- D c] D 5h' ( D 3-DN$hsm$t$,& &&,3hsm38C=3RV34% ! 3-c] 3-  c]  3-  3-< #n% !#n 3-#nc] #n 3- #n c] #n 3-#n 3-#nr