Skip to content

Commit

Permalink
chore: minor fix
Browse files Browse the repository at this point in the history
Signed-off-by: Dheeraj Peri <peri.dheeraj@gmail.com>
  • Loading branch information
peri044 committed Jul 19, 2023
1 parent 077eccc commit 8b77617
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion py/torch_tensorrt/_compile.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ def _get_target_ir(module_type: _ModuleType, ir: str) -> _IRType:
elif module_is_tsable:
logging.log(
logging.Level.Warning,
"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",
"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=torchscript",
)
return _IRType.ts
else:
Expand Down

0 comments on commit 8b77617

Please sign in to comment.