wg ddlZddlZddlmZddlmZmZddlm Z m Z m Z ddl m Z dZdefd Z dd d ddd d e ede ede ede ede ef dZdde eddfdZdde eddfdZdde ededdfdZ dd e ede ede ede ede ef dZ ddd d dddd e ede ede ede ede ef dZy)N)proton)register_triton_hookunregister_triton_hook)set_profiling_offset_profiling_onis_command_line)Optionalrreturnctjjjj j }|dk(ry|dk(ryt d)Ncudacuptihip roctracerz/No backend is available for the current target.)tritonruntimedriveractiveget_current_targetbackend ValueError)rs \/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/triton/profiler/profile.py_select_backendr sHnn##**==?GGG& E JKKshadowtreecontextdatarhooknamerrrr ctry|t}| t}t|r|dk(r t t j ||||S)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 ["cupti"]. 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_NAMErrr libprotonstartr!rrrr s rr%r%sPJ |#!#   ??4$ 88rsessioncbtr|dk7r tdtj|y)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 rr)r)Ks*W\`aa wrcbtr|dk7r tdtj|y)a 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$ deactivater*s rr,r,[s*W\bcc !r output_formatc|*ttj|tyt r|dk7r t dtj ||y)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'r-s rr0r0ksK}-  Ade e7M2rcVtjfd}|S)z Context manager for profiling. Internally use only. Args: See start() for the arguments. Returns: wrapper (function): The wrapped function. cNt }|i|}t||S)Nr)r%r,) argskwargsr'retrrrfuncr r!s rwrapperz_profiling..wrappers2gD'PTUD#F#7 r) functoolswraps)r6r!rrrr r7s`````` r _profilingr:s,$__T Nrr&cH| fd}|St|S)z Decorator for profiling. Usage: ```python @proton.profile def foo(): pass ``` Args: See start() for the arguments. Returns: decorator (function): The decorator function. c&t|S)Nr&r:)frrrr r!s r decoratorzprofile..decoratorsadG$PW^bc crr&r=)r6r!rrrr r?s ````` rprofiler@s64 | d d$T7w]abbr)N)r)Nhatchet)NrrNN)r8rtriton._C.libprotonrr$r rrflagsrrr typingr r#strrr%intr)r,r0r:r@rrrHs 3>GGLL29& ! 29 3-29c]29 3- 29 c] 29 3- 29j  hsm  D  " "d " 3hsm333t32% !  3-c] 3-  c]  3- : "c% !"c 3-"cc] "c 3- "c c] "c 3-"cr