🐛 [Bug] Serde issues with torch-trt save

[peri044]

Bug Description

import torch_tensorrt as trt
inputs = [trt.Input(min_shape=(1, 1, 28, 28), 
                   opt_shape=(50, 1, 28, 28), 
                   max_shape=(64, 1, 28, 28), 
                   dtype=torch.float32)]
exp_program = trt.dynamo.trace(model_from_state, inputs)
trt_gm = trt.dynamo.compile(exp_program, inputs=inputs)

torch_inputs = torch.randn(50, 1, 28, 28).cuda()
trt.save(trt_gm, "trt_model.ep", inputs=torch_inputs)

model = torch.export.load("trt_model.ep")

Error :

torch/fx/experimental/symbolic_shapes.py:4449] s0 is not in var_ranges, defaulting to unknown range. E1002 18:36:09.951000 124112691492352 torch/fx/experimental/recording.py:281] failed while running evaluate_expr(*(s0 >= 0, True), **{'fx_node': None})

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