Skip to content

Commit 1109dd8

Browse files
committed
chore: rebase to dynamo_refactor
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 2cc58bb commit 1109dd8

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

py/torch_tensorrt/_compile.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -61,17 +61,17 @@ def _get_target_ir(module_type: _ModuleType, ir: str) -> _IRType:
6161
else:
6262
if ir == "default":
6363
# Options are listed in order of preference
64-
if module_is_fxable:
64+
if module_is_tsable:
6565
logging.log(
66-
logging.Level.Info, "ir was set to default, using dynamo as ir"
66+
logging.Level.Info, "ir was set to default, using TorchScript as ir"
6767
)
68-
return _IRType.dynamo
69-
elif module_is_tsable:
68+
return _IRType.ts
69+
elif module_is_fxable:
7070
logging.log(
7171
logging.Level.Warning,
72-
"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",
72+
"Input graph is a torch.fx.GraphModule but the ir provided is default (ts). Please set ir=dynamo to suppress the warning.",
7373
)
74-
return _IRType.ts
74+
return _IRType.dynamo
7575
else:
7676
raise ValueError("Module was provided in an unsupported format")
7777
else:

py/versions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
__version__ = "2.0.0.dev"
1+
__version__ = "1.5.0.dev0"
22
__cuda_version__ = "12.1"
33
__cudnn_version__ = "8.8"
44
__tensorrt_version__ = "8.6"

0 commit comments

Comments
 (0)