Skip to content

Commit e3ae4ea

Browse files
committed
Filter out DCP resharding tests from the 2 GPU FusedAdam test matrix, as those tests need to be run in sequence.
Signed-off-by: Cory Ye <cye@nvidia.com>
1 parent f095376 commit e3ae4ea

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/pytorch/distributed/test_torch_fsdp2.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,10 @@ def test_fsdp2_fused_adam_tests():
5555
"-v",
5656
"-s",
5757
"--tb=short",
58+
# The following 2 tests need to be run in sequence,
59+
# as they depend on each other.
60+
"-k",
61+
"not dcp_resharding_save and not dcp_resharding_load",
5862
],
5963
env=os.environ,
6064
timeout=600,

0 commit comments

Comments
 (0)