Skip to content

Commit 41b3928

Browse files
committed
Added kwarg support for dynamo.compile
1 parent 2ee679b commit 41b3928

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

py/torch_tensorrt/dynamo/_compiler.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,8 @@ def compile(
176176
else:
177177
make_refitable = kwargs["refit"]
178178

179+
if kwarg_inputs is None:
180+
kwarg_inputs = {}
179181
engine_capability = EngineCapability._from(engine_capability)
180182

181183
if torch_executed_modules is not None and torch_executed_modules:

0 commit comments

Comments
 (0)