未验证 提交 98e057bb 编写于 作者: D Dong Daxiang 提交者: GitHub

remove role maker in test_fleet_graph_execution_meta_optimizer.py (#26674)

test=develop
上级 dbcef732
...@@ -40,9 +40,7 @@ class TestFleetGraphExecutionMetaOptimizer(unittest.TestCase): ...@@ -40,9 +40,7 @@ class TestFleetGraphExecutionMetaOptimizer(unittest.TestCase):
def node_func(): def node_func():
import paddle.distributed.fleet as fleet import paddle.distributed.fleet as fleet
import paddle.fluid.incubate.fleet.base.role_maker as role_maker fleet.init(is_collective=True)
role = role_maker.PaddleCloudRoleMaker(is_collective=True)
fleet.init(role)
input_x = paddle.fluid.layers.data( input_x = paddle.fluid.layers.data(
name="x", shape=[32], dtype='float32') name="x", shape=[32], dtype='float32')
input_y = paddle.fluid.layers.data( input_y = paddle.fluid.layers.data(
...@@ -98,9 +96,7 @@ class TestFleetGraphExecutionMetaOptimizer(unittest.TestCase): ...@@ -98,9 +96,7 @@ class TestFleetGraphExecutionMetaOptimizer(unittest.TestCase):
def node_func(): def node_func():
import paddle.distributed.fleet as fleet import paddle.distributed.fleet as fleet
import paddle.fluid.incubate.fleet.base.role_maker as role_maker fleet.init(is_collective=True)
role = role_maker.PaddleCloudRoleMaker(is_collective=True)
fleet.init(role)
input_x = paddle.fluid.layers.data( input_x = paddle.fluid.layers.data(
name="x", shape=[32], dtype='float32') name="x", shape=[32], dtype='float32')
input_y = paddle.fluid.layers.data( input_y = paddle.fluid.layers.data(
...@@ -166,9 +162,7 @@ class TestFleetGraphExecutionMetaOptimizer(unittest.TestCase): ...@@ -166,9 +162,7 @@ class TestFleetGraphExecutionMetaOptimizer(unittest.TestCase):
def node_func(): def node_func():
import paddle.distributed.fleet as fleet import paddle.distributed.fleet as fleet
import paddle.fluid.incubate.fleet.base.role_maker as role_maker fleet.init(is_collective=True)
role = role_maker.PaddleCloudRoleMaker(is_collective=True)
fleet.init(role)
input_x = paddle.fluid.layers.data( input_x = paddle.fluid.layers.data(
name="x", shape=[32], dtype='float32') name="x", shape=[32], dtype='float32')
input_y = paddle.fluid.layers.data( input_y = paddle.fluid.layers.data(
...@@ -217,9 +211,7 @@ class TestFleetGraphExecutionMetaOptimizer(unittest.TestCase): ...@@ -217,9 +211,7 @@ class TestFleetGraphExecutionMetaOptimizer(unittest.TestCase):
def node_func(): def node_func():
import paddle.distributed.fleet as fleet import paddle.distributed.fleet as fleet
import paddle.fluid.incubate.fleet.base.role_maker as role_maker fleet.init(is_collective=True)
role = role_maker.PaddleCloudRoleMaker(is_collective=True)
fleet.init(role)
input_x = paddle.fluid.layers.data( input_x = paddle.fluid.layers.data(
name="x", shape=[32], dtype='float32') name="x", shape=[32], dtype='float32')
input_y = paddle.fluid.layers.data( input_y = paddle.fluid.layers.data(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册