diff --git a/tests/integration/test_lists/test-db/l0_a30.yml b/tests/integration/test_lists/test-db/l0_a30.yml index 3adfc02353..64cbb936b6 100644 --- a/tests/integration/test_lists/test-db/l0_a30.yml +++ b/tests/integration/test_lists/test-db/l0_a30.yml @@ -18,8 +18,15 @@ l0_a30: - unittest/_torch/modeling -k "modeling_phi3" - unittest/_torch/modeling -k "modeling_qwen" - unittest/_torch/modeling -k "modeling_qwen_moe" + - unittest/_torch/modeling -k "modeling_out_of_tree" - unittest/_torch/auto_deploy/unit/singlegpu -k "not test_trtllm_bench_backend_comparison" - unittest/_torch/sampler/test_beam_search.py + - test_e2e.py::test_openai_completions_with_logit_bias[torch_sampler] + - test_e2e.py::test_openai_chat_with_logit_bias[torch_sampler] + - test_e2e.py::test_openai_completions_with_logit_bias[trtllm_sampler] + - test_e2e.py::test_openai_chat_with_logit_bias[trtllm_sampler] + - test_e2e.py::test_ptp_quickstart_bert[VANILLA-BertForSequenceClassification-bert/bert-base-uncased-yelp-polarity] + - test_e2e.py::test_ptp_quickstart_bert[TRTLLM-BertForSequenceClassification-bert/bert-base-uncased-yelp-polarity] - condition: ranges: system_gpu_count: diff --git a/tests/integration/test_lists/test-db/l0_l40s.yml b/tests/integration/test_lists/test-db/l0_l40s.yml index ee25ace289..0bb1d4414d 100644 --- a/tests/integration/test_lists/test-db/l0_l40s.yml +++ b/tests/integration/test_lists/test-db/l0_l40s.yml @@ -15,7 +15,6 @@ l0_l40s: tests: # ------------- PyTorch tests --------------- - unittest/_torch/modeling -k "modeling_mllama" - - unittest/_torch/modeling -k "modeling_out_of_tree" - unittest/_torch/modeling -k "modeling_vila" - unittest/_torch/modeling -k "modeling_siglip" - test_e2e.py::test_ptp_scaffolding[DeepSeek-R1-Distill-Qwen-7B-DeepSeek-R1/DeepSeek-R1-Distill-Qwen-7B] @@ -31,12 +30,6 @@ l0_l40s: - test_e2e.py::test_ptp_quickstart_multimodal_phi4mm[audio] - test_e2e.py::test_ptp_quickstart_multimodal_phi4mm[image] - test_e2e.py::test_ptp_quickstart_multimodal_phi4mm[image_audio] - - test_e2e.py::test_ptp_quickstart_bert[VANILLA-BertForSequenceClassification-bert/bert-base-uncased-yelp-polarity] - - test_e2e.py::test_ptp_quickstart_bert[TRTLLM-BertForSequenceClassification-bert/bert-base-uncased-yelp-polarity] - - test_e2e.py::test_openai_completions_with_logit_bias[torch_sampler] - - test_e2e.py::test_openai_chat_with_logit_bias[torch_sampler] - - test_e2e.py::test_openai_completions_with_logit_bias[trtllm_sampler] - - test_e2e.py::test_openai_chat_with_logit_bias[trtllm_sampler] - accuracy/test_llm_api_pytorch.py::TestQwen2_VL_7B::test_auto_dtype - condition: ranges: