[CI] Tune threshold (#5787)

This commit is contained in:
Lianmin Zheng
2025-04-27 04:10:22 -07:00
committed by GitHub
parent 9c088829ee
commit 3c4e0ee64d
3 changed files with 5 additions and 5 deletions

View File

@@ -82,10 +82,10 @@ suites = {
TestFile("test_triton_moe_channel_fp8_kernel.py", 25),
],
"per-commit-2-gpu": [
TestFile("models/lora/test_lora_tp.py", 300),
TestFile("models/lora/test_lora_tp.py", 150),
TestFile("test_data_parallelism.py", 90),
TestFile("test_dp_attention.py", 90),
TestFile("test_mla_tp.py", 420),
TestFile("test_dp_attention.py", 150),
TestFile("test_mla_tp.py", 174),
TestFile("test_moe_ep.py", 220),
TestFile("test_patch_torch.py", 30),
TestFile("test_update_weights_from_distributed.py", 100),