Ë
    øÿæi§  ã                   óú   — d Z ddlZddlZddlmZ ddlmZ ddlmZm	Z	 dZ
deej                     fd„Z ej                  d	«      Ze
s	 ddlZd
Z e	edd¬«      Zdej                  ded   fd„Zy# e$ r dZY Œ*w xY w)aª  Logging utilities for Dynamo and Inductor.

This module provides specialized logging functionality including:
- Step-based logging that prepends step numbers to log messages
- Progress bar management for compilation phases
- Centralized logger management for Dynamo and Inductor components

The logging system helps track the progress of compilation phases and provides structured
logging output for debugging and monitoring.
é    N)ÚCallable)ÚAny)Ú	_FaketqdmÚtqdmTÚreturnc                  ó~   — t        j                  d«      t        j                  d«      t        j                  d«      gS )Nz%torch.fx.experimental.symbolic_shapesztorch._dynamoztorch._inductor)ÚloggingÚ	getLogger© ó    új/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/torch/_dynamo/logging.pyÚget_loggersr      s8   € ä×ÑÐAÓBÜ×Ñ˜/Ó*Ü×ÑÐ+Ó,ðð r   é   é   é   ztorch.compile())ÚtotalÚdescÚdelayÚlogger).Nc                 óü   ‡ ‡— t         sIt        j                  d«       t        t        t        «      s t        j                  ‰ j                  › «       t        t        «      Šdt        dt        dt        dd fˆ ˆfd„}|S )Nr   ÚlevelÚmsgÚkwargsr   c                 óD   •— d|vrd|d<    ‰j                   | d‰|fi |¤Ž y )NÚ
stacklevelr   zStep %s: %s)Úlog)r   r   r   r   Ústeps      €€r   r   zget_step_logger.<locals>.logD   s.   ø€ Ø˜vÑ%Ø#$ˆF�<Ñ Øˆ�
‰
�5˜-¨¨sÑ=°fÓ=r   )Údisable_progressÚpbarÚupdateÚ
isinstancer   Úset_postfix_strÚnameÚnextÚ_step_counterÚintÚstrr   )r   r   r   s   ` @r   Úget_step_loggerr(   <   sa   ù€ ÝÜ�‰�AŒÜœ$¤	Ô*Ü× Ñ  F§K¡K =Ô2ä”Ó€Dð>”3ð >œSð >¬Cð >°Dö >ð
 €Jr   )Ú__doc__Ú	itertoolsr	   Úcollections.abcr   Útypingr   Ú	torch.hubr   r   r   ÚlistÚLoggerr   Úcountr%   ÚtritonÚ	num_stepsÚImportErrorr   r(   r   r   r   Ú<module>r4      sŸ   ðñ	ó Û Ý $Ý ç %ð Ð ð�T˜'Ÿ.™.Ñ)ó ð&  �	—‘ Ó"€ñ ðÛàˆ	ñ �iÐ&7¸qÔA€Dð˜GŸN™Nð ¨x¸	Ñ/Bô øð ò ØŠ	ðús   Á	A0 Á0A:Á9A: