Skip to content

Commit 754e96a

Browse files
committed
distribute lookup table work with parallel executor
1 parent b73806b commit 754e96a

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

paddle/fluid/framework/details/multi_devices_graph_builder.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ std::unique_ptr<Graph> MultiDevSSAGraphBuilder::Apply(
207207
result.Set("ops", new GraphOps);
208208

209209
// find send/recv vars so that we can place the distributed training
210-
// realted op in the place 0
210+
// related op in the place 0
211211
auto send_vars = FindDistTrainSendVars(nodes);
212212
auto recv_vars = FindDistTrainRecvVars(nodes);
213213

python/paddle/fluid/transpiler/distribute_transpiler.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -779,7 +779,9 @@ def _replace_lookup_table_op_with_prefetch(self, program,
779779
outputs={"Out": prefetch_output_vars},
780780
attrs={
781781
"epmap": pserver_endpoints,
782-
RPC_OP_ROLE_ATTR_NAME: RPC_OP_ROLE_ATTR_VALUE
782+
# FIXME(qiao) temporarily disable this config because prefetch
783+
# is not act as other rpc op, it's more like a forward op
784+
# RPC_OP_ROLE_ATTR_NAME: RPC_OP_ROLE_ATTR_VALUE
783785
})
784786

785787
# insert concat_op

0 commit comments

Comments
 (0)