Skip to content

Commit 8ca5206

Browse files
authored
fix SerializeSelectedRows (#36543)
* bug fix for DeserializeSelectedRows. test=develop * fix bug for SerializeSelectedRows. test=develop * update. test=develop
1 parent 6524fa8 commit 8ca5206

File tree

1 file changed

+1
-13
lines changed

1 file changed

+1
-13
lines changed

paddle/fluid/distributed/service/brpc_utils.cc

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -138,23 +138,11 @@ void SerializeSelectedRows(framework::Variable* var,
138138
var_data->clear();
139139
var_data->resize(rows->size() * sizeof(int64_t));
140140
char* data_ptr = const_cast<char*>(var_data->data());
141-
142-
if (platform::is_cpu_place(tensor->place())) {
143-
memcpy(data_ptr, &(*rows)[0], rows->size() * sizeof(int64_t));
144-
} else {
145-
#ifdef PADDLE_WITH_CUDA
146-
auto stream =
147-
reinterpret_cast<const platform::CUDADeviceContext&>(ctx).stream();
148-
memory::Copy(platform::CPUPlace(), data_ptr,
149-
BOOST_GET_CONST(platform::CUDAPlace, tensor->place()),
150-
&(*rows)[0], rows->size() * sizeof(int64_t), stream);
151-
#endif
152-
}
141+
memcpy(data_ptr, &((*rows)[0]), rows->size() * sizeof(int64_t));
153142
var_msg->set_data_type(static_cast<VarMsg::Type>(tensor->type()));
154143
for (auto& dim : framework::vectorize(tensor->dims())) {
155144
var_msg->add_dims(dim);
156145
}
157-
158146
// IO Buffer
159147
if (platform::is_cpu_place(tensor->place())) {
160148
auto data_len = tensor->numel() * framework::SizeOfType(tensor->type());

0 commit comments

Comments
 (0)