Skip to content

Commit ef2c4dd

Browse files
cehongwangChen Fu
andauthored
Fixed python only runtime bug (#3701)
Co-authored-by: Chen Fu <[email protected]>
1 parent 89b629a commit ef2c4dd

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

py/torch_tensorrt/dynamo/debug/_Debugger.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
from unittest import mock
99

1010
import torch
11+
from torch_tensorrt._features import ENABLED_FEATURES
1112
from torch_tensorrt.dynamo._defaults import DEBUG_LOGGING_DIR
1213
from torch_tensorrt.dynamo.debug._DebuggerConfig import DebuggerConfig
1314
from torch_tensorrt.dynamo.debug._supports_debugger import (
@@ -93,7 +94,8 @@ def __init__(
9394

9495
def __enter__(self) -> None:
9596
self.original_lvl = _LOGGER.getEffectiveLevel()
96-
self.rt_level = torch.ops.tensorrt.get_logging_level()
97+
if ENABLED_FEATURES.torch_tensorrt_runtime:
98+
self.rt_level = torch.ops.tensorrt.get_logging_level()
9799
dictConfig(self.get_logging_config(self.log_level))
98100

99101
if self.capture_fx_graph_before or self.capture_fx_graph_after:
@@ -144,7 +146,8 @@ def __enter__(self) -> None:
144146
def __exit__(self, exc_type: Any, exc_value: Any, exc_tb: Any) -> None:
145147

146148
dictConfig(self.get_logging_config(None))
147-
torch.ops.tensorrt.set_logging_level(self.rt_level)
149+
if ENABLED_FEATURES.torch_tensorrt_runtime:
150+
torch.ops.tensorrt.set_logging_level(self.rt_level)
148151
if self.capture_fx_graph_before or self.capture_fx_graph_after:
149152
ATEN_PRE_LOWERING_PASSES.passes, ATEN_POST_LOWERING_PASSES.passes = (
150153
self.old_pre_passes,

0 commit comments

Comments
 (0)