~L idZddlZddlZddlmZmZddlmZmZdZ de ejfdZ ejdZe s ddlZd Zeed d Zd ejdedfdZy#e$rd ZY*wxYw)aLogging 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)AnyCallable) _FaketqdmtqdmTreturnc~tjdtjdtjdgS)Nz%torch.fx.experimental.symbolic_shapesz torch._dynamoztorch._inductor)logging getLogger[/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/torch/_dynamo/logging.py get_loggersrs8AB/*+, r ztorch.compile())totaldescdelaylogger).NctsItjdttts tj j ttdtdtdtddffd }|S)NrlevelmsgkwargsrcDd|vrd|d<j|d|fi|y)N stacklevelrz Step %s: %s)log)rrrrsteps r rzget_step_logger..logCs. v %#$F<  5-s=f=r ) disable_progresspbarupdate isinstancerset_postfix_strnamenext _step_counterintstrr)rrrs` @r get_step_loggerr(;sa  A$ *  FKK= 2  D>3>S>C>D> Jr )__doc__ itertoolsr typingrr torch.hubrrrlistLoggerrcountr%triton num_steps ImportErrorrr(r r r r3s  %T'..)& "   i&7q AD GNN x /B   sA,,A65A6