
    wiP                         d dl Z d dlZd dlZddlmZmZmZ ddlmZ d Z	d Z
d Zd Zd	 Zed
k    r e             dS dS )    N   )startfinalize_select_backend)set_command_linec            	         t          j        ddt           j                  } |                     ddt          d           |                     dd	t          d
d g d           |                     ddt          ddddg           |                     ddt          dddg           |                     ddt          dd d dg           |                     dt           j        d           |                                 }||j        fS )NzBThe proton command utility for profiling scripts and pytest tests.z
    proton [options] script.py [script_args] [script_options]
    proton [options] pytest [pytest_args] [script_options]
    python -m triton.profiler.proton [options] script.py [script_args] [script_options]
)descriptionusageformatter_classz-nz--namezName of the profiling session)typehelpz-bz	--backendzProfiling backend)cupticupti_pcsampling	roctracer)r   r   defaultchoicesz-cz	--contextzProfiling contextshadowpythonz-dz--datazProfiling datatreez-kz--hookzProfiling hooktritontarget_argszSubcommand and its arguments)nargsr   )argparseArgumentParserRawTextHelpFormatteradd_argumentstr	REMAINDER
parse_argsr   )parserargss     f/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/triton/profiler/proton.pyparse_argumentsr#      s?   $X a 24 4 4F hS7VWWW
k:MW[ J J J  L L L
k:MW_!)8 4  6 6 6
hS7GQWbhaijjj
hS7GQU`dfn_oppp
X-?FdeeeD!!!    c                 H    t           j                            |           dk    S )Npytest)ospathbasename)scripts    r"   	is_pytestr+      s    7F##x//r$   c                 v   t           j                            |           }ddd|dt          t          j        t          i}t          j        }| g|z   t          _        t          j                            t           j                            |                     	 t          |d          5 }t          |                                |d          }d d d            n# 1 swxY w Y   t          ||           n)# t          $ r}t          d|            Y d }~nd }~ww xY w|t          _        d S # |t          _        w xY w)N__name____main____file____builtins__rbexecz.An error occurred while executing the script: )r'   r(   abspathr0   sysr-   argvappenddirnameopencompilereadr2   	Exceptionprint)r*   r!   script_pathclean_globalsoriginal_argvfilecodees           r"   execute_as_mainrC      sh   '//&))K 	JKc	M HMx$CHHOOBGOOK00111!+t$$ 	=499;;V<<D	= 	= 	= 	= 	= 	= 	= 	= 	= 	= 	= 	= 	= 	= 	=T=!!!! D D DBqBBCCCCCCCCD !=    sT   C3 '$CC3 CC3 CC3 2D* 3
D=DD* DD* *D8c                    | j         r| j         nt                      }t          | j        | j        | j        || j                   t                       |d         }t          |          dk    r
|dd          ng }t          |          rdd l
}|                    |           nt          ||           t                       d S )N)contextdatabackendhookr   r   )rG   r   r   namerE   rF   rH   r   lenr+   r&   mainrC   r   )r!   r   rG   r*   script_argsr&   s         r"   run_profilingrM   9   s    "lAdll0A0AG	$)T\	7QUQZ[[[[ ^F%(%5%5%9%9+abb//rK -K    ,,,JJJJJr$   c                  H    t                      \  } }t          | |           d S )N)r#   rM   )r!   r   s     r"   rK   rK   L   s(    '))D+$$$$$$r$   r.   )r   r4   r'   profiler   r   r   flagsr   r#   r+   rC   rM   rK   r-    r$   r"   <module>rR      s     



 				 5 5 5 5 5 5 5 5 5 5 # # # # # #" " "&0 0 0! ! !4  &% % %
 zDFFFFF r$   