pytorch - 💡(How to fix) Fix [dynamo] Display tracked side effects on side-effect related issues [1 participants]

import logging

# Set up logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# Define a custom logger for side effects
side_effect_logger = logging.getLogger('side_effects')
side_effect_logger.setLevel(logging.INFO)

# Modify the code to log side effects at the tracking point
def track_side_effects(side_effects):
    # Log side effects
    side_effect_logger.info('Tracked side effects: %s', side_effects)
    # ... (rest of the codegen logic remains the same)

# Update the logging configuration to write to TORCH_LOGS
logging.config.dictConfig({
    'version': 1,
    'formatters': {
        'default': {
            'format': '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
        }
    },
    'handlers': {
        'console': {
            'class': 'logging.StreamHandler',
            'formatter': 'default',
            'stream': 'ext://sys.stdout'
        },
        'file': {
            'class': 'logging.FileHandler',
            'formatter': 'default',
            'filename': 'TORCH_LOGS="side_effects"'
        }
    },
    'root': {
        'level': 'INFO',
        'handlers': ['console', 'file']
    }
})