🐛 [Bug] Failures in test_dyn_models.py

[zewenli98]

Bug Description

The failures were shown in CI (https://github.com/pytorch/TensorRT/actions/runs/8863273616/job/24391143167?pr=2781):

=========================== short test summary info ============================
FAILED models/test_dyn_models.py::test_base_dynamic - torch.fx.experimental.symbolic_shapes.ConstraintViolationError: Constraints violated (L['x'].size()[0])! For more information, run with TORCH_LOGS="+dynamic".
  - Not all values of L['x'].size()[0] = L['x'].size()[0] in the specified range L['x'].size()[0] <= 8 satisfy the generated guard 4 <= L['x'].size()[0] and L['x'].size()[0] <= 8


You can suppress this exception and fall back to eager by setting:
    import torch._dynamo
    torch._dynamo.config.suppress_errors = True
FAILED models/test_dyn_models.py::test_base_dynamic_fallback - torch.fx.experimental.symbolic_shapes.ConstraintViolationError: Constraints violated (L['x'].size()[0])! For more information, run with TORCH_LOGS="+dynamic".
  - Not all values of L['x'].size()[0] = L['x'].size()[0] in the specified range L['x'].size()[0] <= 8 satisfy the generated guard 4 <= L['x'].size()[0] and L['x'].size()[0] <= 8


You can suppress this exception and fall back to eager by setting:
    import torch._dynamo
    torch._dynamo.config.suppress_errors = True
=================== 2 failed, 2 passed, 8 warnings in 31.07s ===================

Environment

• Torch-TensorRT Version): main branch
• PyTorch Version: 2.3
• TensorRT: 10.0.1.6