未验证 提交 500a8bc2 编写于 作者: W wangzhen38 提交者: GitHub

[RM FLUID] rm ps mode (#50704)

上级 4db8e5c7
# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
class PSMode:
"""
There are various mode for fleet, each of them is designed for different model.
"""
TRANSPILER = 1
PSLIB = 2
class DistributedMode:
SYNC = 0
ASYNC = 1
HALF_ASYNC = 2
GEO = 3
...@@ -25,8 +25,8 @@ from dist_test_utils import remove_ps_flag ...@@ -25,8 +25,8 @@ from dist_test_utils import remove_ps_flag
import paddle.fluid as fluid import paddle.fluid as fluid
import paddle.fluid.core as core import paddle.fluid.core as core
from paddle.fluid.framework import Program, program_guard from paddle.fluid.framework import Program, program_guard
from paddle.fluid.incubate.fleet.parameter_server.mode import DistributedMode
from paddle.fluid.op import Operator from paddle.fluid.op import Operator
from paddle.incubate.fleet.parameter_server.mode import DistributedMode
def run_pserver(pserver_id): def run_pserver(pserver_id):
......
...@@ -140,7 +140,7 @@ import paddle.fluid as fluid ...@@ -140,7 +140,7 @@ import paddle.fluid as fluid
from paddle.distributed.communicator import Communicator from paddle.distributed.communicator import Communicator
import paddle.fluid.incubate.fleet.base.role_maker as role_maker import paddle.fluid.incubate.fleet.base.role_maker as role_maker
from paddle.fluid.incubate.fleet.parameter_server.mode import DistributedMode from paddle.incubate.fleet.parameter_server.mode import DistributedMode
import paddle.distributed.fleet as fleet import paddle.distributed.fleet as fleet
from test_communicator_geo import TestCommunicatorGeoEnd2End from test_communicator_geo import TestCommunicatorGeoEnd2End
......
...@@ -49,7 +49,7 @@ from .distribute_transpiler import ( ...@@ -49,7 +49,7 @@ from .distribute_transpiler import (
same_or_split_var, same_or_split_var,
ServerRuntimeConfig, ServerRuntimeConfig,
) )
from paddle.fluid.incubate.fleet.parameter_server.mode import DistributedMode from paddle.incubate.fleet.parameter_server.mode import DistributedMode
from paddle.distributed.distribute_lookup_table import ( from paddle.distributed.distribute_lookup_table import (
find_distributed_lookup_table, find_distributed_lookup_table,
) )
......
...@@ -19,10 +19,10 @@ import warnings ...@@ -19,10 +19,10 @@ import warnings
from functools import reduce from functools import reduce
import paddle import paddle
from paddle.fluid.incubate.fleet.parameter_server.mode import DistributedMode
from paddle.framework import core from paddle.framework import core
from paddle.incubate.fleet.parameter_server.ir import vars_metatools from paddle.incubate.fleet.parameter_server.ir import vars_metatools
from paddle.incubate.fleet.parameter_server.ir.ps_dispatcher import RoundRobin from paddle.incubate.fleet.parameter_server.ir.ps_dispatcher import RoundRobin
from paddle.incubate.fleet.parameter_server.mode import DistributedMode
OP_NAME_SCOPE = "op_namescope" OP_NAME_SCOPE = "op_namescope"
CLIP_OP_NAME_SCOPE = "gradient_clip" CLIP_OP_NAME_SCOPE = "gradient_clip"
......
...@@ -28,7 +28,7 @@ from . import ps_pb2 as pslib ...@@ -28,7 +28,7 @@ from . import ps_pb2 as pslib
from .node import DownpourServer, DownpourWorker from .node import DownpourServer, DownpourWorker
OpRole = core.op_proto_and_checker_maker.OpRole OpRole = core.op_proto_and_checker_maker.OpRole
# this dict is for store info about pull/push sparse ops. # this dict is for storing info about pull/push sparse ops.
FLEET_GLOBAL_DICT = { FLEET_GLOBAL_DICT = {
# global settings # global settings
"enable": False, "enable": False,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册