Files
iGavroche-rocm-ninodes/rocm_nodes/utils/debug.py
T

56 lines
1.4 KiB
Python

"""
Debug utilities for ROCM Ninodes.
Provides debug functions that can be enabled/disabled via environment variables
for performance profiling and data capture during development.
"""
import os
from typing import Any
# Debug mode is controlled by environment variable
DEBUG_MODE = os.environ.get('ROCM_NINODES_DEBUG', '0') == '1'
def save_debug_data(*args: Any, **kwargs: Any) -> None:
"""Save debug data to disk (only active when DEBUG_MODE=True)"""
if not DEBUG_MODE:
return
# Implementation would go here for actual data capture
pass
def capture_timing(*args: Any, **kwargs: Any) -> None:
"""Capture timing information (only active when DEBUG_MODE=True)"""
if not DEBUG_MODE:
return
# Implementation would go here for timing capture
pass
def capture_memory_usage(*args: Any, **kwargs: Any) -> None:
"""Capture memory usage information (only active when DEBUG_MODE=True)"""
if not DEBUG_MODE:
return
# Implementation would go here for memory capture
pass
def log_debug(*args: Any, **kwargs: Any) -> None:
"""Log debug information (only active when DEBUG_MODE=True)"""
if not DEBUG_MODE:
return
# Implementation would go here for debug logging
pass
__all__ = [
'DEBUG_MODE',
'save_debug_data',
'capture_timing',
'capture_memory_usage',
'log_debug',
]