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 8
8
from unittest import mock
9
9
10
10
import torch
11
+ from torch_tensorrt ._features import ENABLED_FEATURES
11
12
from torch_tensorrt .dynamo ._defaults import DEBUG_LOGGING_DIR
12
13
from torch_tensorrt .dynamo .debug ._DebuggerConfig import DebuggerConfig
13
14
from torch_tensorrt .dynamo .debug ._supports_debugger import (
@@ -93,7 +94,8 @@ def __init__(
93
94
94
95
def __enter__ (self ) -> None :
95
96
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 ()
97
99
dictConfig (self .get_logging_config (self .log_level ))
98
100
99
101
if self .capture_fx_graph_before or self .capture_fx_graph_after :
@@ -144,7 +146,8 @@ def __enter__(self) -> None:
144
146
def __exit__ (self , exc_type : Any , exc_value : Any , exc_tb : Any ) -> None :
145
147
146
148
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 )
148
151
if self .capture_fx_graph_before or self .capture_fx_graph_after :
149
152
ATEN_PRE_LOWERING_PASSES .passes , ATEN_POST_LOWERING_PASSES .passes = (
150
153
self .old_pre_passes ,
You can’t perform that action at this time.
0 commit comments