Skip to content

Commit b0dc251

Browse files
committed
chore: address review comments
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent c267dd5 commit b0dc251

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

py/torch_tensorrt/_compile.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,13 +63,13 @@ def _get_target_ir(module_type: _ModuleType, ir: str) -> _IRType:
6363
# Options are listed in order of preference
6464
if module_is_tsable:
6565
logging.log(
66-
logging.Level.Warning,
67-
"Input graph is a Torchscript module but the ir provided is default (dynamo). Please set ir=torchscript to suppress the warning. Compiling the module with ir=ts",
66+
logging.Level.Info, "ir was set to default, using TorchScript as ir"
6867
)
6968
return _IRType.ts
7069
elif module_is_fxable:
7170
logging.log(
72-
logging.Level.Info, "ir was set to default, using dynamo as ir"
71+
logging.Level.Warning,
72+
"Input graph is a torch.fx.GraphModule but the ir provided is default (ts). Please set ir=dynamo to suppress the warning.",
7373
)
7474
return _IRType.dynamo
7575
else:

0 commit comments

Comments
 (0)