Skip to content

Commit c59b4a1

Browse files
committed
fix: tmp skip test_tensorrt_export_reload in torch-tensorrt 2.8.0
1 parent 25c5d09 commit c59b4a1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/tests_pytorch/models/test_torch_tensorrt.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,8 +130,9 @@ def test_tensorrt_save_ir_type(ir, export_type):
130130
"ir",
131131
["default", "dynamo", "ts"],
132132
)
133-
@RunIf(tensorrt=True, min_cuda_gpus=1, min_torch="2.2.0")
133+
@RunIf(tensorrt=True, min_cuda_gpus=1, min_torch="2.2.0", max_torch="2.8.0")
134134
def test_tensorrt_export_reload(output_format, ir, tmp_path):
135+
# todo remove max_torch once https://github.com/pytorch/TensorRT/issues/3775 is fixed
135136
if ir == "ts" and output_format == "exported_program":
136137
pytest.skip("TorchScript cannot be exported as exported_program")
137138

0 commit comments

Comments
 (0)