From 3b07d814f5e0a8803118602c596881bb84168418 Mon Sep 17 00:00:00 2001 From: Chen Fu Date: Mon, 21 Jul 2025 22:22:42 +0000 Subject: [PATCH] Fixed python only runtime bug --- py/torch_tensorrt/dynamo/debug/_Debugger.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/py/torch_tensorrt/dynamo/debug/_Debugger.py b/py/torch_tensorrt/dynamo/debug/_Debugger.py index be5bea358b..ec624ffc5a 100644 --- a/py/torch_tensorrt/dynamo/debug/_Debugger.py +++ b/py/torch_tensorrt/dynamo/debug/_Debugger.py @@ -8,6 +8,7 @@ from unittest import mock import torch +from torch_tensorrt._features import ENABLED_FEATURES from torch_tensorrt.dynamo._defaults import DEBUG_LOGGING_DIR from torch_tensorrt.dynamo.debug._DebuggerConfig import DebuggerConfig from torch_tensorrt.dynamo.debug._supports_debugger import ( @@ -93,7 +94,8 @@ def __init__( def __enter__(self) -> None: self.original_lvl = _LOGGER.getEffectiveLevel() - self.rt_level = torch.ops.tensorrt.get_logging_level() + if ENABLED_FEATURES.torch_tensorrt_runtime: + self.rt_level = torch.ops.tensorrt.get_logging_level() dictConfig(self.get_logging_config(self.log_level)) if self.capture_fx_graph_before or self.capture_fx_graph_after: @@ -144,7 +146,8 @@ def __enter__(self) -> None: def __exit__(self, exc_type: Any, exc_value: Any, exc_tb: Any) -> None: dictConfig(self.get_logging_config(None)) - torch.ops.tensorrt.set_logging_level(self.rt_level) + if ENABLED_FEATURES.torch_tensorrt_runtime: + torch.ops.tensorrt.set_logging_level(self.rt_level) if self.capture_fx_graph_before or self.capture_fx_graph_after: ATEN_PRE_LOWERING_PASSES.passes, ATEN_POST_LOWERING_PASSES.passes = ( self.old_pre_passes,