Skip to content

Commit b54b6bd

Browse files
committed
fix: distingush engines based on compilation settings in addition to graph structure
Signed-off-by: Naren Dasan <[email protected]>
1 parent 2518db5 commit b54b6bd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -557,7 +557,7 @@ def run(
557557
)
558558
assert (
559559
setting_compatiblity
560-
), f"Attempted to refit a prebuilt engine with incompatible settings: {incompattible_settings}, (old_settings: {engine_compilation_settings}, new_settings: {self.compilation_settings})"
560+
), f"Attempted to refit a cached engine with incompatible settings: {incompattible_settings}, (old_settings: {engine_compilation_settings}, new_settings: {self.compilation_settings})"
561561

562562
for i, e in enumerate(
563563
[
@@ -567,7 +567,7 @@ def run(
567567
):
568568
assert (
569569
e
570-
), f"Found that cached engine was built for a different input size (input: {i}, cached size: {cached_engine_input_specs[i]}, new size: {self.input_specs[i]}"
570+
), f"Attempted to refit a cached engine built for a different input size (input: {i}, cached size: {cached_engine_input_specs[i]}, new size: {self.input_specs[i]}"
571571

572572
_LOGGER.info(
573573
"Found the cached engine that corresponds to this graph. It is directly loaded."

0 commit comments

Comments
 (0)