Added kwarg support for dynamo.compile

pytorch · Jul 9, 2024 · fb899ac · fb899ac
1 parent 131c158
commit fb899ac
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/py/torch_tensorrt/dynamo/_compiler.py b/py/torch_tensorrt/dynamo/_compiler.py
@@ -176,6 +176,8 @@ def compile(
         else:
             make_refitable = kwargs["refit"]
 
+    if kwarg_inputs is None:
+        kwarg_inputs = {}
     engine_capability = EngineCapability._from(engine_capability)
 
     if torch_executed_modules is not None and torch_executed_modules: