From 5efda075b7249b961ff9754f23b168513c562ed9 Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Thu, 9 Oct 2025 18:35:01 -0700 Subject: [PATCH 1/2] fix test --- tests/py/ts/api/test_export_serde.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/py/ts/api/test_export_serde.py b/tests/py/ts/api/test_export_serde.py index 21bf57082b..bdf02098a8 100644 --- a/tests/py/ts/api/test_export_serde.py +++ b/tests/py/ts/api/test_export_serde.py @@ -15,8 +15,9 @@ assertions = unittest.TestCase() + @pytest.mark.unit -def test_save_load_ts(ir): +def test_save_load_ts(): """ This tests save/load API on Torchscript format (model still compiled using ts workflow) """ From fd811e2b69dd361bd2b5300d29894e9f5603a7ae Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Thu, 9 Oct 2025 19:32:37 -0700 Subject: [PATCH 2/2] fix test --- tests/py/ts/api/test_export_serde.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/tests/py/ts/api/test_export_serde.py b/tests/py/ts/api/test_export_serde.py index bdf02098a8..eb1850b132 100644 --- a/tests/py/ts/api/test_export_serde.py +++ b/tests/py/ts/api/test_export_serde.py @@ -10,7 +10,6 @@ from torch_tensorrt.dynamo.utils import ( COSINE_THRESHOLD, cosine_similarity, - get_model_device, ) assertions = unittest.TestCase() @@ -42,8 +41,6 @@ def forward(self, x): ir="ts", inputs=[input], min_block_size=1, - cache_built_engines=False, - reuse_cached_engines=False, ) outputs_trt = trt_gm(input) # Save it as torchscript representation