m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

remove role maker in test_fleet_graph_execution_meta_optimizer.py (#26674)

Browse Source 
test=develop
revert-26856-strategy_example2
Dong Daxiang 5 years ago committed by GitHub
parent dbcef732d8
commit 98e057bb1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File

16
python/paddle/fluid/tests/unittests/test_fleet_graph_execution_meta_optimizer.py
Unescape View File

@ -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):
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):
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):
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(

Write Preview
Loading…
Cancel
Save