Deleted sample inputs for lowering everywhere

Author: cehongwang

py/torch_tensorrt/dynamo/backend/backends.py

@@ -98,7 +98,7 @@ def _pretraced_backend(
 
             logger.debug("Post-AOT Autograd graph:\n" + str(gm.graph))
 
-            gm = post_lowering(gm, sample_inputs)
+            gm = post_lowering(gm)
 
             logger.debug("Lowered Input graph:\n " + str(gm.graph))
 

tests/py/dynamo/conversion/harness.py

@@ -219,7 +219,7 @@ def generate_graph(
         torch_inputs = get_torch_inputs(original_inputs, _defaults.DEVICE)
         if use_dynamo_tracer:
             exported_program = torch_tensorrt.dynamo.trace(mod, tuple(original_inputs))
-            exported_program = pre_export_lowering(exported_program, torch_inputs)
+            exported_program = pre_export_lowering(exported_program)
             exported_program = exported_program.run_decompositions(
                 get_decompositions(False)
             )
@@ -228,7 +228,7 @@ def generate_graph(
             fx_module = torch.fx.symbolic_trace(mod)
 
         if enable_passes:
-            fx_module = post_lowering(fx_module, original_inputs)
+            fx_module = post_lowering(fx_module)
 
         if propagate_shapes:
             # TODO: This is currently being used to test embedding_bag_aten due to https://github.com/pytorch/TensorRT/issues/2843

tests/py/dynamo/models/test_model_refit.py

@@ -62,12 +62,12 @@ def test_mapping():
     engine_info = trt_gm._run_on_acc_0.engine.__getstate__()[0]
     engine = get_engine_from_encoded_engine(engine_info[3], runtime)
 
-    exp_program2 = pre_export_lowering(exp_program2, inputs)
+    exp_program2 = pre_export_lowering(exp_program2)
     exp_program2 = exp_program2.run_decompositions(
         get_decompositions(settings.enable_experimental_decompositions)
     )
     new_gm = exp_program2.module()
-    new_gm = post_lowering(new_gm, inputs)
+    new_gm = post_lowering(new_gm)
     mapping = construct_refit_mapping(new_gm, trt_input, settings)
 
     refitter = trt.Refitter(engine, TRT_LOGGER)

tests/py/dynamo/testing_utilities.py

@@ -50,7 +50,7 @@ def fx_dynamo_testing_backend(
             decompositions=get_decompositions(),
         )
 
-        gm = post_lowering(gm, sample_inputs)
+        gm = post_lowering(gm)
 
         trt_compiled = custom_backend(
             gm,