Ë
    cQêjÿ  ã                   ó,   — d dl Z d dlmZ  G d„ d«      Zy)é    N)Úcudartc                   ó&   — e Zd Zd„ Zdd„Zd„ Zd„ Zy)Ú
NvtxHelperc                 óž   — || _         i | _        |D ]3  }dD ],  }t        j                  «       d   | j                  |dz   |z   <   Œ. Œ5 i | _        y )N)ÚstartÚstopé   ú-)ÚstagesÚeventsr   ÚcudaEventCreateÚmarkers)Úselfr   ÚstageÚmarkers       úu/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/onnxruntime/transformers/models/sam2/nvtx_helper.pyÚ__init__zNvtxHelper.__init__
   s`   € ØˆŒØˆŒØò 	PˆEØ+ò P�Ü4:×4JÑ4JÓ4LÈQÑ4O�—‘˜E C™K¨&Ñ0Ò1ñPð	Pð ˆ�ó    c                 óº   — t        j                  ||¬«      | j                  |<   |dz   }|| j                  v r$t	        j
                  | j                  |   d«       y y )N)ÚmessageÚcolorú-startr   )ÚnvtxÚstart_ranger   r   r   ÚcudaEventRecord)r   r   r   Ú
event_names       r   Ústart_profilezNvtxHelper.start_profile   sQ   € Ü"×.Ñ.°uÀEÔJˆ�‰�UÑØ˜XÑ%ˆ
Ø˜Ÿ™Ñ$Ü×"Ñ" 4§;¡;¨zÑ#:¸AÕ>ð %r   c                 ó´   — |dz   }|| j                   v r#t        j                  | j                   |   d«       t        j                  | j
                  |   «       y )Nú-stopr   )r   r   r   r   Ú	end_ranger   )r   r   r   s      r   Ústop_profilezNvtxHelper.stop_profile   sE   € Ø˜W‘_ˆ
Ø˜Ÿ™Ñ$Ü×"Ñ" 4§;¡;¨zÑ#:¸AÔ>Ü�‰�t—|‘| EÑ*Õ+r   c                 ó¼   — | j                   D ]M  }t        j                  | j                  |› d�   | j                  |› d�   «      d   }t	        |› d|d›d�«       ŒO y )Nr   r   r	   z: z.2fz ms)r   r   ÚcudaEventElapsedTimer   Úprint)r   r   Úlatencys      r   Úprint_latencyzNvtxHelper.print_latency   sj   € Ø—[‘[ò 	0ˆEÜ×1Ñ1°$·+±+ÀÀÀvÐ>NÑ2OÐQU×Q\ÑQ\Ð`eÐ_fÐfkÐ]lÑQmÓnÐopÑqˆGÜ�U�G˜2˜g c˜]¨#Ð.Õ/ñ	0r   N)Úblue)Ú__name__Ú
__module__Ú__qualname__r   r   r!   r&   © r   r   r   r   	   s   „ òó?ò,ó0r   r   )r   Úcudar   r   r+   r   r   ú<module>r-      s   ðó
 Ý ÷0ò 0r   