Skip to content

Commit 3a9617b

Browse files
committed
update
1 parent a600f41 commit 3a9617b

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

tests/tests_pytorch/models/test_torch_tensorrt.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,17 @@ def test_tensorrt_save_ir_type(ir, export_type):
136136
)
137137
@pytest.mark.parametrize(
138138
"ir",
139-
["default", "dynamo", "ts"],
139+
[
140+
"default",
141+
"dynamo",
142+
pytest.param(
143+
"ts",
144+
marks=pytest.mark.skipif(
145+
_TORCH_EQUAL_2_9,
146+
reason="TorchScript IR crashes with torch_tensorrt on PyTorch 2.9",
147+
),
148+
),
149+
],
140150
)
141151
@RunIf(tensorrt=True, min_cuda_gpus=1, min_torch="2.2.0")
142152
def test_tensorrt_export_reload(output_format, ir, tmp_path):

0 commit comments

Comments
 (0)