Skip to content

Commit 20ac8d7

Browse files
committed
Implemented basic pipeline for Refitting (#2886)
1 parent 936f0ae commit 20ac8d7

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

py/torch_tensorrt/dynamo/_compiler.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,18 @@ def compile(
176176

177177
if kwarg_inputs is None:
178178
kwarg_inputs = {}
179+
180+
if "refit" in kwargs.keys():
181+
warnings.warn(
182+
"Refit is deprecated. Please use make_refitable=True if you want to enable refitting of the engine.",
183+
DeprecationWarning,
184+
stacklevel=2,
185+
)
186+
if make_refitable:
187+
raise ValueError("Use flag make_refitable only. Flag refit is deprecated.")
188+
else:
189+
make_refitable = kwargs["refit"]
190+
179191
engine_capability = EngineCapability._from(engine_capability)
180192

181193
if torch_executed_modules is not None and torch_executed_modules:

0 commit comments

Comments
 (0)