Fixed comments

Author: cehongwang

py/torch_tensorrt/_compile.py

@@ -243,11 +243,11 @@ def compile(
     elif target_ir == _IRType.dynamo:
         # Prepare torch and torchtrt inputs
         if not arg_inputs and not inputs:
-            raise AssertionError("'arg_input' and 'input' should not both be None.")
+            raise AssertionError("'arg_inputs' and 'inputs' should not both be None.")
 
         elif arg_inputs and inputs:
             raise AssertionError(
-                "'arg_input' and 'input' should not be used at the same time."
+                "'arg_inputs' and 'inputs' should not be used at the same time."
             )
         arg_inputs = inputs or arg_inputs
 
@@ -367,11 +367,11 @@ def convert_method_to_trt_engine(
     elif target_ir == _IRType.dynamo:
         # Prepare torch and torchtrt inputs
         if not arg_inputs and not inputs:
-            raise AssertionError("'arg_input' and 'input' should not both be None.")
+            raise AssertionError("'arg_inputs' and 'inputs' should not both be None.")
 
         elif arg_inputs and inputs:
             raise AssertionError(
-                "'arg_input' and 'input' should not be used at the same time."
+                "'arg_inputs' and 'inputs' should not be used at the same time."
             )
         arg_inputs = arg_inputs or inputs
 
@@ -475,7 +475,7 @@ def save(
         )
     if arg_inputs and inputs:
         raise AssertionError(
-            "'arg_input' and 'input' should not be used at the same time."
+            "'arg_inputs' and 'inputs' should not be used at the same time."
         )
 
     arg_inputs = inputs or arg_inputs

py/torch_tensorrt/dynamo/_compiler.py

@@ -187,11 +187,11 @@ def compile(
 
     # Aliasing inputs to arg_inputs for better understanding
     if not arg_inputs and not inputs:
-        raise AssertionError("'arg_input' and 'input' should not both be None.")
+        raise AssertionError("'arg_inputs' and 'inputs' should not both be None.")
 
     elif arg_inputs and inputs:
         raise AssertionError(
-            "'arg_input' and 'input' should not be used at the same time."
+            "'arg_inputs' and 'inputs' should not be used at the same time."
         )
 
     arg_inputs = inputs or arg_inputs
@@ -609,11 +609,11 @@ def convert_module_to_trt_engine(
             stacklevel=2,
         )
     if not arg_inputs and not inputs:
-        raise AssertionError("'arg_input' and 'input' should not both be None.")
+        raise AssertionError("'arg_inputs' and 'inputs' should not both be None.")
 
     elif arg_inputs and inputs:
         raise AssertionError(
-            "'arg_input' and 'input' should not be used at the same time."
+            "'arg_inputs' and 'inputs' should not be used at the same time."
         )
 
     arg_inputs = inputs or arg_inputs

py/torch_tensorrt/dynamo/_tracer.py

@@ -59,11 +59,11 @@ def trace(
 
     # Set log level at the top of compilation (torch_tensorrt.dynamo)
     if not arg_inputs and not inputs:
-        raise AssertionError("'arg_input' and 'input' should not both be None.")
+        raise AssertionError("'arg_inputs' and 'inputs' should not both be None.")
 
     elif arg_inputs and inputs:
         raise AssertionError(
-            "'arg_input' and 'input' should not be used at the same time."
+            "'arg_inputs' and 'inputs' should not be used at the same time."
         )
     arg_inputs = inputs or arg_inputs
 
@@ -79,7 +79,6 @@ def trace(
     torch_kwarg_inputs = get_torch_inputs(kwarg_inputs, device)
     dynamic_shapes = get_dynamic_shapes_args(mod, arg_inputs)
     dynamic_shapes.update(get_dynamic_shapes_kwargs(kwarg_inputs))
-    # breakpoint()
     exp_program = export(
         mod,
         tuple(torch_arg_inputs),