kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-02-16 15:55:08 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[#11234][test] Move test_ad_export_onnx to integration examples (#11260)

Browse Source 
Signed-off-by: yocox <yocox@nvidia.com>
This commit is contained in:
nvyocox 2026-02-06 00:32:57 +08:00 committed by GitHub
parent d3d951d837
commit e52eb82780
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/unittest/_torch/auto_deploy/unit/singlegpu/test_ad_export_onnx.py → tests/integration/defs/examples/test_ad_export_onnx.py
View File

@ -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

1
tests/integration/test_lists/test-db/l0_h100.yml
View File

@ -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]