Skip to content

Commit b8c3abe

Browse files
committed
chore: invalid log level check
1 parent 9ed18a7 commit b8c3abe

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

py/torch_tensorrt/dynamo/utils.py

+3
Original file line numberDiff line numberDiff line change
@@ -204,6 +204,7 @@ def set_log_level(parent_logger: Any, level: Any) -> None:
204204
Sets the log level to the user provided level.
205205
This is used to set debug logging at a global level
206206
at entry points of tracing, dynamo and torch_compile compilation.
207+
It also set log level for c++ torch trt logger
207208
"""
208209
if parent_logger:
209210
parent_logger.setLevel(level)
@@ -218,6 +219,8 @@ def set_log_level(parent_logger: Any, level: Any) -> None:
218219
log_level = trt.ILogger.Severity.ERROR
219220
elif level == logging.CRITICAL:
220221
log_level = trt.ILogger.Severity.INTERNAL_ERROR
222+
else:
223+
raise AssertionError(f"{level} is valid log level")
221224

222225
torch.ops.tensorrt.set_logging_level(int(log_level))
223226

0 commit comments

Comments
 (0)