From d6a899ff67b51e0ff37ebe2052adcbed936e374b Mon Sep 17 00:00:00 2001
From: zhaoyingli <86812880+zhaoyinglia@users.noreply.github.com>
Date: Thu, 6 Jul 2023 14:36:53 +0800
Subject: [PATCH] recovery launch devices in test_pass_sharding (#55194)

---
 test/auto_parallel/test_pass_sharding.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/test/auto_parallel/test_pass_sharding.py b/test/auto_parallel/test_pass_sharding.py
index d9f514e9473..d6062f216f6 100644
--- 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,
-- 
GitLab