o
    پi                     @   s>   d dl mZmZmZ g Zddee fddZG dd dZdS )	    )CallableListOptionalNop_namec                    s   dt f fdd}|S )Nop_funcc                    s    p| j }td|  | S )Nzsglang.)__name__	SPLIT_OPSappend)r   namer    ]/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/compilation/compilation_config.py	decorator	   s   
z$register_split_op.<locals>.decorator)r   )r   r   r   r   r   register_split_op   s   r   c                   @   s`   e Zd Z		ddee dedefddZdefd	d
ZdefddZ	dd Z
dd Zdd ZdS )CompilationConfigeagerFcapture_sizescompilerenable_debug_modec                 C   s0   t  | _|| _|| _|| _g | _| jt d S N)settraced_filesr   r   r   	split_opsextendr   )selfr   r   r   r   r   r   __init__   s   zCompilationConfig.__init__opc                 C      | j | d S r   )r   r	   )r   r   r   r   r   add_split_op       zCompilationConfig.add_split_op	file_pathc                 C   r   r   )r   add)r   r    r   r   r   add_traced_file#   r   z!CompilationConfig.add_traced_filec                 C      | j S r   )r   r   r   r   r   get_traced_files&      z"CompilationConfig.get_traced_filesc                 C   r#   r   )r   r$   r   r   r   get_capture_sizes)   r&   z#CompilationConfig.get_capture_sizesc                 C   r#   r   )r   r$   r   r   r   get_enable_debug_mode,   r&   z'CompilationConfig.get_enable_debug_modeN)r   F)r   
__module____qualname__r   intstrboolr   r   r"   r%   r'   r(   r   r   r   r   r      s    
r   r   )typingr   r   r   r   r,   r   r   r   r   r   r   <module>   s   
