Ϧi SSKrSSKrSSKrSSKJr SSKJrJr SSK J r J r J r SSK Jr SrS\4S jrS \SS4S jrSS S SSS.S\\S\\S\\S \\S\\4 SjjjrSS\\SS4SjjrSS\\SS4SjjrSS\\S\SS4SjjrSS\\S\\S\\S \\S\\4 SjjrSSS S SSS.S\\S\\S\\S \\S\\4 Sjjjrg) N)proton)register_triton_hookunregister_triton_hook)set_profiling_offset_profiling_onis_command_line)Optionalrreturnc[RRRR 5R nUS:XagUS:Xag[ S5e)Ncudacuptihip roctracerz/No backend is available for the current target.)tritonruntimedriveractiveget_current_targetbackend ValueError)rs V/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/triton/profiler/profile.py_select_backendr sHnn##**==?GGG& E JKKrczUS: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[#rshadowtreecontextdatarhooknamer&r'r(c[5(agUc[nUc [5n[U5 [ 5 U(aUS:Xa [ 5 [ R"XX#5$)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.hatchet". backend (str, optional): The backend to use for profiling. Available options are [None, "cupti", "cupti_pcsampling", "roctracer"]. Defaults to None, which automatically selects the backend matching the current active runtime. 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"]. Defaults to "tree". hook (str, optional): The hook to use for profiling. Available options are [None, "triton"]. Defaults to None. Returns: session (int): The session ID of the profiling session. Nr)r DEFAULT_PROFILE_NAMErr"rr libprotonstartr)r&r'rr(s rr-r-!sWJ |#!#w   ??4$ 88rsessioncp[5(aUS:wa [S5e[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 0 (the first session started.) Returns: None rzEOnly one session can be activated when running from the command line.N)r rr,activater/s rr1r1Xs-W\`aa wrcp[5(aUS:wa [S5e[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 0 (the first session started.) Returns: None rzGOnly one session can be deactivated when running from the command line.N)r rr, deactivater2s rr4r4hs-W\bcc !r output_formatcUc+[5 [R"U5 [5 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. Aavailable options are ["hatchet"]. Returns: None NrzEOnly one session can be finalized when running from the command line.)rr, finalize_allrr rfinalize)r/r5s rr8r8xsL}-   Ade e72rcZ^^^^^^[R"T5UUUUUU4Sj5nU$)z Context manager for profiling. Internally use only. Args: See start() for the arguments. Returns: wrapper (function): The wrapped function. cH>[T TTTTS9nT"U0UD6n[U5 U$)Nr%)r-r4) argskwargsr/retrr&r'funcr(r)s rwrapper_profiling..wrappers2gD'PTUD#F#7 r) functoolswraps)r>r)r&r'rr(r?s`````` r _profilingrCs,$__T Nrr.c F^^^^^Uc UUUUU4SjnU$[UTTTTTS9$)z Decorator for profiling. Usage: ```python @proton.profile def foo(): pass ``` Args: See start() for the arguments. Returns: decorator (function): The decorator function. c  >[UTTTTTS9$)Nr.rC)frr&r'r(r)s r decoratorprofile..decoratorsadG$PW^bc crr.rF)r>r)r&r'rr(rHs ````` rprofilerJs64 | d d$T7w]abbr)N)r)Nhatchet)Nr#r$NN)rArrtriton._C.libprotonrr,r(rrflagsrrr typingr r+strrr"r-intr1r4r8rCrJrrrRs 3>GGLL49& ! 49 3-49c]49 3- 49 c] 49 3- 49n  hsm  D  " "d " 3hsm333t32% !  3-c] 3-  c]  3- : "c% !"c 3-"cc] "c 3- "c c] "c 3-"cr