🐛 [Bug] MutableModule Test Failures

[narendasan]

Bug Description

FAILED runtime/test_mutable_torchtrt_module.py::test_model_complex_dynamic_shape_with_saving - KeyError: 'enabled_precisions'
FAILED runtime/test_mutable_torchtrt_module.py::test_resnet18 - AssertionError: False is not true : The output of saved and reloaded Mutable Module is not correct.
FAILED runtime/test_mutable_torchtrt_module.py::test_save - AssertionError: False is not true : The output of saved and reloaded Mutable Module is not correct.
FAILED runtime/test_mutable_torchtrt_module.py::test_resnet18_modify_attribute - AssertionError: False is not true : The output of refitted Mutable Module is not correct.
FAILED runtime/test_mutable_torchtrt_module.py::test_resnet18_modify_attribute_no_refit - AssertionError: <RefitFlag.NEEDS_REFIT: 2> != <RefitFlag.LIVE: 4> : update_refit_condition() failed to set the flag to LIVE.
FAILED runtime/test_mutable_torchtrt_module.py::test_custom_model_with_kwarg - AssertionError: False is not true : The output of original and refitted Mutable Module is not the same.
FAILED runtime/test_mutable_torchtrt_module.py::test_custom_model_with_inplace_init - AssertionError: False is not true : The output of original and refitted Mutable Module is not the same.
FAILED runtime/test_mutable_torchtrt_module.py::test_custom_model_with_init_recompile - AssertionError: False is not true : The output of original and refitted Mutable Module is not the same.
FAILED runtime/test_mutable_torchtrt_module.py::test_custom_model_with_kwarg_different_input - AssertionError: False is not true : The output of original and refitted Mutable Module is not the same.

To Reproduce

Steps to reproduce the behavior:

Expected behavior

Environment

Build information about Torch-TensorRT can be found by turning on debug messages

• Torch-TensorRT Version (e.g. 1.0.0):
• PyTorch Version (e.g. 1.0):
• CPU Architecture:
• OS (e.g., Linux):
• How you installed PyTorch (conda, pip, libtorch, source):
• Build command you used (if compiling from source):
• Are you using local sources or building from archives:
• Python version:
• CUDA version:
• GPU models and configuration:
• Any other relevant information:

Additional context