
    h                     ,    S SK r S SKJr   " S S5      rg)    N)cudartc                   0    \ rS rSrS rSS jrS rS rSrg)	
NvtxHelper	   c                     Xl         0 U l        U H7  nS H.  n[        R                  " 5       S   U R                  US-   U-   '   M0     M9     0 U l        g )N)startstop   -)stageseventsr   cudaEventCreatemarkers)selfr   stagemarkers       j/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/transformers/models/sam2/nvtx_helper.py__init__NvtxHelper.__init__
   sP    E+4:4J4J4LQ4OECK&01 ,      c                     [         R                  " XS9U R                  U'   US-   nX0R                  ;   a%  [        R
                  " U R                  U   S5        g g )N)messagecolor-startr   )nvtxstart_ranger   r   r   cudaEventRecord)r   r   r   
event_names       r   start_profileNvtxHelper.start_profile   sM    "..uJUX%
$""4;;z#:A> %r   c                     US-   nX R                   ;   a$  [        R                  " U R                   U   S5        [        R                  " U R
                  U   5        g )N-stopr   )r   r   r   r   	end_ranger   )r   r   r   s      r   stop_profileNvtxHelper.stop_profile   sC    W_
$""4;;z#:A>t||E*+r   c                     U R                    HO  n[        R                  " U R                  U S3   U R                  U S3   5      S   n[	        U SUS S35        MQ     g )Nr   r"   r
   z: z.2fz ms)r   r   cudaEventElapsedTimer   print)r   r   latencys      r   print_latencyNvtxHelper.print_latency   se    [[E11$++v>N2OQUQ\Q\`e_ffk]lQmnopqGUG2gc]#./ !r   )r   r   r   N)blue)	__name__
__module____qualname____firstlineno__r   r   r$   r*   __static_attributes__ r   r   r   r   	   s    ?,0r   r   )r   cudar   r   r2   r   r   <module>r4      s   
  0 0r   