
    .`i3              	           d dl Z d dlmZ d dlmZmZmZ d Zd Z G d d          Z		 dd	e
d
edeege
f         e
z  de
fdZdedefdZdedefdZde
fdZdede
fdZdede
fdZdedeegef         de
fdZdS )    N)Callable)
_EventType_ProfilerEvent_TensorMetadatac                     t          |           |k    r?t          |           |z
  dz   }| |d          } t          |           dk    rd| dd          z   } | S N   z...lenstringwidthoffsets      g/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/profiler/utils.pytrim_string_frontr      s[    
6{{UVu$q(v;;??VABBZ'FM    c                     t          |           |k    r8t          |           |z
  dz   }| d |          } t          |           dk    r| dz   } | S r   r
   r   s      r   trim_string_backr      sU    
6{{UVu$q(&!v;;??e^FMr   c                   x    e Zd Zdeej                 deeef         fdZ	de
ej                 fdZd Zd Zd Zd	S )
TablePrinterrow_clscolumn_widthsc                     || _         d t          j        |          D             | _        || _        t          | j                                                  t          | j                  k    sJ d S )Nc                     g | ]	}|j         
S  )name.0xs     r   
<listcomp>z)TablePrinter.__init__.<locals>.<listcomp>%   s    GGGa16GGGr   )r   dataclassesfields
fieldnamesr   setkeys)selfr   r   s      r   __init__zTablePrinter.__init__!   sk     GG;+=g+F+FGGG*4%**,,--T_1E1EEEEEEEr   rowsc                     |                                   |                                  |D ]}|                     |           d S N)_print_header_print_line
_print_row)r&   r(   rows      r   print_tablezTablePrinter.print_table)   sT     	! 	!COOC    	! 	!r   c                     t          | j                  D ]c\  }}|t          | j                  dz
  k    }| j        |         }t	          t          ||                              |          |sdnd           dd S )N    | 
end)	enumerater#   r   r   printr   ljust)r&   iflast	col_widths        r   r+   zTablePrinter._print_header/   s    do.. 	 	DAqDO,,q00D*1-I I..44Y??!%/EE4    	 	r   c                 (   t          || j                  sJ t          | j                  D ]\  }}|t	          | j                  dz
  k    }| j        |         }t          ||          }d}t          |t                    r$t          ||          	                    |          }nXt          |          t          t          fv r%t          |          d                    |          }n|                     |          }t          ||sdnd           d S )Nr1    z>.2fr2   r3   r4   )
isinstancer   r6   r#   r   r   getattrstrr   r8   typefloatintrjustr7   )r&   r.   r9   r:   r;   r<   valval_strs           r   r-   zTablePrinter._print_row8   s   #t|,,,,,do.. 	< 	<DAqDO,,q00D*1-I#q//CG#s## 4*3	::@@KKcucl**"3ZZ..44Y?? (..33'D:uud;;;;;	< 	<r   c                     d}| j                                         D ]}||z  }t          d|dt          | j                   dz
  z  z   z             d S )Nr   =r	   r1   )r   valuesr7   r   )r&   total_col_widthcolumn_widths      r   r,   zTablePrinter._print_lineI   sd     .5577 	, 	,L|+OOc_qC0B,C,Ca,G'HHIJJJJJr   N)__name__
__module____qualname__rB   r!   	dataclassdictrA   rD   r'   listr/   r+   r-   r,   r   r   r   r   r       s        FK12FCGS>F F F F![%: ; ! ! ! !  < < <"K K K K Kr   r    r   indentindent_stylereturnc                 `    |r+t          |t                    r||z  | z   S  ||          | z   S | S r*   )r?   rA   )r   rT   rU   s      r   indent_stringrX   P   sH      lC(( 	1&(611<''&00r   eventc                 L    | j         \  }}|t          j        k    r	|j        d uS dS )NF)typedr   PyCallmodule)rY   
event_typetyped_events      r   event_has_moduler`   a   s/    #kJZ&&&!--5r   c                 ,    | j         t          j        k    S r*   )tagr   TorchOp)rY   s    r   event_is_torch_oprd   h   s    9
***r   c                 @   | )t          |           t          t          t          t          fv r|  S t          | t                    r#dd                    d | D                        dS t          | t                    r#dd                    d | D                        dS t          | t                    sJ dt          |                        d                    d	 | j
        D                       }t	          | j                                      d
d           d| dS )N[, c                 ,    g | ]}t          |          S r   event_arg_reprr   s     r   r    z"event_arg_repr.<locals>.<listcomp>p        ===AnQ//===r   ](c                 ,    g | ]}t          |          S r   ri   r   s     r   r    z"event_arg_repr.<locals>.<listcomp>r   rk   r   )zUnsupported type: c                 ,    g | ]}t          |          S r   )rA   r   s     r   r    z"event_arg_repr.<locals>.<listcomp>u   s    999!s1vv999r   ztorch.r>   )rB   rC   rD   boolrA   r?   rR   jointupler   sizesdtypereplace)arg	sizes_strs     r   rj   rj   l   s#   
{d3iiE3c#:::x	C		 FA499=====>>AAAA	C		 FA499=====>>AAAA#//QQ1Qd3ii1Q1QQQ/II99sy999::	ci..((266EEEEEEr   c                     | j         t          j        k    sJ d                    d | j        d         j        D                       }| j         d| d                    dd          S )Nrg   c                 ,    g | ]}t          |          S r   ri   r   s     r   r    z'event_torch_op_repr.<locals>.<listcomp>{   s     KKK.++KKKr   r1   rm   ro   zaten::r>   )rb   r   rc   rr   r[   inputsr   rv   )rY   args_strs     r   event_torch_op_reprr}   y   sj    9
*****yyKKU[^5JKKKLLHj&&8&&&..x<<<r   c                     t          |           sJ | j        d         j        }|j        rIt	          |j                  dk    r1d                    d |j        D                       }|j         d| dS |j        S )Nr1   r   rg   c                 N    g | ]"}|d           dt          |d                    #S )r   rI   r1   ri   r   s     r   r    z%event_module_repr.<locals>.<listcomp>   s6    III!!--~ad++--IIIr   rm   ro   )r`   r[   r]   
parametersr   rr   cls_name)rY   r]   r|   s      r   event_module_reprr      s    E"""""[^"F S!233a7799IIv7HIII
 
 ///H////r   
curr_eventuntilc                     d}| j         } | rX ||           sMt          |           r*t          |          dk    r|dz  }|t          |           z  }| j         } | r ||           M|S )Nr>   r   z <- )parentrd   r   r}   )r   r   traces      r   event_torch_op_stack_tracer      s     E"J
 'UU:.. 'Z(( 	55zzA~~(444E&
  'UU:.. ' Lr   )rS   )r!   collections.abcr   torch._C._profilerr   r   r   r   r   r   rA   rD   rX   rq   r`   rd   rj   r}   r   r   r   r   r   <module>r      s       $ $ $ $ $ $ J J J J J J J J J J    -K -K -K -K -K -K -K -Kb JM	 			,4cUCZ,@3,F		 	 	 	"N t    +^ + + + + +
F3 
F 
F 
F 
F=~ =# = = = =	^ 	 	 	 	 	'/0@$0F'G     r   