update worker_num for MPISymetricRoleMaker (#20798)

test=develop

update worker_num for MPISymetricRoleMaker (#20798)
test=develop
ccbdd7aa · Dong Daxiang · GitHub · 01a96463 · ccbdd7aa
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

python/paddle/fluid/incubate/fleet/base/role_maker.py python/paddle/fluid/incubate/fleet/base/role_maker.py +1 -2

未找到文件。
--- a/python/paddle/fluid/incubate/fleet/base/role_maker.py
+++ b/python/paddle/fluid/incubate/fleet/base/role_maker.py
@@ -252,8 +252,7 @@ class MPISymetricRoleMaker(MPIRoleMaker):
        return the current number of worker
        """
        if self._check_role_generation():
-            if self.is_worker():
-                return self._get_size() / self._proc_per_node
+            return self._get_size() / self._proc_per_node
        return 0

    def _server_num(self):