
    ϦiP                     v    S SK r S SKrS SKrSSKJrJrJr  SSKJr  S r	S r
S rS rS	 r\S
:X  a  \" 5         gg)    N   )startfinalize_select_backend)set_command_linec            	         [         R                  " SS[         R                  S9n U R                  SS[        SS9  U R                  SS	[        S
S / SQS9  U R                  SS[        SSSS/S9  U R                  SS[        SSS/S9  U R                  SS[        SS S S/S9  U R                  S[         R
                  SS9  U R                  5       nXR                  4$ )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     U/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/triton/profiler/proton.pyparse_argumentsr#      s   $$X a 224F hS7VW
k:MW[ J  L
k:MW_!)8 4  6
hS7GQWbhaij
hS7GQU`dfn_op
X-?-?FdeD!!!!    c                 F    [         R                  R                  U 5      S:H  $ )Npytest)ospathbasename)scripts    r"   	is_pytestr+      s    77F#x//r$   c                 T   [         R                  R                  U 5      nSSSUS[        [        R
                  [        0n[        R                  nU /U-   [        l        [        R                  R                  [         R                  R                  U5      5         [        US5       n[        UR                  5       US5      nS S S 5        [        WU5        U[        l        g ! , (       d  f       N&= f! [         a  n[        SU 35         S nAN:S nAff = f! U[        l        f = f)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      s    ''//&)K 	JKc	M HHMx$CHHHOOBGGOOK01!+t$499;V<D %T=! ! %$  D>qcBCCD !sB   C5 (C$C5 $
C2.C5 5
D?DD DD D'c                 z   U R                   (       a  U R                   O	[        5       n[        U R                  U R                  U R
                  X R                  S9  [        5         US   n[        U5      S:  a  USS  O/ n[        U5      (       a  SS K
nUR                  U5        O[        X45        [        5         g )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    "lldll0AG	$))T\\		7QZQZ[ ^F%(%5%9+ab/rKK ,Jr$   c                  2    [        5       u  p[        X5        g )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      sD     
 	 5 5 #"&0!4&%
 zF r$   