Skip to content

Commit 8fc05e0

Browse files
authored
fix cpu build test=develop (#14260)
1 parent 4dbc018 commit 8fc05e0

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

paddle/fluid/operators/ref_by_trainer_id_op.h

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ class RefByTrainerIdKernel : public framework::OpKernel<T> {
2626
auto* out = context.Output<framework::Tensor>("Out");
2727
auto in_list = context.MultiInput<framework::Tensor>("X");
2828
auto* trainer_id_t = context.Input<framework::Tensor>("TrainerId");
29-
int64_t trainer_id;
29+
int64_t trainer_id = 0;
3030
auto* trainer_id_data = trainer_id_t->data<int64_t>();
3131
if (platform::is_gpu_place(context.GetPlace())) {
3232
#ifdef PADDLE_WITH_CUDA
@@ -38,7 +38,6 @@ class RefByTrainerIdKernel : public framework::OpKernel<T> {
3838
} else {
3939
trainer_id = *trainer_id_data;
4040
}
41-
printf("after get trainer_id %lu\n", trainer_id);
4241
PADDLE_ENFORCE_LT(trainer_id, in_list.size());
4342
out->mutable_data<T>(context.GetPlace());
4443
out->ShareDataWith(*(in_list[trainer_id]));

python/paddle/fluid/transpiler/distribute_transpiler.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1588,7 +1588,6 @@ def _append_dc_asgd_ops(self, block, param_var, grad_var):
15881588
ref_inputs = []
15891589
for p, p_bak in self.param_bak_list:
15901590
if p.name == param_var.name:
1591-
print("#### ref inputs: ", param_var.name, p_bak.name)
15921591
ref_inputs.append(p_bak)
15931592
block.append_op(
15941593
type="ref_by_trainer_id",

0 commit comments

Comments
 (0)