Skip to content

Commit 584ed86

Browse files
MrGevamikeiovine
authored andcommitted
[https://nvbugs/5572320][fix] Ported test_ad_trtllm_bench.py from main (#8671)
Signed-off-by: Eran Geva <[email protected]> Signed-off-by: Mike Iovine <[email protected]>
1 parent 870da18 commit 584ed86

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

tests/integration/test_lists/test-db/l0_b200.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,8 @@ l0_b200:
7777
- unittest/_torch/modeling -k "modeling_mixtral"
7878
- unittest/_torch/modeling -k "modeling_gpt_oss"
7979
- unittest/_torch/modeling/test_modeling_exaone4.py::TestEXAONE4::test_llm_load_1_FP8
80-
- unittest/_torch/auto_deploy/unit/singlegpu -k "not test_trtllm_bench_backend_comparison"
8180
# ------------- AutoDeploy tests ---------------
81+
- unittest/_torch/auto_deploy/unit/singlegpu -k "not test_trtllm_bench_backend_comparison"
8282
- accuracy/test_llm_api_autodeploy.py::TestLlama3_1_8B::test_auto_dtype[False-1]
8383
- unittest/_torch/auto_deploy/unit/singlegpu
8484
- condition:

tests/integration/test_lists/test-db/l0_h100.yml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,9 @@ l0_h100:
3232
- unittest/_torch/modeling -k "modeling_nemotron"
3333
- unittest/_torch/modeling -k "modeling_gemma3"
3434
- unittest/_torch/modeling -k "modeling_gpt_oss"
35+
- unittest/disaggregated/test_disagg_utils.py
36+
- unittest/disaggregated/test_router.py
37+
- unittest/disaggregated/test_remoteDictionary.py
3538
- accuracy/test_llm_api_pytorch.py::TestGemma3_1BInstruct::test_auto_dtype
3639
- accuracy/test_llm_api_pytorch.py::TestGemma3_1BInstruct::test_auto_dtype_vswa_without_reuse
3740
- accuracy/test_llm_api_pytorch.py::TestGemma3_1BInstruct::test_auto_dtype_vswa_reuse

0 commit comments

Comments
 (0)