From e52eb8278084dbf10193c2aaa7b0b483310309c4 Mon Sep 17 00:00:00 2001 From: nvyocox Date: Fri, 6 Feb 2026 00:32:57 +0800 Subject: [PATCH] [#11234][test] Move test_ad_export_onnx to integration examples (#11260) Signed-off-by: yocox --- .../defs/examples}/test_ad_export_onnx.py | 8 ++++++++ tests/integration/test_lists/test-db/l0_h100.yml | 1 + 2 files changed, 9 insertions(+) rename tests/{unittest/_torch/auto_deploy/unit/singlegpu => integration/defs/examples}/test_ad_export_onnx.py (92%) diff --git a/tests/unittest/_torch/auto_deploy/unit/singlegpu/test_ad_export_onnx.py b/tests/integration/defs/examples/test_ad_export_onnx.py similarity index 92% rename from tests/unittest/_torch/auto_deploy/unit/singlegpu/test_ad_export_onnx.py rename to tests/integration/defs/examples/test_ad_export_onnx.py index 3de8527263..af5c36eb2c 100644 --- a/tests/unittest/_torch/auto_deploy/unit/singlegpu/test_ad_export_onnx.py +++ b/tests/integration/defs/examples/test_ad_export_onnx.py @@ -1,7 +1,15 @@ import os +import sys +from pathlib import Path import onnx import pytest + +# Import utility from unittest directory +sys.path.insert( + 0, + str(Path(__file__).parent.parent.parent.parent / "unittest/_torch/auto_deploy/_utils_test"), +) from _model_test_utils import get_small_model_config from tensorrt_llm._torch.auto_deploy.export import export_onnx diff --git a/tests/integration/test_lists/test-db/l0_h100.yml b/tests/integration/test_lists/test-db/l0_h100.yml index d7f55c98d1..2831bc95bc 100644 --- a/tests/integration/test_lists/test-db/l0_h100.yml +++ b/tests/integration/test_lists/test-db/l0_h100.yml @@ -446,3 +446,4 @@ l0_h100: - examples/test_ad_speculative_decoding.py::test_eagle_model_with_weights - examples/test_ad_speculative_decoding.py::test_eagle_wrapper_forward[1] - examples/test_ad_speculative_decoding.py::test_eagle_wrapper_forward[2] + - examples/test_ad_export_onnx.py::test_ad_export_onnx[Qwen/Qwen2.5-3B-Instruct-/tmp/test_ad_export_onnx_qwen2.5-3b-36]