Skip to content

Commit 14539f1

Browse files
authored
[None][test] Fix DGX_B200 CI timeout by splitting multimodal tests an… (NVIDIA#12978)
Signed-off-by: nv-guomingz <137257613+nv-guomingz@users.noreply.github.com>
1 parent 04915ad commit 14539f1

File tree

2 files changed

+20
-4
lines changed

2 files changed

+20
-4
lines changed

tests/integration/defs/.test_durations

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -790,6 +790,7 @@
790790
"test_unittests.py::test_unittests_v2[unittest/_torch/compilation]": 31.94,
791791
"test_unittests.py::test_unittests_v2[unittest/_torch/debugger]": 36.69,
792792
"test_unittests.py::test_unittests_v2[unittest/_torch/executor]": 170.86,
793+
"test_unittests.py::test_unittests_v2[unittest/_torch/flashinfer/test_trtllm_flashinfer_symbol_collision.py]": 1004.0,
793794
"test_unittests.py::test_unittests_v2[unittest/_torch/misc]": 600.5,
794795
"test_unittests.py::test_unittests_v2[unittest/_torch/modeling -k \"modeling_llama\"]": 718.749935634085,
795796
"test_unittests.py::test_unittests_v2[unittest/_torch/modeling -k \"modeling_mixtral\"]": 208.1838396479725,
@@ -801,11 +802,21 @@
801802
"test_unittests.py::test_unittests_v2[unittest/_torch/modeling -k \"modeling_qwen_moe\"]": 401.2630233000382,
802803
"test_unittests.py::test_unittests_v2[unittest/_torch/modeling -k \"modeling_vila\"]": 79.90315388399176,
803804
"test_unittests.py::test_unittests_v2[unittest/_torch/modules]": 158.5,
805+
"test_unittests.py::test_unittests_v2[unittest/_torch/modules/moe/test_moe_backend.py::test_moe_backend -k \"CUTEDSL\"]": 68.0,
806+
"test_unittests.py::test_unittests_v2[unittest/_torch/modules/moe/test_moe_backend.py::test_moe_backend -k \"CUTLASS\"]": 1598.0,
807+
"test_unittests.py::test_unittests_v2[unittest/_torch/modules/moe/test_moe_backend.py::test_moe_backend -k \"DEEPGEMM\"]": 22.0,
808+
"test_unittests.py::test_unittests_v2[unittest/_torch/modules/moe/test_moe_backend.py::test_moe_backend -k \"DENSEGEMM\"]": 149.0,
809+
"test_unittests.py::test_unittests_v2[unittest/_torch/modules/moe/test_moe_backend.py::test_moe_backend -k \"TRTLLM\"]": 998.0,
804810
"test_unittests.py::test_unittests_v2[unittest/_torch/multi_gpu_modeling -k \"deepseek\"]": 393.0210295501165,
805-
"test_unittests.py::test_unittests_v2[unittest/_torch/multimodal]": 23.54,
806-
"test_unittests.py::test_unittests_v2[unittest/_torch/sampler]": 107.66,
811+
"test_unittests.py::test_unittests_v2[unittest/_torch/multimodal/test_external_embedding.py]": 30.0,
812+
"test_unittests.py::test_unittests_v2[unittest/_torch/multimodal/test_find_num_image_tokens.py]": 240.0,
813+
"test_unittests.py::test_unittests_v2[unittest/_torch/multimodal/test_fuse_input_embeds.py]": 60.0,
814+
"test_unittests.py::test_unittests_v2[unittest/_torch/multimodal/test_mm_encoder_standalone.py]": 1800.0,
815+
"test_unittests.py::test_unittests_v2[unittest/_torch/multimodal/test_multimodal_runtime.py]": 180.0,
816+
"test_unittests.py::test_unittests_v2[unittest/_torch/multimodal/test_share_multiparams.py]": 30.0,
817+
"test_unittests.py::test_unittests_v2[unittest/_torch/sampler]": 1020.0,
807818
"test_unittests.py::test_unittests_v2[unittest/_torch/speculative]": 1850.16,
808-
"test_unittests.py::test_unittests_v2[unittest/_torch/thop/parallel]": 311.58,
819+
"test_unittests.py::test_unittests_v2[unittest/_torch/thop/parallel]": 1463.0,
809820
"test_unittests.py::test_unittests_v2[unittest/_torch/thop/serial]": 18.96,
810821
"test_unittests.py::test_unittests_v2[unittest/api_stability]": 33.137137457728386,
811822
"test_unittests.py::test_unittests_v2[unittest/bindings]": 1119.2564616799355,

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

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,12 @@ l0_b200:
120120
# ------------- MoE: FlashInfer & TRTLLM symbol collision tests ---------------
121121
- unittest/_torch/flashinfer/test_trtllm_flashinfer_symbol_collision.py
122122
# --- MoE end
123-
- unittest/_torch/multimodal
123+
- unittest/_torch/multimodal/test_mm_encoder_standalone.py
124+
- unittest/_torch/multimodal/test_multimodal_runtime.py
125+
- unittest/_torch/multimodal/test_find_num_image_tokens.py
126+
- unittest/_torch/multimodal/test_fuse_input_embeds.py
127+
- unittest/_torch/multimodal/test_external_embedding.py
128+
- unittest/_torch/multimodal/test_share_multiparams.py
124129
- unittest/_torch/sampler
125130
- unittest/_torch/speculative
126131
- unittest/_torch/thop/parallel TIMEOUT (90)

0 commit comments

Comments
 (0)