o
    پiW                     @   s:   d dl Z d dlZd dlmZ ejG dd dZe ZdS )    N)contextmanagerc                   @   s   e Zd ZU dZeed< dZeed< dZeed< dZeed< dZ	eed< dZ
eed< dZeed< dZeed	< dZeed
< dZeed< dZeed< dZeed< dddZedd ZdS )CompilationCounterr   num_models_seennum_graphs_seennum_piecewise_graphs_seen$num_piecewise_capturable_graphs_seennum_backend_compilationsnum_gpu_runner_capture_triggersnum_cudagraph_capturednum_inductor_compilesnum_eager_compilesnum_cache_entries_updatednum_compiled_artifacts_saveddynamo_as_is_countreturnc                 C   s
   t | S )N)copydeepcopy)self r   ^/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/compilation/compilation_counter.pyclone    s   
zCompilationCounter.clonec              	   k   sh    |   }d V  | D ]%\}}t| |t|| |ks1J | dt|| dt| | d| qd S )Nz not as expected, before it is z, after it is z, expected diff is )r   itemsgetattr)r   kwargsoldkvr   r   r   expect#   s   zCompilationCounter.expectN)r   r   )__name__
__module____qualname__r   int__annotations__r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r      s    
 
r   )r   dataclasses
contextlibr   	dataclassr   compilation_counterr   r   r   r   <module>   s   
&