diff --git a/src/forge/observability/perf_tracker.py b/src/forge/observability/perf_tracker.py index 12c5e904a..c3bf137b9 100644 --- a/src/forge/observability/perf_tracker.py +++ b/src/forge/observability/perf_tracker.py @@ -15,7 +15,11 @@ import torch -from forge.env_constants import DISABLE_PERF_METRICS, METRIC_TIMER_USES_GPU +from forge.env_constants import ( + DISABLE_PERF_METRICS, + FORGE_DISABLE_METRICS, + METRIC_TIMER_USES_GPU, +) from forge.observability.metrics import record_metric, Reduce logger = logging.getLogger(__name__) @@ -108,7 +112,10 @@ def __init__( self.prefix = prefix self.track_memory = track_memory self.time_with_gpu = timer == "gpu" - self._disable = os.getenv(DISABLE_PERF_METRICS, "false") == "true" + self._disable = ( + os.getenv(DISABLE_PERF_METRICS, "false").lower() == "true" + or os.getenv(FORGE_DISABLE_METRICS, "false").lower() == "true" + ) self._active = False # Timing state