+
    ړLjJ                         ^ RI Ht ^ RIHt ^ RIt^RIHt RR R lltRR R lltRR	 R
 llt	RR R llt
RR R lltR# )    )Optional)protonN)flagsc                F    V ^8  d   QhR\         \        ,          R\        /#    sessionphaser   int)formats   "@/app/.local/lib/python3.14/site-packages/triton/profiler/data.py__annotate__r      s     : :# :3 :    c                    V f   R# \         P                  '       d   V ^ 8w  d   \        R4      h\        P                  ! \
        P                  ! W4      4      # )z
Retrieves profiling data for a given session.

Args:
    session (Optional[int]): The session ID of the profiling session, or None if profiling is inactive.
Returns:
    str: The profiling data in JSON format.
NEOnly one session can be retrieved when running from the command line.)r   command_line
ValueErrorjsonloads	libprotonget_datar	   r
   s   &&r   getr      sD     gl`aa::i((899r   c                F    V ^8  d   QhR\         \        ,          R\        /# r   r   )r   s   "r   r   r      s     6 6# 63 6r   c                    V f   R# \         P                  '       d   V ^ 8w  d   \        R4      h\        P                  ! W4      # )z
Retrieves profiling data for a given session encoded with MessagePack.

Args:
    session (Optional[int]): The session ID of the profiling session, or None if profiling is inactive.

Returns:
    bytes: The profiling data encoded with MessagePack.
Nr   )r   r   r   r   get_data_msgpackr   s   &&r   get_msgpackr      s;     gl`aa%%g55r   c                \    V ^8  d   QhR\         \        ,          R\         \        ,          /# )r   r	   returnr   )r   s   "r   r   r   (   s"     1 18C= 1# 1r   c                    V f   R# \         P                  '       d   V ^ 8w  d   \        R4      h\        P                  ! V 4      # )z
Advances the profiling phase for a given session.

Args:
    session (Optional[int]): The session ID of the profiling session, or None if profiling is inactive.

Returns:
    Optional[int]: The next phase number after advancing.
NzFOnly one session can advance phase when running from the command line.)r   r   r   r   advance_data_phase)r	   s   &r   advance_phaser#   (   s;     glabb''00r   c                R    V ^8  d   QhR\         \        ,          R\        R\        /# )r   r	   r
   r    r   r   bool)r   s   "r   r   r   9   s%     < <x} < <T <r   c                    V f   R# \         P                  '       d   V ^ 8w  d   \        R4      h\        P                  ! W4      # )a  
Checks if the profiling data for a given session and phase is complete.

A "complete" phase is safe to read/clear because all device-side records for
the phase have been flushed to the host and the phase will no longer receive
new records.

Args:
    session (Optional[int]): The session ID of the profiling session, or None if profiling is inactive.
    phase (int): The phase number to check. Defaults to 0.

Returns:
    bool: True if the phase data is complete, False otherwise.
FzVOnly one session can check phase completion status when running from the command line.)r   r   r   r   is_data_phase_completer   s   &&r   is_phase_completer)   9   s;     glqrr++G;;r   c                V    V ^8  d   QhR\         \        ,          R\        R\        RR/# )r   r	   r
   clear_up_to_phaser    Nr%   )r   s   "r   r   r   O   s4     < <c]<< < 
	<r   c                    V f   R# \         P                  '       d   V ^ 8w  d   \        R4      h\        P                  ! WV4       R# )a+  
Clears profiling data for a given session.

Args:
    session (Optional[int]): The session ID of the profiling session, or None if profiling is inactive.
    phase (int): The phase number to clear. Defaults to 0.
    clear_up_to_phase (bool): If True, clear all phases up to and including `phase`.
NzCOnly one session can be cleared when running from the command line.)r   r   r   r   
clear_data)r	   r
   r+   s   &&&r   clearr.   O   s;     gl^__):;r   )r   r   )r   )r   r   F)typingr   triton._C.libprotonr   r   r   r   r   r   r#   r)   r.    r   r   <module>r2      s/     3  : 6"1"<,< <r   