File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed
py/torch_tensorrt/dynamo/debug Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change 88from unittest import mock
99
1010import torch
11+ from torch_tensorrt ._features import ENABLED_FEATURES
1112from torch_tensorrt .dynamo ._defaults import DEBUG_LOGGING_DIR
1213from torch_tensorrt .dynamo .debug ._DebuggerConfig import DebuggerConfig
1314from 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 ,
You can’t perform that action at this time.
0 commit comments