louis_lifu
/
AReaL
mirror of https://github.com/inclusionAI/AReaL
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

.

This commit is contained in:
bowei.fw 2025-03-22 14:58:19 +08:00
parent bd8df23a13
commit 0300f4b94b
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/experiments/test_math_ppo.py
View File

@ -53,10 +53,10 @@ def math_code_dataset(request, save_path):
@pytest.mark.parametrize( @pytest.mark.parametrize(
"dp,pp,mp", "dp,pp,mp",
[ [
(2, 2, 1), (1, 1, 1),
# (2, 1, 2), (2, 1, 2),
# (1, 2, 1), (1, 2, 1),
# (1, 1, 2), (1, 1, 2),
], ],
) )
def test_ppo_symm( def test_ppo_symm(
@ -120,7 +120,6 @@ def test_ppo_symm(
run_test_exp(exp_cfg) run_test_exp(exp_cfg)
@pytest.mark.skip("")
# The global resharding strategy, where all MFCs # The global resharding strategy, where all MFCs
# occupy the same device mesh but with different # occupy the same device mesh but with different
# parallelization strategies. # parallelization strategies.
@ -242,7 +241,6 @@ def test_ppo_global_reshard(
run_test_exp(exp_cfg) run_test_exp(exp_cfg)
@pytest.mark.skip("")
# Actor/critic train and ref_inf/rew_inf are on disjoint # Actor/critic train and ref_inf/rew_inf are on disjoint
# device meshes and executed concurrently. # device meshes and executed concurrently.
@pytest.mark.parametrize("actor_gen", [(2, 2, 1)]) @pytest.mark.parametrize("actor_gen", [(2, 2, 1)])
@ -359,7 +357,6 @@ def test_ppo_param_realloc_sub_device_mesh(
run_test_exp(exp_cfg) run_test_exp(exp_cfg)
@pytest.mark.skip("")
@pytest.mark.parametrize("freq_step", [3, 4, 7]) @pytest.mark.parametrize("freq_step", [3, 4, 7])
@pytest.mark.parametrize("freq_epoch", [1, 2, 3]) @pytest.mark.parametrize("freq_epoch", [1, 2, 3])
@pytest.mark.parametrize("bs", [30, 80, 100]) @pytest.mark.parametrize("bs", [30, 80, 100])