We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 936f0ae commit 20ac8d7Copy full SHA for 20ac8d7
py/torch_tensorrt/dynamo/_compiler.py
@@ -176,6 +176,18 @@ def compile(
176
177
if kwarg_inputs is None:
178
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
191
engine_capability = EngineCapability._from(engine_capability)
192
193
if torch_executed_modules is not None and torch_executed_modules:
0 commit comments