Change arg_inputs back to inputs

Author: cehongwang

docs/_downloads/6dc7949e3e7f3cb855e4a7b28eadc851/vgg16_fp8_ptq.py

@@ -229,7 +229,7 @@ def calibrate_loop(model):
         exp_program = torch.export.export(model, (input_tensor,))
         trt_model = torchtrt.dynamo.compile(
             exp_program,
-            arg_inputs=[input_tensor],
+            inputs=[input_tensor],
             enabled_precisions={torch.float8_e4m3fn},
             min_block_size=1,
             debug=False,

examples/dynamo/vgg16_fp8_ptq.py

@@ -229,7 +229,7 @@ def calibrate_loop(model):
         exp_program = torch.export.export(model, (input_tensor,))
         trt_model = torchtrt.dynamo.compile(
             exp_program,
-            arg_inputs=[input_tensor],
+            inputs=[input_tensor],
             enabled_precisions={torch.float8_e4m3fn},
             min_block_size=1,
             debug=False,

py/torch_tensorrt/dynamo/_compiler.py

@@ -47,9 +47,8 @@
 
 def compile(
     exported_program: ExportedProgram,
+    inputs: Sequence[Any],
     *,
-    arg_inputs: Optional[Sequence[Any]] = None,
-    inputs: Optional[Sequence[Any]] = None,
     kwarg_inputs: Optional[dict[Any, Any]] = None,
     device: Optional[Union[Device, torch.device, str]] = _defaults.DEVICE,
     disable_tf32: bool = _defaults.DISABLE_TF32,
@@ -195,19 +194,13 @@ def compile(
             f"Detected torch_executed_modules was non-empty: {torch_executed_modules}"
             "\nThis feature is unimplemented in Torch-TRT Dynamo currently."
         )
-    if inputs is not None:
-        logger.warning(
-            "'inputs' is deprecated. Please use 'args_inputs' in the future."
-        )
-        if arg_inputs is None:
-            arg_inputs = inputs
-        else:
-            logger.warning(
-                "Both 'arg_inputs' and 'inputs' are received. 'inputs' will be ignored."
-            )
-    else:
-        if arg_inputs is None:
-            raise AssertionError("'arg_inputs' cannot be empty")
+
+    # Aliasing inputs to arg_inputs for better understanding
+    arg_inputs = inputs
+
+    if kwarg_inputs is None:
+        kwarg_inputs = {}
+
     if not isinstance(arg_inputs, collections.abc.Sequence):
         arg_inputs = [arg_inputs]