update

Author: bhimrazy

src/lightning/pytorch/utilities/imports.py

@@ -28,6 +28,7 @@
 _TORCHMETRICS_GREATER_EQUAL_0_11 = RequirementCache("torchmetrics>=0.11.0")  # using new API with task
 _TORCHMETRICS_GREATER_EQUAL_1_0_0 = RequirementCache("torchmetrics>=1.0.0")
 _TORCH_EQUAL_2_8 = RequirementCache("torch>=2.8.0,<2.9.0")
+_TORCH_EQUAL_2_9 = RequirementCache("torch>=2.9.0,<2.10.0")
 _TORCH_GREATER_EQUAL_2_8 = compare_version("torch", operator.ge, "2.8.0")
 
 _OMEGACONF_AVAILABLE = package_available("omegaconf")

tests/tests_pytorch/models/test_torch_tensorrt.py

@@ -10,6 +10,7 @@
 from lightning.pytorch.core.module import _TORCH_TRT_AVAILABLE
 from lightning.pytorch.demos.boring_classes import BoringModel
 from lightning.pytorch.utilities.exceptions import MisconfigurationException
+from lightning.pytorch.utilities.imports import _TORCH_EQUAL_2_9
 from tests_pytorch.helpers.runif import RunIf
 
 
@@ -110,7 +111,14 @@ def test_tensorrt_saves_on_multi_gpu(tmp_path):
     [
         ("default", torch.fx.GraphModule),
         ("dynamo", torch.fx.GraphModule),
-        ("ts", torch.jit.ScriptModule),
+        pytest.param(
+            "ts",
+            torch.jit.ScriptModule,
+            marks=pytest.mark.skipif(
+                _TORCH_EQUAL_2_9,
+                reason="TorchScript IR crashes with torch_tensorrt on PyTorch 2.9",
+            ),
+        ),
     ],
 )
 @RunIf(tensorrt=True, min_cuda_gpus=1, min_torch="2.2.0")