recovery launch devices in test_pass_sharding (#55194)

d6a899ff · zhaoyingli · GitHub · 190e8074 · d6a899ff
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

test/auto_parallel/test_pass_sharding.py test/auto_parallel/test_pass_sharding.py +2 -0

未找到文件。
--- a/test/auto_parallel/test_pass_sharding.py
+++ b/test/auto_parallel/test_pass_sharding.py
@@ -36,6 +36,8 @@ class TestShardingPass(unittest.TestCase):
            + [
                "-m",
                "paddle.distributed.launch",
+                "--devices",
+                "0,1",
                "--log_dir",
                tmp_dir.name,
                launch_model_path,