[CI] update verlengine ci to 4-gpu test (#6007)
This commit is contained in:
@@ -101,7 +101,7 @@ suites = {
|
||||
TestFile("test_moe_ep.py", 181),
|
||||
TestFile("test_patch_torch.py", 19),
|
||||
TestFile("test_update_weights_from_distributed.py", 103),
|
||||
TestFile("test_verl_engine.py", 64),
|
||||
TestFile("test_verl_engine_2_gpu.py", 64),
|
||||
],
|
||||
"per-commit-2-gpu-amd": [
|
||||
TestFile("test_mla_tp.py", 170),
|
||||
@@ -109,6 +109,7 @@ suites = {
|
||||
"per-commit-4-gpu": [
|
||||
TestFile("test_local_attn.py", 250),
|
||||
TestFile("test_pp_single_node.py", 150),
|
||||
TestFile("test_verl_engine_4_gpu.py", 64),
|
||||
],
|
||||
"per-commit-8-gpu": [
|
||||
# Disabled deepep tests temporarily because it takes too much time.
|
||||
|
||||
Reference in New Issue
Block a user