^jJ ddlmZddlmZddlZddlmZddeedefdZddeedefd Z ddeed eefd Z ddeeded e fd Z ddeeded e d dfdZ y))Optional)protonN)flagssessionphasec|ytjr|dk7r tdtjt j ||S)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. NrEOnly one session can be retrieved when running from the command line.)r command_line ValueErrorjsonloads libprotonget_datarrs _/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/triton/profiler/data.pygetrsC gl`aa ::i((%8 99ct|ytjr|dk7r tdtj||S)a 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. Nrr )rr r rget_data_msgpackrs r get_msgpackrs: gl`aa  % %gu 55rreturncr|ytjr|dk7r tdtj|S)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. NrzFOnly one session can advance phase when running from the command line.)rr r radvance_data_phase)rs r advance_phaser(s8 glabb  ' ' 00rct|ytjr|dk7r tdtj||S)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. FrzVOnly one session can check phase completion status when running from the command line.)rr r ris_data_phase_completers ris_phase_completer9s: glqrr  + +GU ;;rclear_up_to_phasecx|ytjr|dk7r tdtj|||y)aC 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`. NrzCOnly one session can be cleared when running from the command line.)rr r r clear_data)rrrs rclearr"Os: gl^__ %):;r)rr)r)rrF)typingrtriton._C.libprotonrrr rintrrrboolrr"rrr(s3 :# :3 : 6#636"18C=1#1"